public static QueryExecutor onSqlServer() { return testContext().getDependency(QueryExecutor.class, "sqlserver"); }
private void onTestFinished(ITestResult testResult, TestStatus testStatus) { if (!testContextIfSet().isPresent()) { return; } boolean runAfterSucceeded = false; try { runAfterWithContextMethods(testResult, (GuiceTestContext) testContext()); runAfterSucceeded = true; } finally { TestContextStack<GuiceTestContext> testContextStack = (TestContextStack) popAllTestContexts(); doCleanup(testContextStack, testMethodLevelFulfillers, runAfterSucceeded ? testStatus : FAILURE); cleanLoggingMdc(); } }
public static QueryExecutor onPresto() { return testContext().getDependency(QueryExecutor.class, "presto"); }
public static QueryExecutor connectToPresto(String prestoConfig) { return testContext().getDependency(QueryExecutor.class, prestoConfig); }
public static QueryExecutor onHive() { return testContext().getDependency(QueryExecutor.class, "hive"); }
public static QueryExecutor onMySql() { return testContext().getDependency(QueryExecutor.class, "mysql"); }
private static void executeHiveQuery(String query) { testContext().getDependency(QueryExecutor.class, "hive").executeQuery(query); } }
private static TableInstance mutableTableInstanceOf(TableHandle tableHandle) { return testContext().getDependency(MutableTablesState.class).get(tableHandle); }
private static TableInstance mutableTableInstanceOf(TableHandle tableHandle) { return testContext().getDependency(MutableTablesState.class).get(tableHandle); }
public static Optional<TestContext> testContextIfSet() { if (testContextStackThreadLocal.get() == null) { return Optional.empty(); } return Optional.of(testContext()); }
public static ImmutableTablesState immutableTablesState() { return testContext().getDependency(ImmutableTablesState.class); }
static TableManagerDispatcher getTableManagerDispatcher() { return testContext().getDependency(TableManagerDispatcher.class); } }
static QueryExecutor defaultQueryExecutor() { return testContext().getDependency(QueryExecutor.class, DEFAULT_DB_NAME); }
public static MutableTablesState mutableTablesState() { return testContext().getDependency(MutableTablesState.class); }
private QueryExecutor getQueryExecutor(SqlQueryDescriptor sqlQueryDescriptor) { String database = sqlQueryDescriptor.getDatabaseName(); try { return testContext().getDependency(QueryExecutor.class, database); } catch (RuntimeException e) { throw new RuntimeException("Cannot get query executor for database '" + database + "'", e); } } }
public static void runWithChildTestContext(Runnable runnable) { runWithTestContext(testContext().createChildContext(), runnable); }
public static IndexedRunnable withChildTestContext(IndexedRunnable runnable) { return (int threadIndex) -> { pushTestContext(testContext().createChildContext()); try { runnable.run(threadIndex); } finally { popTestContext(); } }; }
@Test(groups = "query") @Requires(MultipleTablesTestRequirements.class) public void selectAllFromMultipleTables() { MutableTablesState mutableTablesState = testContext().getDependency(MutableTablesState.class); TableInstance tableInstance = mutableTablesState.get("table"); assertThat(query("select * from " + tableInstance.getNameInDatabase())).hasAnyRows(); }