JTableFixture_withMocks_Test.java

/**
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
 * the License. You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
 * an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
 * specific language governing permissions and limitations under the License.
 *
 * Copyright 2012-2015 the original author or authors.
 */
package org.assertj.swing.fixture;

import static java.awt.Color.BLUE;
import static org.assertj.core.api.Assertions.assertThat;
import static org.assertj.swing.core.MouseButton.LEFT_BUTTON;
import static org.assertj.swing.core.MouseButton.MIDDLE_BUTTON;
import static org.assertj.swing.core.MouseClickInfo.middleButton;
import static org.assertj.swing.data.TableCell.row;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

import java.awt.Font;
import java.awt.Point;
import java.util.regex.Pattern;

import javax.swing.JPopupMenu;
import javax.swing.JTable;
import javax.swing.table.JTableHeader;

import org.assertj.swing.cell.JTableCellReader;
import org.assertj.swing.cell.JTableCellWriter;
import org.assertj.swing.core.MouseClickInfo;
import org.assertj.swing.core.Robot;
import org.assertj.swing.data.TableCell;
import org.assertj.swing.data.TableCellFinder;
import org.assertj.swing.driver.JTableDriver;
import org.junit.Before;
import org.junit.Test;

/**
 * Tests for {@link JTableFixture}.
 *
 * @author Alex Ruiz
 * @author Yvonne Wang
 */
public class JTableFixture_withMocks_Test {
  private TableCell cell;
  private JTableDriver driver;
  private JTable target;

  private JTableFixture fixture;

  @Before
  public void setUp() {
    cell = row(6).column(8);
    fixture = new JTableFixture(mock(Robot.class), mock(JTable.class));
    fixture.replaceDriverWith(mock(JTableDriver.class));
    driver = fixture.driver();
    target = fixture.target();
  }

  @Test
  public void should_Return_Font_At_Cell_Using_Driver() {
    Font font = mock(Font.class);
    when(driver.font(target, cell)).thenReturn(font);
    FontFixture fontFixture = fixture.fontAt(cell);
    assertThat(fontFixture.target()).isSameAs(font);
    verify(driver).font(target, cell);
  }

  @Test
  public void should_Return_Background_At_Cell_Using_Driver() {
    when(driver.background(target, cell)).thenReturn(BLUE);
    ColorFixture colorFixture = fixture.backgroundAt(cell);
    assertThat(colorFixture.target()).isSameAs(BLUE);
    verify(driver).background(target, cell);
  }

  @Test
  public void should_Return_Foreground_At_Cell_Using_Driver() {
    when(driver.foreground(target, cell)).thenReturn(BLUE);
    ColorFixture colorFixture = fixture.foregroundAt(cell);
    assertThat(colorFixture.target()).isSameAs(BLUE);
    verify(driver).foreground(target, cell);
  }

  @Test
  public void should_Return_Cell_With_Value_Using_Driver() {
    when(driver.cell(target, "Hello")).thenReturn(cell);
    JTableCellFixture cellFixture = fixture.cell("Hello");
    assertThat(cellFixture.cell()).isSameAs(cell);
    assertThat(cellFixture.tableFixture()).isSameAs(fixture);
    verify(driver).cell(target, "Hello");
  }

  @Test
  public void should_Return_Cell_With_Pattern_Using_Driver() {
    Pattern pattern = Pattern.compile("Hello");
    when(driver.cell(target, pattern)).thenReturn(cell);
    JTableCellFixture cellFixture = fixture.cell(pattern);
    assertThat(cellFixture.cell()).isSameAs(cell);
    assertThat(cellFixture.tableFixture()).isSameAs(fixture);
    verify(driver).cell(target, pattern);
  }

  @Test
  public void should_Return_Cell_With_TableCellFinder_Using_Driver() {
    TableCellFinder cellFinder = mock(TableCellFinder.class);
    when(driver.cell(target, cellFinder)).thenReturn(cell);
    JTableCellFixture cellFixture = fixture.cell(cellFinder);
    assertThat(cellFixture.cell()).isSameAs(cell);
    assertThat(cellFixture.tableFixture()).isSameAs(fixture);
    verify(driver).cell(target, cellFinder);
  }

  @Test
  public void should_Return_Cell_Using_Driver() {
    JTableCellFixture cellFixture = fixture.cell(cell);
    assertThat(cellFixture.cell()).isSameAs(cell);
    assertThat(cellFixture.tableFixture()).isSameAs(fixture);
    verify(driver).checkCellIndicesInBounds(target, cell);
  }

