@Test
public void failsOnFirstOfTwoAssertions() throws Throwable {
thrown.expectMessage(equalTo("Assertion failed."));
rule.checkAssertionAfterwards(INVALID_ASSERTION);
rule.checkAssertionAfterwards(VALID_ASSERTION);
executeRuleWithoutExitCall();
}
@Test
public void failsOnExitWithInvalidAssertion() throws Throwable {
thrown.expectMessage(equalTo("Assertion failed."));
rule.expectSystemExit();
rule.checkAssertionAfterwards(INVALID_ASSERTION);
executeRuleWithExitStatus0();
}
@Test
public void succeedOnExitWithValidAssertion() throws Throwable {
rule.expectSystemExit();
rule.checkAssertionAfterwards(
new Assertion() {
public void checkAssertion() throws Exception {
assertTrue(true);
}
});
executeRuleWithExitStatus0();
}
@Test
public void failBecauseOfMissingSystemExitCall() throws Throwable {
thrown.expectMessage("System.exit has not been called.");
rule.expectSystemExit();
executeRuleWithoutExitCall();
}
@Test
public void failForWrongStatus() throws Throwable {
thrown.expectMessage("Wrong exit status");
rule.expectSystemExitWithStatus(1);
executeRuleWithExitStatus0();
}
@Test
public void succeedOnExitWithSelectedStatusCode() throws Throwable {
rule.expectSystemExitWithStatus(0);
executeRuleWithExitStatus0();
}
@Test
public void succeedOnExitWithArbitraryStatusCode() throws Throwable {
rule.expectSystemExit();
executeRuleWithExitStatus0();
}
public class ExpectedSystemExitTest {
private static final Object ARBITRARY_CONTEXT = new Object();
private static final int ARBITRARY_EXIT_STATUS = 216843;
private static final Assertion INVALID_ASSERTION =
new Assertion() {
public void checkAssertion() throws Exception {
fail("Assertion failed.");
}
};
private static final Assertion VALID_ASSERTION =
new Assertion() {
public void checkAssertion() throws Exception {}
};
@Rule public final ExpectedException thrown = none().handleAssertionErrors();
private final ExpectedSystemExit rule = ExpectedSystemExit.none();
@Test
public void succeedWithoutExit() throws Throwable {
executeRuleWithoutExitCall();
}
@Test
public void succeedOnExitWithArbitraryStatusCode() throws Throwable {
rule.expectSystemExit();
executeRuleWithExitStatus0();
}
@Test
public void succeedOnExitWithSelectedStatusCode() throws Throwable {
rule.expectSystemExitWithStatus(0);
executeRuleWithExitStatus0();
}
@Test
public void failForUnexpectedSystemExit() throws Throwable {
thrown.expectMessage("Unexpected call of System.exit(0).");
executeRuleWithExitStatus0();
}
@Test
public void failBecauseOfMissingSystemExitCall() throws Throwable {
thrown.expectMessage("System.exit has not been called.");
rule.expectSystemExit();
executeRuleWithoutExitCall();
}
@Test
public void failForWrongStatus() throws Throwable {
thrown.expectMessage("Wrong exit status");
rule.expectSystemExitWithStatus(1);
executeRuleWithExitStatus0();
}
@Test
public void succeedOnExitWithValidAssertion() throws Throwable {
rule.expectSystemExit();
rule.checkAssertionAfterwards(
new Assertion() {
public void checkAssertion() throws Exception {
assertTrue(true);
}
});
executeRuleWithExitStatus0();
}
@Test
public void failsOnExitWithInvalidAssertion() throws Throwable {
thrown.expectMessage(equalTo("Assertion failed."));
rule.expectSystemExit();
rule.checkAssertionAfterwards(INVALID_ASSERTION);
executeRuleWithExitStatus0();
}
@Test
public void failsOnFirstOfTwoAssertions() throws Throwable {
thrown.expectMessage(equalTo("Assertion failed."));
rule.checkAssertionAfterwards(INVALID_ASSERTION);
rule.checkAssertionAfterwards(VALID_ASSERTION);
executeRuleWithoutExitCall();
}
@Test
public void failsOnSecondOfTwoAssertions() throws Throwable {
thrown.expectMessage(equalTo("Assertion failed."));
rule.checkAssertionAfterwards(VALID_ASSERTION);
rule.checkAssertionAfterwards(INVALID_ASSERTION);
executeRuleWithoutExitCall();
}
@Test
public void restoreOldSecurityManager() throws Throwable {
SecurityManager manager = new ArbitrarySecurityManager();
setSecurityManager(manager);
executeRuleWithoutExitCall();
assertThat(getSecurityManager(), sameInstance(manager));
}
@Test
public void delegateToOldSecurityManager() throws Throwable {
SecurityManager manager = new ArbitrarySecurityManager();
setSecurityManager(manager);
executeRuleWithStatement(new CheckContext());
}
@Test
public void succeedsOnExitInThread() throws Throwable {
rule.expectSystemExitWithStatus(ARBITRARY_EXIT_STATUS);
executeRuleWithStatement(new SystemExitInThread());
}
private void executeRuleWithoutExitCall() throws Throwable {
executeRuleWithStatement(new EmptyStatement());
}
private void executeRuleWithExitStatus0() throws Throwable {
executeRuleWithStatement(new SystemExit0());
}
private void executeRuleWithStatement(Statement statement) throws Throwable {
rule.apply(statement, null).evaluate();
}
private static class SystemExit0 extends Statement {
@Override
public void evaluate() throws Throwable {
System.exit(0);
}
}
private static class CheckContext extends Statement {
@Override
public void evaluate() throws Throwable {
assertEquals(ARBITRARY_CONTEXT, getSecurityManager().getSecurityContext());
}
}
private static class SystemExitInThread extends Statement {
@Override
public void evaluate() throws Throwable {
Runnable callSystemExit =
new Runnable() {
public void run() {
System.exit(ARBITRARY_EXIT_STATUS);
}
};
Thread thread = new Thread(callSystemExit);
thread.start();
sleep(1000); // wait until the thread exits
}
}
private static class ArbitrarySecurityManager extends SecurityManager {
@Override
public Object getSecurityContext() {
return ARBITRARY_CONTEXT;
}
@Override
public void checkPermission(Permission perm) {
// allow anything.
}
}
}
private void executeRuleWithStatement(Statement statement) throws Throwable {
rule.apply(statement, null).evaluate();
}
@Test
public void succeedsOnExitInThread() throws Throwable {
rule.expectSystemExitWithStatus(ARBITRARY_EXIT_STATUS);
executeRuleWithStatement(new SystemExitInThread());
}
