source: TraumaOntologies/src/test/java/tudelft/healthpsychology/traumaontologies/answerstate/ListOfAnswersStateTest.java

package tudelft.healthpsychology.traumaontologies.answerstate;

import static org.junit.Assert.assertEquals;
import static org.mockito.Matchers.anyString;
import static org.mockito.Matchers.eq;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

import java.util.Arrays;
import java.util.Collection;
import java.util.Collections;

import org.junit.Test;

import tudelft.healthpsychology.traumaontologies.questiontypes.TypedQuestion;
import tudelft.utilities.tree.Tree;

public class ListOfAnswersStateTest {

        private Tree<String, Collection<Property>, OntologyNode> tree = mock(
                        Tree.class);

        @Test
        public void smokeTest() {
                new ListOfAnswersState(Collections.emptyList());
        }

        @Test
        public void testgetOptions() {
                // we should get options from first AnswerState with non-null options
                AnswerState state1 = mock(AnswerState.class);
                AnswerState state2 = mock(AnswerState.class);
                when(state1.getOptions(tree)).thenReturn(null);
                TypedQuestion options = mock(TypedQuestion.class);
                when(state2.getOptions(tree)).thenReturn(options);

                ListOfAnswersState las = new ListOfAnswersState(
                                Arrays.asList(state1, state2));
                assertEquals(options, las.getOptions(tree));

        }

        @Test
        public void testgetOptionsFinished() {
                // we should get options from first AnswerState with non-null options
                AnswerState state1 = mock(AnswerState.class);
                when(state1.getOptions(tree)).thenReturn(null);

                ListOfAnswersState las = new ListOfAnswersState(Arrays.asList(state1));
                assertEquals(null, las.getOptions(tree));

        }

        @Test
        public void testWith() {
                // with should be passed to first non-null state
                AnswerState state1 = mock(AnswerState.class);
                AnswerState state2 = mock(AnswerState.class);
                when(state1.getOptions(tree)).thenReturn(null);
                TypedQuestion options = mock(TypedQuestion.class);
                when(options.fits(anyString())).thenReturn(true);
                when(state2.getOptions(tree)).thenReturn(options);

                ListOfAnswersState las = new ListOfAnswersState(
                                Arrays.asList(state1, state2));
                las.with("bla", tree);
                verify(state2, times(1)).with(eq("bla"), eq(tree));
        }

        @Test(expected = IllegalStateException.class)
        public void testWithFinished() {
                // we can't do with on final state.
                AnswerState state1 = mock(AnswerState.class);
                when(state1.getOptions(tree)).thenReturn(null);

                ListOfAnswersState las = new ListOfAnswersState(Arrays.asList(state1));
                las.with("bla", tree);
        }

}