  @Test
  public void should_Return_JTableHeader_Using_Driver() {
    JTableHeader header = mock(JTableHeader.class);
    when(driver.tableHeaderOf(target)).thenReturn(header);
    JTableHeaderFixture headerFixture = fixture.tableHeader();
    assertThat(headerFixture.target()).isSameAs(header);
    verify(driver).tableHeaderOf(target);
  }

  @Test
  public void should_Return_Selection_Using_Driver() {
    when(driver.selectionValue(target)).thenReturn("Hello");
    assertThat(fixture.selectionValue()).isEqualTo("Hello");
    verify(driver).selectionValue(target);
  }

  @Test
  public void should_Convert_Cell_To_Point_Using_Driver() {
    Point p = new Point(6, 8);
    when(driver.pointAt(target, cell)).thenReturn(p);
    assertThat(fixture.pointAt(cell)).isSameAs(p);
    verify(driver).pointAt(target, cell);
  }

  @Test
  public void should_Return_Contents_Using_Driver() {
    String[][] contents = { { "0", "1", "2" } };
    when(driver.contents(target)).thenReturn(contents);
    assertThat(fixture.contents()).isSameAs(contents);
    verify(driver).contents(target);
  }

  @Test
  public void should_Return_Row_Count_Using_Driver() {
    when(driver.rowCountOf(target)).thenReturn(6);
    assertThat(fixture.rowCount()).isEqualTo(6);
    verify(driver).rowCountOf(target);
  }

  @Test
  public void should_Return_Value_At_Cell_Using_Driver() {
    when(driver.value(target, cell)).thenReturn("Hello");
    assertThat(fixture.valueAt(cell)).isEqualTo("Hello");
    verify(driver).value(target, cell);
  }

  @Test
  public void should_Call_SelectCell_In_Driver_And_Return_Self() {
    assertThat(fixture.selectCell(cell)).isSameAs(fixture);
    verify(driver).selectCell(target, cell);
  }

  @Test
  public void should_Call_SelectCells_In_Driver_And_Return_Self() {
    TableCell[] cells = { cell };
    assertThat(fixture.selectCells(cells)).isSameAs(fixture);
    verify(driver).selectCells(target, cells);
  }

  @Test
  public void should_Call_SelectRows_In_Driver_And_Return_Self() {
    int rows[] = { 6, 8 };
    assertThat(fixture.selectRows(rows)).isSameAs(fixture);
    verify(driver).selectRows(target, rows);
  }

  @Test
  public void should_Call_UnselectCell_In_Driver_And_Return_Self() {
    assertThat(fixture.unselectCell(cell)).isSameAs(fixture);
    verify(driver).unselectCell(target, cell);
  }

  @Test
  public void should_Call_UnselectCells_In_Driver_And_Return_Self() {
    TableCell[] cells = { cell };
    assertThat(fixture.unselectCells(cells)).isSameAs(fixture);
    verify(driver).unselectCells(target, cells);
  }

  @Test
  public void should_Call_UnselectRows_In_Driver_And_Return_Self() {
    int rows[] = { 6, 8 };
    assertThat(fixture.unselectRows(rows)).isSameAs(fixture);
    verify(driver).unselectRows(target, rows);
  }

  @Test
  public void should_Call_Drag_In_Driver_And_Return_Self() {
    assertThat(fixture.drag(cell)).isSameAs(fixture);
    verify(driver).drag(target, cell);
  }

  @Test
  public void should_Call_Drop_In_Driver_And_Return_Self() {
    assertThat(fixture.drop(cell)).isSameAs(fixture);
    verify(driver).drop(target, cell);
  }

  @Test
  public void should_Call_Click_With_MouseButton_In_Driver_And_Return_Self() {
    assertThat(fixture.click(cell, LEFT_BUTTON)).isSameAs(fixture);
    verify(driver).click(target, cell, LEFT_BUTTON, 1);
  }

  @Test
  public void should_Call_Click_With_MouseClickInfo_In_Driver_And_Return_Self() {
    MouseClickInfo info = middleButton().times(3);
    assertThat(fixture.click(cell, info)).isSameAs(fixture);
    verify(driver).click(target, cell, MIDDLE_BUTTON, 3);
  }

