org.mule.transport.jdbc.functional.JdbcEndpointWithNestedQueriesTestCase.java Source code

Introduction

Here is the source code for org.mule.transport.jdbc.functional.JdbcEndpointWithNestedQueriesTestCase.java

Source

/*
 * $Id$
 * --------------------------------------------------------------------------------------
 * Copyright (c) MuleSoft, Inc.  All rights reserved.  http://www.mulesoft.com
 *
 * The software in this package is published under the terms of the CPAL v1.0
 * license, a copy of which has been included with this distribution in the
 * LICENSE.txt file.
 */

package org.mule.transport.jdbc.functional;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.fail;

import org.mule.api.client.MuleClient;

import java.sql.Connection;
import java.util.Arrays;
import java.util.Collection;
import java.util.List;

import org.apache.commons.dbutils.QueryRunner;
import org.apache.commons.dbutils.handlers.ArrayListHandler;
import org.junit.Test;
import org.junit.runners.Parameterized;

/**
 * Tests Mule disposal in the context of nested queries on JDBC endpoints. This test is related to MULE-6607.
 */
public class JdbcEndpointWithNestedQueriesTestCase extends AbstractJdbcFunctionalTestCase {
    public JdbcEndpointWithNestedQueriesTestCase(ConfigVariant variant, String configResources) {
        super(variant, configResources);
        setPopulateTestData(false);
    }

    @Parameterized.Parameters
    public static Collection<Object[]> parameters() {
        return Arrays
                .asList(new Object[][] { { ConfigVariant.FLOW, "jdbc-endpoint-with-nested-queries-test.xml" } });
    }

    @Test
    public void testDisposeAfterQueryExecution() throws Exception {
        QueryRunner queryRunner = jdbcConnector.getQueryRunner();
        Connection connection = jdbcConnector.getConnection();

        // Send a message to trigger nested query on JDBC outbound endpoint to execute.
        MuleClient client = muleContext.getClient();
        client.send("vm://in", "some test data", null);

        // Assert that query executed correctly.
        List<?> results = (List<?>) queryRunner.query(connection, "SELECT * FROM TEST", new ArrayListHandler());
        assertEquals(1, results.size());

        // Try to dispose gracefully.
        try {
            muleContext.dispose();
        } catch (Exception ex) {
            fail("Server disposal failed");
        }
    }
}