@Test public void picksCorrectDrawColor() throws Exception { // Arrange when(resultMock.isDraw()).thenReturn(true); // Act Color pickedColor = colorPicker.pickResultColor(resultMock); // Assert assertThat(pickedColor, is(Colors.DRAW_BG_COLOR)); }
@Before public void setUp() throws Exception { MockitoAnnotations.initMocks(this); when(resultMock.isHomeWin()).thenReturn(false); when(resultMock.isDraw()).thenReturn(false); when(resultMock.isAwayWin()).thenReturn(false); colorPicker = new ColorPicker(); }