  @Test
  public void should_Call_EnterValueInCell_In_Driver_And_Return_Self() {
    assertThat(fixture.enterValue(cell, "Hello")).isSameAs(fixture);
    verify(driver).enterValueInCell(target, cell, "Hello");
  }

  @Test
  public void should_Call_ReplaceCellReader_In_Driver() {
    JTableCellReader cellReader = mock(JTableCellReader.class);
    fixture.replaceCellReader(cellReader);
    verify(driver).replaceCellReader(cellReader);
  }

  @Test
  public void should_Call_RequireRowCount_In_Driver_And_Return_Self() {
    assertThat(fixture.requireRowCount(6)).isSameAs(fixture);
    verify(driver).requireRowCount(target, 6);
  }

  @Test
  public void should_Call_RequireSelectedRows_In_Driver_And_Return_Self() {
    int[] rows = { 6, 8 };
    assertThat(fixture.requireSelectedRows(rows)).isSameAs(fixture);
    verify(driver).requireSelectedRows(target, rows);
  }

  @Test
  public void should_Call_RequireColumnCount_In_Driver_And_Return_Self() {
    assertThat(fixture.requireColumnCount(6)).isSameAs(fixture);
    verify(driver).requireColumnCount(target, 6);
  }

  @Test
  public void should_Call_RequireEditable_In_Driver_And_Return_Self() {
    assertThat(fixture.requireEditable(cell)).isSameAs(fixture);
    verify(driver).requireEditable(target, cell);
  }

  @Test
  public void should_Call_RequireNotEditable_In_Driver_And_Return_Self() {
    assertThat(fixture.requireNotEditable(cell)).isSameAs(fixture);
    verify(driver).requireNotEditable(target, cell);
  }

  @Test
  public void should_Call_RequireNoSelection_In_Driver_And_Return_Self() {
    assertThat(fixture.requireNoSelection()).isSameAs(fixture);
    verify(driver).requireNoSelection(target);
  }

  @Test
  public void should_Call_RequireCellValue_With_Text_In_Driver_And_Return_Self() {
    assertThat(fixture.requireCellValue(cell, "Hello")).isSameAs(fixture);
    verify(driver).requireCellValue(target, cell, "Hello");
  }

  @Test
  public void should_Call_RequireCellValue_With_Pattern_In_Driver_And_Return_Self() {
    Pattern pattern = Pattern.compile("Hello");
    assertThat(fixture.requireCellValue(cell, pattern)).isSameAs(fixture);
    verify(driver).requireCellValue(target, cell, pattern);
  }

  @Test
  public void should_Call_RequireContents_In_Driver_And_Return_Self() {
    String[][] contents = { { "0", "1", "2" } };
    assertThat(fixture.requireContents(contents)).isSameAs(fixture);
    verify(driver).requireContents(target, contents);
  }

  @Test
  public void should_Call_ReplaceCellWriter_In_Driver() {
    JTableCellWriter cellWriter = mock(JTableCellWriter.class);
    fixture.replaceCellWriter(cellWriter);
    verify(driver).replaceCellWriter(cellWriter);
  }

  @Test
  public void should_Return_Column_Index_Using_Driver() {
    when(driver.columnIndex(target, "Name")).thenReturn(2);
    assertThat(fixture.columnIndexFor("Name")).isEqualTo(2);
    verify(driver).columnIndex(target, "Name");
  }

  @Test
  public void should_Return_Self_When_Column_Exists() {
    when(driver.columnIndex(target, "Name")).thenReturn(2);
    assertThat(fixture.requireColumnNamed("Name")).isSameAs(fixture);
    verify(driver).columnIndex(target, "Name");
  }

  @Test(expected = AssertionError.class)
  public void should_Throw_Error_When_Column_Not_Exists() {
    when(driver.columnIndex(target, "Name")).thenThrow(new AssertionError());
    fixture.requireColumnNamed("Name");
  }

  @Test
  public void should_Return_JPopupMenu_With_Cell_Using_Driver() {
    JPopupMenu popupMenu = mock(JPopupMenu.class);
    when(driver.showPopupMenuAt(target, cell)).thenReturn(popupMenu);
    JPopupMenuFixture popupMenuFixture = fixture.showPopupMenuAt(cell);
    assertThat(popupMenuFixture.target()).isSameAs(popupMenu);
    verify(driver).showPopupMenuAt(target, cell);
  }
}