Search in sources :

Example 1 with TestingWarningCollector

use of io.prestosql.testing.TestingWarningCollector in project hetu-core by openlookeng.

the class TestJdbcWarnings method testStatementWarnings.

@Test
public void testStatementWarnings() throws SQLException {
    assertFalse(statement.execute("CREATE SCHEMA blackhole.test_schema"));
    SQLWarning warning = statement.getWarnings();
    assertNotNull(warning);
    TestingWarningCollectorConfig warningCollectorConfig = new TestingWarningCollectorConfig().setPreloadedWarnings(PRELOADED_WARNINGS);
    TestingWarningCollector warningCollector = new TestingWarningCollector(new WarningCollectorConfig(), warningCollectorConfig);
    List<PrestoWarning> expectedWarnings = warningCollector.getWarnings();
    assertStartsWithExpectedWarnings(warning, fromPrestoWarnings(expectedWarnings));
    statement.clearWarnings();
    assertNull(statement.getWarnings());
}
Also used : SQLWarning(java.sql.SQLWarning) WarningCollectorConfig(io.prestosql.execution.warnings.WarningCollectorConfig) TestingWarningCollectorConfig(io.prestosql.testing.TestingWarningCollectorConfig) TestingWarningCollectorConfig(io.prestosql.testing.TestingWarningCollectorConfig) PrestoWarning(io.prestosql.spi.PrestoWarning) TestingWarningCollector(io.prestosql.testing.TestingWarningCollector) Test(org.testng.annotations.Test)

Example 2 with TestingWarningCollector

use of io.prestosql.testing.TestingWarningCollector in project hetu-core by openlookeng.

the class TestJdbcWarnings method testExecuteQueryWarnings.

@Test
public void testExecuteQueryWarnings() throws SQLException {
    try (ResultSet rs = statement.executeQuery("SELECT a FROM (VALUES 1, 2, 3) t(a)")) {
        assertNull(statement.getConnection().getWarnings());
        assertNull(statement.getWarnings());
        assertNull(rs.getWarnings());
        Set<WarningEntry> currentWarnings = new HashSet<>();
        while (rs.next()) {
            assertWarnings(rs.getWarnings(), currentWarnings);
        }
        TestingWarningCollectorConfig warningCollectorConfig = new TestingWarningCollectorConfig().setPreloadedWarnings(PRELOADED_WARNINGS).setAddWarnings(true);
        TestingWarningCollector warningCollector = new TestingWarningCollector(new WarningCollectorConfig(), warningCollectorConfig);
        List<PrestoWarning> expectedWarnings = warningCollector.getWarnings();
        for (PrestoWarning prestoWarning : expectedWarnings) {
            assertTrue(currentWarnings.contains(new WarningEntry(toPrestoSqlWarning(prestoWarning))));
        }
    }
}
Also used : WarningCollectorConfig(io.prestosql.execution.warnings.WarningCollectorConfig) TestingWarningCollectorConfig(io.prestosql.testing.TestingWarningCollectorConfig) TestingWarningCollectorConfig(io.prestosql.testing.TestingWarningCollectorConfig) ResultSet(java.sql.ResultSet) PrestoWarning(io.prestosql.spi.PrestoWarning) TestingWarningCollector(io.prestosql.testing.TestingWarningCollector) HashSet(java.util.HashSet) Test(org.testng.annotations.Test)

Example 3 with TestingWarningCollector

use of io.prestosql.testing.TestingWarningCollector in project hetu-core by openlookeng.

the class TestTestingWarningCollector method testAddWarnings.

@Test
public void testAddWarnings() {
    TestingWarningCollector collector = new TestingWarningCollector(new WarningCollectorConfig(), new TestingWarningCollectorConfig().setAddWarnings(true));
    ImmutableList.Builder<PrestoWarning> expectedWarningsBuilder = ImmutableList.builder();
    expectedWarningsBuilder.add(createTestWarning(1));
    assertEquals(collector.getWarnings(), expectedWarningsBuilder.build());
}
Also used : TestingWarningCollectorConfig(io.prestosql.testing.TestingWarningCollectorConfig) TestingWarningCollectorConfig(io.prestosql.testing.TestingWarningCollectorConfig) ImmutableList(com.google.common.collect.ImmutableList) PrestoWarning(io.prestosql.spi.PrestoWarning) TestingWarningCollector(io.prestosql.testing.TestingWarningCollector) Test(org.testng.annotations.Test)

Example 4 with TestingWarningCollector

use of io.prestosql.testing.TestingWarningCollector in project hetu-core by openlookeng.

the class TestCompletedEventWarnings method testCompletedEventWarnings.

@Test
public void testCompletedEventWarnings() throws InterruptedException {
    TestingWarningCollectorConfig warningCollectorConfig = new TestingWarningCollectorConfig().setPreloadedWarnings(TEST_WARNINGS);
    TestingWarningCollector testingWarningCollector = new TestingWarningCollector(new WarningCollectorConfig(), warningCollectorConfig);
    assertWarnings("select 1", ImmutableMap.of(), testingWarningCollector.getWarnings().stream().map(PrestoWarning::getWarningCode).collect(toImmutableList()));
}
Also used : WarningCollectorConfig(io.prestosql.execution.warnings.WarningCollectorConfig) TestingWarningCollectorConfig(io.prestosql.testing.TestingWarningCollectorConfig) TestingWarningCollectorConfig(io.prestosql.testing.TestingWarningCollectorConfig) PrestoWarning(io.prestosql.spi.PrestoWarning) TestingWarningCollector(io.prestosql.testing.TestingWarningCollector) Test(org.testng.annotations.Test)

Aggregations

PrestoWarning (io.prestosql.spi.PrestoWarning)4 TestingWarningCollector (io.prestosql.testing.TestingWarningCollector)4 TestingWarningCollectorConfig (io.prestosql.testing.TestingWarningCollectorConfig)4 Test (org.testng.annotations.Test)4 WarningCollectorConfig (io.prestosql.execution.warnings.WarningCollectorConfig)3 ImmutableList (com.google.common.collect.ImmutableList)1 ResultSet (java.sql.ResultSet)1 SQLWarning (java.sql.SQLWarning)1 HashSet (java.util.HashSet)1