FailOnTimeoutStatementBuilderTest.java

package de.bechte.junit.runners.context.statements.builder;

import de.bechte.junit.runners.model.TestClassPool;
import de.bechte.junit.stubs.singletests.SingleTestWithTimeoutAnnotation;
import de.bechte.junit.stubs.singletests.SingleTestWithoutAnnotations;
import org.junit.Before;
import org.junit.Test;
import org.junit.internal.runners.statements.FailOnTimeout;
import org.junit.runner.Description;
import org.junit.runner.notification.RunNotifier;
import org.junit.runners.model.FrameworkMethod;
import org.junit.runners.model.Statement;
import org.junit.runners.model.TestClass;
import org.mockito.Mock;

import static org.hamcrest.Matchers.*;
import static org.junit.Assert.assertThat;
import static org.mockito.MockitoAnnotations.initMocks;

public class FailOnTimeoutStatementBuilderTest {
    @Mock
    private Object target;
    @Mock private Statement next;
    @Mock private Description description;
    @Mock private RunNotifier notifier;
    @Mock private Test testAnnotation;

    private FailOnTimeoutStatementBuilder builder = new FailOnTimeoutStatementBuilder();

    @Before
    public void setUp() throws Exception {
        initMocks(this);
    }

    @Test
    public void whenTestAnnotationDoesNotSpecifyATimeout_returnNextStatement() throws Exception {
        TestClass testClass = TestClassPool.forClass(SingleTestWithoutAnnotations.class);
        FrameworkMethod method = testClass.getAnnotatedMethods(Test.class).get(0);

        Statement actual = builder.createStatement(testClass, method, target, next, description, notifier);
        assertThat(actual, is(equalTo(next)));
    }

    @Test
    public void whenTestAnnotationDoesSpecifyATimeout_returnFailOnTimeoutStatement() throws Exception {
        TestClass testClass = TestClassPool.forClass(SingleTestWithTimeoutAnnotation.class);
        FrameworkMethod method = testClass.getAnnotatedMethods(Test.class).get(0);

        Statement actual = builder.createStatement(testClass, method, target, next, description, notifier);
        assertThat(actual, is(instanceOf(FailOnTimeout.class)));
    }
}