@Test
public void with_circuit_breaker() throws Throwable {
// exactly as before but with a circuit breaker
CircuitBreakerPolicy.addCircuitBreaker("cb1", 500, 1);
WithIntegrationPoint example = new WithIntegrationPoint();
try {
example.run(1);
} catch (MyOwnException e) {
try {
// repeat but get different exception
example.run(1);
} catch (CircuitOpenException ee) {
try {
new DejaVuEngine().replay(callback.getTrace());
Assert.fail("Must throw CircuitOpenException");
} catch (CircuitOpenException eee) {
}
}
}
}
@Test
public void exceed_threshold() {
CircuitBreaker breaker = new CircuitBreaker("cb1", 500, 2);
CircuitBreakerPolicy.addCircuitBreaker(breaker);
WithIntegrationPoint example = new WithIntegrationPoint();
try {
Assert.assertTrue(breaker.isClosed());
example.run(1);
Assert.fail("first crash");
} catch (MyOwnException e) {
Assert.assertTrue("breaker must still be closed", breaker.isClosed());
try {
example.run(1);
Assert.fail("second crash");
} catch (MyOwnException ee) {
Assert.assertTrue(breaker.isOpen());
try {
example.run(1);
Assert.fail("third must be of type CircuitOpenException");
} catch (CircuitOpenException eee) {
Assert.assertTrue(breaker.isOpen());
try {
Thread.sleep(600);
// should now be set to half open
Assert.assertTrue(breaker.isHalfOpen());
// should succeed
example.run(0);
Assert.assertTrue(breaker.isClosed());
} catch (InterruptedException e1) {
Assert.fail();
}
}
}
}
}
