package de.bechte.junit.runners.context.statements.builder;
import org.junit.AfterClass;
import org.junit.Before;
import org.junit.Test;
import org.junit.internal.runners.statements.RunAfters;
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 java.util.Arrays;
import java.util.Collections;
import java.util.List;
import static org.hamcrest.Matchers.*;
import static org.junit.Assert.assertThat;
import static org.mockito.Mockito.when;
import static org.mockito.MockitoAnnotations.initMocks;
public class AfterClassStatementBuilderTest {
@Mock private TestClass testClass;
@Mock private Statement next;
@Mock private Description description;
@Mock private RunNotifier notifier;
@Mock private FrameworkMethod method1;
@Mock private FrameworkMethod method2;
private AfterClassStatementBuilder builder = new AfterClassStatementBuilder();
@Before
public void setUp() throws Exception {
initMocks(this);
}
@Test
public void givenTestClassWithoutAfterClassAnnotatedMethods_returnsNextStatement() throws Exception {
when(testClass.getAnnotatedMethods(AfterClass.class)).thenReturn(Collections.EMPTY_LIST);
Statement statement = builder.createStatement(testClass, next, description, notifier);
assertThat(statement, is(equalTo(next)));
}
@Test
public void givenTestClassWithAfterClassAnnotatedMethods_returnsRunAfterStatement() throws Exception {
List<FrameworkMethod> afters = Arrays.asList(method1, method2);
when(testClass.getAnnotatedMethods(AfterClass.class)).thenReturn(afters);
Statement actual = builder.createStatement(testClass, next, description, notifier);
assertThat(actual, is(instanceOf(RunAfters.class)));
}
}