MainWindowTest.java :  » Database-Client » SQL-Workbench » workbench » gui » Java Open Source

/*
 * MainWindowTest.java
 *
 * This file is part of SQL Workbench/J, http://www.sql-workbench.net
 *
 * Copyright 2002-2008, Thomas Kellerer
 * No part of this code maybe reused without the permission of the author
 *
 * To contact the author please send an email to: support@sql-workbench.net
 *
 */
package workbench.gui;

import javax.swing.JMenuBar;
import javax.swing.JMenuItem;
import javax.swing.JTabbedPane;
import junit.framework.TestCase;
import org.netbeans.jemmy.QueueTool;
import org.netbeans.jemmy.operators.JButtonOperator;
import org.netbeans.jemmy.operators.JComboBoxOperator;
import org.netbeans.jemmy.operators.JComponentOperator;
import org.netbeans.jemmy.operators.JDialogOperator;
import org.netbeans.jemmy.operators.JFrameOperator;
import org.netbeans.jemmy.operators.JLabelOperator;
import org.netbeans.jemmy.operators.JListOperator;
import org.netbeans.jemmy.operators.JMenuBarOperator;
import org.netbeans.jemmy.operators.JMenuItemOperator;
import org.netbeans.jemmy.operators.JMenuOperator;
import org.netbeans.jemmy.operators.JTableOperator;
import org.netbeans.jemmy.operators.JTextFieldOperator;
import org.netbeans.jemmy.operators.Operator.DefaultStringComparator;
import org.netbeans.jemmy.operators.Operator.StringComparator;
import workbench.db.ConnectionMgr;
import workbench.gui.actions.AppendResultsAction;
import workbench.gui.sql.SqlPanel;
import workbench.util.StringUtil;
import workbench.util.WbFile;

/**
 * @author support@sql-workbench.net
 */
