@SuppressWarnings("static-method") @Test public void uncheckedInjection() { // set exceptions via a string final String exceptionDesc = "<<injected from descPassthroughMethod()>>"; final String exceptionClassName = "java.lang.RuntimeException"; final String jsonString = "{\"injections\":[{" + "\"type\":\"exception\"," + "\"siteClass\":\"com.dremio.exec.testing.TestExceptionInjection$DummyClass\"," + "\"desc\":\"" + exceptionDesc + "\"," + "\"nSkip\":0," + "\"nFire\":1," + "\"exceptionClass\":\"" + exceptionClassName + "\"" + "}]}"; ControlsInjectionUtil.setControls(session, jsonString); final QueryContext context = new QueryContext(session, nodes[0].getContext(), QueryId.getDefaultInstance()); // test that the exception gets thrown final DummyClass dummyClass = new DummyClass(context); assertPassthroughThrows(dummyClass, exceptionClassName, exceptionDesc); try { context.close(); } catch (Exception e) { fail(); } }
@SuppressWarnings("static-method") @Test public void checkedInjection() { // set the injection via the parsing POJOs final String controls = Controls.newBuilder() .addException(DummyClass.class, DummyClass.THROWS_IOEXCEPTION, IOException.class, 0, 1) .build(); ControlsInjectionUtil.setControls(session, controls); final QueryContext context = new QueryContext(session, nodes[0].getContext(), QueryId.getDefaultInstance()); // test that the expected exception (checked) gets thrown final DummyClass dummyClass = new DummyClass(context); try { dummyClass.throwsIOException(); fail(NO_THROW_FAIL); } catch (IOException e) { assertEquals(DummyClass.THROWS_IOEXCEPTION, e.getMessage()); } try { context.close(); } catch (Exception e) { fail(); } }
@SuppressWarnings("static-method") @Test public void skipAndLimit() { final String passthroughDesc = "<<injected from descPassthrough>>"; final int nSkip = 7; final int nFire = 3; final Class<? extends Throwable> exceptionClass = RuntimeException.class; final String controls = Controls.newBuilder() .addException(DummyClass.class, passthroughDesc, exceptionClass, nSkip, nFire) .build(); ControlsInjectionUtil.setControls(session, controls); final QueryContext context = new QueryContext(session, nodes[0].getContext(), QueryId.getDefaultInstance()); final DummyClass dummyClass = new DummyClass(context); // these shouldn't throw for (int i = 0; i < nSkip; ++i) { dummyClass.descPassthroughMethod(passthroughDesc); } // these should throw for (int i = 0; i < nFire; ++i) { assertPassthroughThrows(dummyClass, exceptionClass.getName(), passthroughDesc); } // this shouldn't throw dummyClass.descPassthroughMethod(passthroughDesc); try { context.close(); } catch (Exception e) { fail(); } }
@Test public void pauseInjected() { final long expectedDuration = 1000L; final ExtendedLatch trigger = new ExtendedLatch(1); final Pointer<Exception> ex = new Pointer<>(); final String controls = Controls.newBuilder() .addPause(DummyClass.class, DummyClass.PAUSES) .build(); ControlsInjectionUtil.setControls(session, controls); final QueryContext queryContext = new QueryContext(session, nodes[0].getContext(), QueryId.getDefaultInstance()); (new ResumingThread(queryContext, trigger, ex, expectedDuration)).start(); // test that the pause happens final DummyClass dummyClass = new DummyClass(queryContext, trigger); final long actualDuration = dummyClass.pauses(); assertTrue(String.format("Test should stop for at least %d milliseconds.", expectedDuration), expectedDuration <= actualDuration); assertTrue("No exception should be thrown.", ex.value == null); try { queryContext.close(); } catch (final Exception e) { fail("Failed to close query context: " + e); } }
queryContext.close(); } catch (final Exception e) { fail("Failed to close query context: " + e);