@AfterTestWithContext public void afterTest() { assertThat(testContextIfSet().isPresent()).isTrue(); }
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 IndexedRunnable withChildTestContext(IndexedRunnable runnable) { return (int threadIndex) -> { pushTestContext(testContext().createChildContext()); try { runnable.run(threadIndex); } finally { popTestContext(); } }; }
public static void runWithTestContext(TestContext testContext, Runnable runnable) { pushTestContext(testContext); try { runnable.run(); } finally { popTestContext(); } }
@Override public void onTestStart(ITestResult testResult) { setupLoggingMdcForTest(testResult); checkState(suiteTestContextStack.isPresent(), "test suite not initialized"); GuiceTestContext initTestContext = suiteTestContextStack.get().peek().createChildContext(emptyList(), getTestModules(testResult)); TestContextStack<GuiceTestContext> testContextStack = new TestContextStack<>(); testContextStack.push(initTestContext); try { Set<Requirement> testSpecificRequirements = getTestSpecificRequirements(testResult.getMethod()); doFulfillment(testContextStack, testMethodLevelFulfillers, testSpecificRequirements); } catch (RuntimeException e) { LOGGER.debug("error within test initialization", e); throw e; } assertTestContextNotSet(); pushAllTestContexts(testContextStack); GuiceTestContext topTestContext = testContextStack.peek(); topTestContext.injectMembers(testResult.getInstance()); runBeforeWithContextMethods(testResult, topTestContext); }
public static TestContext testContext() { assertTestContextSet(); return testContextStackThreadLocal.get().peek(); }
public static void pushTestContext(TestContext testContext) { ensureTestContextStack(); testContextStackThreadLocal.get().push(testContext); }
private void runBeforeWithContextMethods(ITestResult testResult, GuiceTestContext testContext) { try { invokeMethodsAnnotatedWith(BeforeTestWithContext.class, testResult, testContext); } catch (RuntimeException e) { TestContextStack<GuiceTestContext> testContextStack = (TestContextStack) popAllTestContexts(); doCleanup(testContextStack, testMethodLevelFulfillers, FAILURE); throw e; } }
public static TestContext popTestContext() { assertTestContextSet(); TestContextStack<TestContext> testContextStack = testContextStackThreadLocal.get(); TestContext testContext = testContextStack.pop(); if (testContextStack.empty()) { testContextStackThreadLocal.remove(); } return testContext; }
public static QueryExecutor onPresto() { return testContext().getDependency(QueryExecutor.class, "presto"); }
@BeforeTestWithContext public void beforeTest() { assertThat(testContextIfSet().isPresent()).isTrue(); }
public static QueryExecutor connectToPresto(String prestoConfig) { return testContext().getDependency(QueryExecutor.class, prestoConfig); }
@BeforeTestWithContext public void beforeTest() { assertThat(testContextIfSet().isPresent()).isTrue(); }
public static QueryExecutor onHive() { return testContext().getDependency(QueryExecutor.class, "hive"); }
@AfterTestWithContext public void afterTest() { assertThat(testContextIfSet().isPresent()).isTrue(); }
public static QueryExecutor onMySql() { return testContext().getDependency(QueryExecutor.class, "mysql"); }
private Optional<String> getCurrentTestLogFileName() { Optional<TestContext> testContext = testContextIfSet(); try { String testName = "SUITE"; if (testContext.isPresent()) { Optional<TestMetadata> testMetadata = testContext.get().getOptionalDependency(TestMetadata.class); if (testMetadata.isPresent()) { testName = testMetadata.get().testName; } } return Optional.of(logsDirectory + "/" + testName); } catch (ConfigurationException e) { System.err.append("Could not load TestMetadata from guice context"); return Optional.empty(); } }
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); }