Examples with ClientFixture org.apache.drill.test.ClientFixture used on opensource projects

Example 31 with ClientFixture

use of org.apache.drill.test.ClientFixture in project drill by apache.

the class TestLateralPlans method testUnnestTableAndColumnAlias.

@Test
public void testUnnestTableAndColumnAlias() throws Exception {
    String sql = "select t.c_name from cp.`lateraljoin/nested-customer.json` t, unnest(t.orders) ";
    ClusterFixtureBuilder builder = ClusterFixture.builder(dirTestWatcher).setOptionDefault(PlannerSettings.ENABLE_UNNEST_LATERAL_KEY, true);
    try (ClusterFixture cluster = builder.build();
        ClientFixture client = cluster.clientFixture()) {
        client.queryBuilder().sql(sql).run();
    } catch (UserRemoteException ex) {
        assertTrue(ex.getMessage().contains("Alias table and column name are required for UNNEST"));
    }
}

Example 32 with ClientFixture

use of org.apache.drill.test.ClientFixture in project drill by apache.

the class TestConfigLinkage method testSessionPrecedence.

/* Test if altering session option takes precedence over system option */
@Test
public void testSessionPrecedence() throws Exception {
    ClusterFixtureBuilder builder = ClusterFixture.builder(dirTestWatcher).systemOption(ExecConstants.SLICE_TARGET, 100000);
    try (ClusterFixture cluster = builder.build();
        ClientFixture client = cluster.clientFixture()) {
        client.queryBuilder().sql("ALTER SESSION SET `planner.slice_target` = 10000").run();
        String slice_target = client.queryBuilder().sql("SELECT val FROM sys.%s where name='planner.slice_target' and optionScope = 'SESSION'", SystemTable.OPTIONS.getTableName()).singletonString();
        assertEquals(slice_target, "10000");
    }
}

Example 33 with ClientFixture

use of org.apache.drill.test.ClientFixture in project drill by apache.

the class StatusResourcesTest method testRetrievePublicOption.

@Test
public void testRetrievePublicOption() throws Exception {
    ClusterFixtureBuilder builder = ClusterFixture.builder(dirTestWatcher).configProperty(ExecConstants.HTTP_ENABLE, true).configProperty(ExecConstants.HTTP_PORT_HUNT, true).configProperty(ExecConstants.SYS_STORE_PROVIDER_LOCAL_ENABLE_WRITE, false).systemOption(ExecConstants.SLICE_TARGET, 20);
    try (ClusterFixture cluster = builder.build();
        ClientFixture client = cluster.clientFixture();
        RestClientFixture restClientFixture = cluster.restClientFixture()) {
        Assert.assertNull(restClientFixture.getStatusInternalOption(ExecConstants.SLICE_TARGET));
        StatusResources.OptionWrapper option = restClientFixture.getStatusOption(ExecConstants.SLICE_TARGET);
        Assert.assertEquals(20, option.getValue());
        client.alterSystem(ExecConstants.SLICE_TARGET, 30);
        Assert.assertNull(restClientFixture.getStatusInternalOption(ExecConstants.SLICE_TARGET));
        option = restClientFixture.getStatusOption(ExecConstants.SLICE_TARGET);
        Assert.assertEquals(30, option.getValue());
    }
}

Example 34 with ClientFixture

use of org.apache.drill.test.ClientFixture in project drill by apache.

the class StatusResourcesTest method testRetrieveInternalOption.

@Test
public void testRetrieveInternalOption() throws Exception {
    OptionDefinition optionDefinition = createMockPropOptionDefinition();
    ClusterFixtureBuilder builder = ClusterFixture.builder(dirTestWatcher).configProperty(ExecConstants.HTTP_ENABLE, true).configProperty(ExecConstants.bootDefaultFor(MOCK_PROPERTY), "a").configProperty(ExecConstants.HTTP_PORT_HUNT, true).configProperty(ExecConstants.SYS_STORE_PROVIDER_LOCAL_ENABLE_WRITE, false).putDefinition(optionDefinition);
    try (ClusterFixture cluster = builder.build();
        ClientFixture client = cluster.clientFixture();
        RestClientFixture restClientFixture = cluster.restClientFixture()) {
        Assert.assertNull(restClientFixture.getStatusOption(MOCK_PROPERTY));
        StatusResources.OptionWrapper option = restClientFixture.getStatusInternalOption(MOCK_PROPERTY);
        Assert.assertEquals("a", option.getValueAsString());
        client.alterSystem(MOCK_PROPERTY, "c");
        Assert.assertNull(restClientFixture.getStatusOption(MOCK_PROPERTY));
        option = restClientFixture.getStatusInternalOption(MOCK_PROPERTY);
        Assert.assertEquals("c", option.getValueAsString());
    }
}

Example 35 with ClientFixture

use of org.apache.drill.test.ClientFixture in project drill by apache.

the class TestHtpasswdFileUserAuthenticator method tryCredentials.

private static void tryCredentials(String user, String password, ClusterFixture cluster, boolean shouldSucceed) throws Exception {
    try {
        ClientFixture client = cluster.clientBuilder().property(DrillProperties.USER, user).property(DrillProperties.PASSWORD, password).build();
        // Run few queries using the new client
        List<String> queries = Arrays.asList("SHOW SCHEMAS", "USE INFORMATION_SCHEMA", "SHOW TABLES", "SELECT * FROM INFORMATION_SCHEMA.`TABLES` WHERE TABLE_NAME LIKE 'COLUMNS'", "SELECT * FROM cp.`region.json` LIMIT 5");
        for (String query : queries) {
            client.queryBuilder().sql(query).run();
        }
        if (!shouldSucceed) {
            fail("Expected connect to fail because of incorrect username / password combination, but it succeeded");
        }
    } catch (IllegalStateException e) {
        if (shouldSucceed) {
            throw e;
        }
    }
}