@Override public void onTransmissionSent(TransmissionHandlerArgs args) { validateTransmissionAndSend(args); }
private boolean generateTransmissionWithStatusCodeAndHeader(int code, String retryHeader) { TransmissionPolicyManager tpm = new TransmissionPolicyManager(true); TransmissionDispatcher mockedDispatcher = Mockito.mock(TransmissionDispatcher.class); TransmissionHandlerArgs args = new TransmissionHandlerArgs(); args.setResponseCode(code); args.setTransmission(new Transmission(new byte[] { 0 }, "testcontent", "testencoding")); args.setTransmissionDispatcher(mockedDispatcher); args.setRetryHeader(new BasicHeader(RESPONSE_THROTTLING_HEADER, retryHeader)); ThrottlingHandler eh = new ThrottlingHandler(tpm); boolean result = eh.validateTransmissionAndSend(args); return result; }
private boolean generateTransmissionWithStatusCode(int code) { TransmissionPolicyManager tpm = new TransmissionPolicyManager(true); TransmissionDispatcher mockedDispatcher = Mockito.mock(TransmissionDispatcher.class); TransmissionHandlerArgs args = new TransmissionHandlerArgs(); args.setResponseCode(code); args.setTransmission(new Transmission(new byte[] { 0 }, "testcontent", "testencoding")); args.setTransmissionDispatcher(mockedDispatcher); ThrottlingHandler eh = new ThrottlingHandler(tpm); boolean result = eh.validateTransmissionAndSend(args); return result; }
@Test public void failException() { TransmissionPolicyManager tpm = new TransmissionPolicyManager(true); TransmissionDispatcher mockedDispatcher = Mockito.mock(TransmissionDispatcher.class); TransmissionHandlerArgs args = new TransmissionHandlerArgs(); args.setResponseCode(0); args.setTransmission(null); args.setTransmissionDispatcher(mockedDispatcher); args.setException(new Exception("Mocked")); ThrottlingHandler eh = new ThrottlingHandler(tpm); boolean result = eh.validateTransmissionAndSend(args); Assert.assertFalse(result); }