Package io.trino.testing

Class QueryAssertions

java.lang.Object

io.trino.testing.QueryAssertions

public final class QueryAssertions
extends Object

Method Summary

Modifier and Type	Method	Description
static void	assertContains(MaterializedResult all, MaterializedResult expectedSubset)
static void	assertContainsEventually(Supplier<MaterializedResult> all, MaterializedResult expectedSubset, io.airlift.units.Duration timeout)
static void	assertEqualsIgnoreOrder(Iterable<?> actual, Iterable<?> expected)
static void	assertEqualsIgnoreOrder(Iterable<?> actual, Iterable<?> expected, String message)
static void	assertQuery(QueryRunner actualQueryRunner, Session session, String actual, H2QueryRunner h2QueryRunner, String expected, boolean ensureOrdering, boolean compareUpdate)
static void	assertQuery(QueryRunner actualQueryRunner, Session session, String actual, H2QueryRunner h2QueryRunner, String expected, boolean ensureOrdering, boolean compareUpdate, Consumer<Plan> planAssertion)
static void	assertQueryEventually(QueryRunner actualQueryRunner, Session session, String actual, H2QueryRunner h2QueryRunner, String expected, boolean ensureOrdering, boolean compareUpdate, Optional<Consumer<Plan>> planAssertion, io.airlift.units.Duration timeout)
protected static void	assertQueryFails(QueryRunner queryRunner, Session session, String sql, String expectedMessageRegExp)
protected static void	assertQueryFailsEventually(QueryRunner queryRunner, Session session, String sql, String expectedMessageRegExp, io.airlift.units.Duration timeout)
protected static void	assertQueryReturnsEmptyResult(QueryRunner queryRunner, Session session, String sql)
protected static void	assertQuerySucceeds(QueryRunner queryRunner, Session session, String sql)
static void	assertUpdate(QueryRunner queryRunner, Session session, String sql, OptionalLong count, Optional<Consumer<Plan>> planAssertion)
static void	copyTable(QueryRunner queryRunner, QualifiedObjectName table, Session session)
static void	copyTable(QueryRunner queryRunner, String sourceCatalog, String sourceSchema, String sourceTable, Session session)
static void	copyTpchTables(QueryRunner queryRunner, String sourceCatalog, String sourceSchema, Session session, Iterable<io.trino.tpch.TpchTable<?>> tables)
static RuntimeException	getTrinoExceptionCause(Throwable e)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Details

assertUpdate

public static void assertUpdate(QueryRunner queryRunner,
 Session session,
 String sql,
 OptionalLong count,
 Optional<Consumer<Plan>> planAssertion)

assertQuery

public static void assertQuery(QueryRunner actualQueryRunner,
 Session session,
 String actual,
 H2QueryRunner h2QueryRunner,
 String expected,
 boolean ensureOrdering,
 boolean compareUpdate)

assertQuery

public static void assertQuery(QueryRunner actualQueryRunner,
 Session session,
 String actual,
 H2QueryRunner h2QueryRunner,
 String expected,
 boolean ensureOrdering,
 boolean compareUpdate,
 Consumer<Plan> planAssertion)

assertQueryEventually

public static void assertQueryEventually(QueryRunner actualQueryRunner,
 Session session,
 String actual,
 H2QueryRunner h2QueryRunner,
 String expected,
 boolean ensureOrdering,
 boolean compareUpdate,
 Optional<Consumer<Plan>> planAssertion,
 io.airlift.units.Duration timeout)

assertEqualsIgnoreOrder

public static void assertEqualsIgnoreOrder(Iterable<?> actual,
 Iterable<?> expected)

assertEqualsIgnoreOrder

public static void assertEqualsIgnoreOrder(Iterable<?> actual,
 Iterable<?> expected,
 String message)

assertContainsEventually

public static void assertContainsEventually(Supplier<MaterializedResult> all,
 MaterializedResult expectedSubset,
 io.airlift.units.Duration timeout)

assertContains

public static void assertContains(MaterializedResult all,
 MaterializedResult expectedSubset)

assertQuerySucceeds

protected static void assertQuerySucceeds(QueryRunner queryRunner,
 Session session,
 String sql)

assertQueryFailsEventually

protected static void assertQueryFailsEventually(QueryRunner queryRunner,
 Session session,
 String sql,
 String expectedMessageRegExp,
 io.airlift.units.Duration timeout)

assertQueryFails

protected static void assertQueryFails(QueryRunner queryRunner,
 Session session,
 String sql,
 String expectedMessageRegExp)

assertQueryReturnsEmptyResult

protected static void assertQueryReturnsEmptyResult(QueryRunner queryRunner,
 Session session,
 String sql)

copyTpchTables

public static void copyTpchTables(QueryRunner queryRunner,
 String sourceCatalog,
 String sourceSchema,
 Session session,
 Iterable<io.trino.tpch.TpchTable<?>> tables)

copyTable

public static void copyTable(QueryRunner queryRunner,
 String sourceCatalog,
 String sourceSchema,
 String sourceTable,
 Session session)

copyTable

public static void copyTable(QueryRunner queryRunner,
 QualifiedObjectName table,
 Session session)

getTrinoExceptionCause

public static RuntimeException getTrinoExceptionCause(Throwable e)