public class MainWindowTest
  extends TestCase
{
  private GuiTestUtil testUtil;

  public MainWindowTest(String testName)
  {
    super(testName);
    this.testUtil = new GuiTestUtil("MainWindowTest");
  }

  private void startApplication()
  {
    try
    {
      testUtil.startApplication();
    }
    catch (Exception e)
    {
      e.printStackTrace();
      fail(e.getMessage());
    }
  }

  private void aboutTest()
  {
    JFrameOperator mainWindow = new JFrameOperator("SQL Workbench");
    new JMenuBarOperator(mainWindow).pushMenuNoBlock("Help|About", "|");
    
    JDialogOperator dialog = new JDialogOperator(mainWindow, "About SQL Workbench/J");
    dialog.setVisible(false);
    
    QueueTool tool = new QueueTool();
    tool.waitEmpty();
    dialog.dispose();
    
  }  
  
  private void whatsNewTest()
  {
    JFrameOperator mainWindow = new JFrameOperator("SQL Workbench");
    new JMenuBarOperator(mainWindow).pushMenuNoBlock("Help|What's new", "|");
    QueueTool tool = new QueueTool();
    tool.waitEmpty();
    
    JDialogOperator dialog = new JDialogOperator(mainWindow, "What's new");
    dialog.setVisible(false);
    dialog.dispose();
  }  
  
  private void settingsTest()
  {
    JFrameOperator mainWindow = new JFrameOperator("SQL Workbench");
    new JMenuBarOperator(mainWindow).pushMenuNoBlock("Tools|Options", "|");
    JDialogOperator dialog = new JDialogOperator(mainWindow, "Settings");
    final JListOperator pages = new JListOperator(dialog);

    int count = pages.getModel().getSize();
    assertEquals(10, count);

    NamedComponentChooser chooser = new NamedComponentChooser();
    chooser.setName("pagetitle");

    QueueTool tool = new QueueTool();
    for (int i = 1; i < count; i++)
    {
      final int index = i;
      pages.selectItem(index);
      tool.waitEmpty();

      String pg = pages.getSelectedValue().toString();
      JLabelOperator title = new JLabelOperator(dialog, chooser);
      assertEquals(pg, title.getText());
    }

    new JButtonOperator(dialog, "Cancel").push();
  }

//  private void createDriver()
//  {
//    JFrameOperator mainWindow = new JFrameOperator("SQL Workbench");
//    new JMenuBarOperator(mainWindow).pushMenuNoBlock("File|Manage Drivers", "|");
//    JDialogOperator dialog = new JDialogOperator(mainWindow, "Manage drivers");
//    JListOperator list = new JListOperator(dialog);
//    list.selectItem("H2 Database Engine");
//    new JButtonOperator(dialog, "Cancel").push();
//  }

  private void definePKTest()
  {
    NamedComponentChooser chooser = new NamedComponentChooser();
    chooser.setName("sqleditor1");
    JFrameOperator mainWindow = new JFrameOperator("SQL Workbench");
    JMenuBarOperator mainMenu = new JMenuBarOperator(mainWindow);

    QueueTool tool = new QueueTool();
    
    chooser.setName("sqlpanel1");
    JComponentOperator panel = new JComponentOperator(mainWindow, chooser);
    final SqlPanel sqlPanel = (SqlPanel)panel.getSource();

    JMenuOperator dataMenu = new JMenuOperator(mainMenu.getMenu(3));
    JMenuItem saveItem = (JMenuItem)dataMenu.getMenuComponent(1);
    final JMenuItemOperator save = new JMenuItemOperator(saveItem);
//    assertFalse(save.isEnabled());
    
    runSql(sqlPanel, "create table nopk (id1 integer, data varchar(20));\n" +
      "commit;\n" +
      "insert into nopk (id1, data) values (1,'Ford');\n" +
      "insert into nopk (id1, data) values (2,'Zaphod');\n" +
      "commit;\n");
    runSql(sqlPanel, "select id1, data from nopk;");

    JTableOperator result = new JTableOperator(mainWindow);
    int rows = result.getRowCount();
    assertEquals(2, rows);
    assertEquals(2, result.getColumnCount());
    result.setValueAt("Arthur", 0, 1);
    
    tool.waitEmpty();
    assertEquals(3, result.getColumnCount());
    assertTrue(save.isEnabled());
    
    new JMenuBarOperator(mainWindow).pushMenuNoBlock("Data|Define key columns", "|");    
    JDialogOperator definePK = new JDialogOperator("Select Key Columns");

    JTableOperator cols = new JTableOperator(definePK);
    cols.setValueAt(Boolean.TRUE, 0, 1);
    JButtonOperator ok = new JButtonOperator(definePK, "OK");
    ok.push();
    tool.waitEmpty();
    
    saveChanges(sqlPanel);
    new JMenuBarOperator(mainWindow).pushMenuNoBlock("Data|Save changes to database", "|");    
    testUtil.waitWhileBusy(sqlPanel);
    tool.waitEmpty();
    
    assertEquals(2, result.getColumnCount());
    runSql(sqlPanel, "select id1, data from nopk where data = 'Arthur';");
    
    result = new JTableOperator(mainWindow);
    rows = result.getRowCount();
    assertEquals(1, rows);
  }  
  
  private void pkWarningsTest()
  {
    NamedComponentChooser chooser = new NamedComponentChooser();
    chooser.setName("sqleditor1");
    JFrameOperator mainWindow = new JFrameOperator("SQL Workbench");
    JMenuBarOperator mainMenu = new JMenuBarOperator(mainWindow);

    QueueTool tool = new QueueTool();
    
    chooser.setName("sqlpanel1");
    JComponentOperator panel = new JComponentOperator(mainWindow, chooser);
    final SqlPanel sqlPanel = (SqlPanel)panel.getSource();

    JMenuOperator dataMenu = new JMenuOperator(mainMenu.getMenu(3));
    JMenuItem saveItem = (JMenuItem)dataMenu.getMenuComponent(1);
    final JMenuItemOperator save = new JMenuItemOperator(saveItem);
    assertFalse(save.isEnabled());
    
    runSql(sqlPanel, "create table jtest (id1 integer, id2 integer, data varchar(20), primary key (id1, id2));\n" +
      "commit;\n" +
      "insert into jtest (id1, id2, data) values (1,1,'Ford');\n" +
      "commit;\n");
    runSql(sqlPanel, "select id1, data from jtest;");

    JTableOperator result = new JTableOperator(mainWindow);
    int rows = result.getRowCount();
    assertEquals(1, rows);
    assertEquals(2, result.getColumnCount());
    result.setValueAt("Arthur", 0, 1);
    
    tool.waitEmpty();
    assertEquals(3, result.getColumnCount());
    assertTrue(save.isEnabled());
    
    new JMenuBarOperator(mainWindow).pushMenuNoBlock("Data|Save changes to database", "|");    
    JDialogOperator warning = new JDialogOperator("Missing key columns");
    JButtonOperator cancel = new JButtonOperator(warning, "Cancel");
    cancel.push();
    tool.waitEmpty();
    assertEquals(3, result.getColumnCount());
  }
  
  private void connect()
  {
    // Make sure not profiles exist
    ConnectionMgr.getInstance().clearProfiles();
    
    JFrameOperator mainWindow = new JFrameOperator("SQL Workbench");
    JMenuBar bar = mainWindow.getJMenuBar();
    //    JMenuBarOperator mainMenu = new JMenuBarOperator(mainWindow);
    JMenuBarOperator menu = new JMenuBarOperator(bar);
    menu.pushMenuNoBlock("File|Connect", "|");

    JDialogOperator dialog = new JDialogOperator(mainWindow, "Select Connection Profile");
    JTextFieldOperator profileName = new JTextFieldOperator(dialog, "New Profile");
    profileName.setText("Test Connection");

    NamedComponentChooser chooser = new NamedComponentChooser();
    chooser.setName("driverclass");

    JComboBoxOperator driver = new JComboBoxOperator(dialog, chooser);
    driver.setToolTipText("test");
    StringComparator comp = new DefaultStringComparator(false, false);

    int index = driver.findItemIndex("H2 Database Engine", comp);
    if (index <= 0) fail("H2 Driver not found");
    driver.selectItem(index);

    chooser.setName("url");
    JTextFieldOperator url = new JTextFieldOperator(dialog, chooser);
    WbFile db = new WbFile(testUtil.getBaseDir(), "testdb");
    url.setText("jdbc:h2:" + db.getFullPath());

    chooser.setName("username");
    JTextFieldOperator username = new JTextFieldOperator(dialog, chooser);
    username.setText("sa");
    new JButtonOperator(dialog, "OK").push();
    
    // Connecting can take some time...
    QueueTool tool = new QueueTool();
    tool.waitEmpty();
    chooser.setName("sqlpanel1");
    JComponentOperator panel = new JComponentOperator(mainWindow, chooser);
    SqlPanel sqlPanel = (SqlPanel)panel.getSource();
    testUtil.waitUntilConnected(sqlPanel);
  }

  private void runSql()
  {
    NamedComponentChooser chooser = new NamedComponentChooser();
    chooser.setName("sqleditor1");
    JFrameOperator mainWindow = new JFrameOperator("SQL Workbench");
    JMenuBarOperator mainMenu = new JMenuBarOperator(mainWindow);

    chooser.setName("sqlpanel1");
    JComponentOperator panel = new JComponentOperator(mainWindow, chooser);
    final SqlPanel sqlPanel = (SqlPanel)panel.getSource();

    String msg = runSql(sqlPanel, "create table person (nr integer primary key, firstname varchar(20), lastname varchar(20));");
    System.out.println("Create message: " + msg);
    assertTrue(msg.indexOf("Object created") > -1);

    msg = runSql(sqlPanel, "insert into person (nr, firstname, lastname) values (42, 'Ford', 'Prefect');\ncommit;");
    assertNotNull(msg);
    assertTrue(msg.indexOf("1 row(s) affected") > -1);

    runSql(sqlPanel, "select nr, firstname, lastname from person");

    JTableOperator result = new JTableOperator(mainWindow);
    int rows = result.getRowCount();
    assertEquals(1, rows);
    assertEquals(3, result.getColumnCount());

    Object nr = result.getValueAt(0, 0);
    assertEquals(nr, new Integer(42));

    JMenuOperator dataMenu = new JMenuOperator(mainMenu.getMenu(3));
    JMenuItem saveItem = (JMenuItem)dataMenu.getMenuComponent(1);
    JMenuItemOperator save = new JMenuItemOperator(saveItem);
    assertFalse(save.isEnabled());

    result.setValueAt("Arthur", 0, 1);

    // The first call to setValueAt() will make the result table display
    // the status column
    assertEquals(4, result.getColumnCount());

    // because of the status column the lastname column 
    // is the column with index 3 (not 2)
    result.setValueAt("Dent", 0, 3);

    assertTrue(save.isEnabled());

    saveChanges(sqlPanel);

    // Make sure the status column is turned off after saving
    assertEquals(3, result.getColumnCount());

    runSql(sqlPanel, "select nr, firstname, lastname from person where lastname = 'Dent';");

    // Obtain a new referenct to the result table as the 
    // SQLPanel has created a new instance when running the select
    result = new JTableOperator(mainWindow);
    assertEquals(1, result.getRowCount());
    assertEquals(3, result.getColumnCount());

    String firstname = (String)result.getValueAt(0, 1);
    assertEquals("Arthur", firstname);

    msg = runSql(sqlPanel, "update person set firstname = null where nr = 42;\ncommit;");
    System.out.println("update message: " + msg);
    assertTrue(msg.indexOf("1 row(s) affected") > -1);

    msg = runSql(sqlPanel, "select nr, firstname, lastname from person where lastname = 'Dent';");
    System.out.println("Message: " + msg);

    result = new JTableOperator(mainWindow);
    firstname = (String)result.getValueAt(0, 1);
    assertTrue(StringUtil.isWhitespaceOrEmpty(firstname));

    result.setValueAt("Arthur", 0, 1);
    QueueTool tool = new QueueTool();
    tool.waitEmpty();
    msg = saveChanges(sqlPanel);
    tool.waitEmpty();
  }

  private void appendTest()
  {
    JFrameOperator mainWindow = new JFrameOperator("SQL Workbench");
    NamedComponentChooser chooser = new NamedComponentChooser();
    chooser.setName("sqlpanel1");
    JComponentOperator panel = new JComponentOperator(mainWindow, chooser);
    final SqlPanel sqlPanel = (SqlPanel)panel.getSource();

    JMenuBarOperator mainMenu = new JMenuBarOperator(mainWindow);
    JMenuOperator sqlMenu = new JMenuOperator(mainMenu.getMenu(4));

    JMenuItem appendItem = sqlMenu.getItem(18);
    JMenuItemOperator append = new JMenuItemOperator(appendItem);
    AppendResultsAction action = (AppendResultsAction)appendItem.getAction();
    assertFalse(appendItem.isSelected());
    
    runSql(sqlPanel, "select * from person");
    
    chooser.setName("resultspane");
    JComponentOperator comp = new JComponentOperator(mainWindow, chooser);
    JTabbedPane resultTab = (JTabbedPane)comp.getSource();
    
    assertEquals(2, resultTab.getTabCount());
    
    QueueTool tool = new QueueTool();
    mainMenu.pushMenu("SQL|Append new results", "|");    
    tool.waitEmpty();
    assertTrue(sqlPanel.getAppendResults());
    assertTrue(appendItem.isSelected());
    assertTrue(action.getButton().isSelected());
    
    runSql(sqlPanel, "select * from person");
    assertEquals(3, resultTab.getTabCount());
    
    mainMenu.pushMenu("SQL|Append new results", "|");    
    tool.waitEmpty();
    assertFalse(sqlPanel.getAppendResults());
    assertFalse(appendItem.isSelected());
    assertFalse(action.getButton().isSelected());
    
    runSql(sqlPanel, "select * from person");
    tool.waitEmpty();
    try { Thread.sleep(500); } catch (Throwable th) {}
    assertEquals(2, resultTab.getTabCount());
  }

  private String saveChanges(final SqlPanel panel)
  {
    Runnable r = new Runnable()
      {
        public void run()
        {
          panel.updateDb();
        }
      };
    testUtil.execute(r);
    testUtil.waitWhileBusy(panel);
    return panel.getLogMessage();
  }

  private String runSql(final SqlPanel panel, final String sql)
  {
    Runnable r = new Runnable()
      {
        public void run()
        {
          panel.getEditor().setText(sql);
          panel.runAll();
        }
      };
    testUtil.execute(r);
    testUtil.waitWhileBusy(panel);
    return panel.getLogMessage();
  }

  public void testWindow()
  {
    try
    {
      startApplication();
      connect();
      whatsNewTest();
      aboutTest();
      settingsTest();
      runSql();
      appendTest();
      pkWarningsTest();
      definePKTest();
      testUtil.stopApplication();
    }
    catch (Exception e)
    {
      e.printStackTrace();
      fail(e.getMessage());
    }
  }
}