[Squirrel-sql-commits] SF.net SVN: squirrel-sql:[6216] trunk/sql12/plugins/dbcopy/src

[<man...@us...>]

Revision: 6216
          http://squirrel-sql.svn.sourceforge.net/squirrel-sql/?rev=6216&view=rev
Author:   manningr
Date:     2011-03-26 21:38:05 +0000 (Sat, 26 Mar 2011)

Log Message:
-----------
Added tests for coverage.

Modified Paths:
--------------
    trunk/sql12/plugins/dbcopy/src/main/java/net/sourceforge/squirrel_sql/plugins/dbcopy/DBCopyPlugin.java
    trunk/sql12/plugins/dbcopy/src/test/java/net/sourceforge/squirrel_sql/plugins/dbcopy/DBCopyPluginTest.java

Modified: trunk/sql12/plugins/dbcopy/src/main/java/net/sourceforge/squirrel_sql/plugins/dbcopy/DBCopyPlugin.java
===================================================================
--- trunk/sql12/plugins/dbcopy/src/main/java/net/sourceforge/squirrel_sql/plugins/dbcopy/DBCopyPlugin.java	2011-03-26 21:37:42 UTC (rev 6215)
+++ trunk/sql12/plugins/dbcopy/src/main/java/net/sourceforge/squirrel_sql/plugins/dbcopy/DBCopyPlugin.java	2011-03-26 21:38:05 UTC (rev 6216)
@@ -195,11 +195,16 @@
 	/**
 	 * @param enabled
 	 */
-	public void setCopyMenuEnabled(boolean enabled)
+	public void setCopyMenuEnabled(final boolean enabled)
 	{
-		final ActionCollection coll = getApplication().getActionCollection();
-		CopyTableAction copyAction = (CopyTableAction) coll.get(CopyTableAction.class);
-		copyAction.setEnabled(enabled);
+		GUIUtils.processOnSwingEventThread(new Runnable()
+		{
+			public void run()
+			{
+				final ActionCollection coll = getApplication().getActionCollection();
+				coll.get(CopyTableAction.class).setEnabled(enabled);
+			}
+		});
 	}
 
 	/**
@@ -212,8 +217,7 @@
 			public void run()
 			{
 				final ActionCollection coll = getApplication().getActionCollection();
-				PasteTableAction pasteAction = (PasteTableAction) coll.get(PasteTableAction.class);
-				pasteAction.setEnabled(enabled);
+				coll.get(PasteTableAction.class).setEnabled(true);
 			}
 		});
 	}

Modified: trunk/sql12/plugins/dbcopy/src/test/java/net/sourceforge/squirrel_sql/plugins/dbcopy/DBCopyPluginTest.java
===================================================================
--- trunk/sql12/plugins/dbcopy/src/test/java/net/sourceforge/squirrel_sql/plugins/dbcopy/DBCopyPluginTest.java	2011-03-26 21:37:42 UTC (rev 6215)
+++ trunk/sql12/plugins/dbcopy/src/test/java/net/sourceforge/squirrel_sql/plugins/dbcopy/DBCopyPluginTest.java	2011-03-26 21:38:05 UTC (rev 6216)
@@ -19,19 +19,66 @@
 package net.sourceforge.squirrel_sql.plugins.dbcopy;
 
 
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertNotNull;
+import static org.junit.Assert.assertNull;
 import net.sourceforge.squirrel_sql.client.plugin.AbstractSessionPluginTest;
 import net.sourceforge.squirrel_sql.client.plugin.IPlugin;
+import net.sourceforge.squirrel_sql.client.plugin.PluginSessionCallback;
+import net.sourceforge.squirrel_sql.fw.sql.IDatabaseObjectInfo;
 
+import org.junit.Test;
 import org.junit.runner.RunWith;
+import org.mockito.Mock;
 import org.mockito.runners.MockitoJUnitRunner;
 
 @RunWith(MockitoJUnitRunner.class)
 public class DBCopyPluginTest extends AbstractSessionPluginTest
 {
 	
+	@Mock
+	IDatabaseObjectInfo mockDatabaseObjectInfo;
+	
 	@Override
 	protected IPlugin getPluginToTest() throws Exception
 	{
 		return new DBCopyPlugin();
 	}
+	
+	@Test
+	public void testSessionStarted() {
+		PluginSessionCallback result = ((DBCopyPlugin) super.classUnderTest).sessionStarted(mockSession);
+		assertNotNull(result);		
+	}
+	
+	@Test
+	public void testSessionEnding() {
+		DBCopyPlugin plugin = (DBCopyPlugin) super.classUnderTest;
+		plugin.setCopySourceSession(mockSession);
+		plugin.sessionEnding(mockSession);
+		assertNull(plugin.getCopySourceSession());
+	}
+	
+	@Test
+	public void testSetSelectedDatabaseObjects() {
+		DBCopyPlugin plugin = (DBCopyPlugin) super.classUnderTest;
+		IDatabaseObjectInfo[] dbObjArr = new IDatabaseObjectInfo[] { mockDatabaseObjectInfo };
+		plugin.setSelectedDatabaseObjects(dbObjArr);
+		IDatabaseObjectInfo[] result = plugin.getSourceSelectedDatabaseObjects();
+		assertEquals(1, result.length);
+	}
+	
+	@Test
+	public void testSetDestSelectedDatabaseObject() {
+		DBCopyPlugin plugin = (DBCopyPlugin) super.classUnderTest;
+		plugin.setDestSelectedDatabaseObject(mockDatabaseObjectInfo);
+		assertEquals(mockDatabaseObjectInfo, plugin.getDestSelectedDatabaseObject());
+	}
+	
+	@Test
+	public void testSetDestCopySession() {
+		DBCopyPlugin plugin = (DBCopyPlugin) super.classUnderTest;
+		plugin.setDestCopySession(mockSession);
+		assertEquals(mockSession, plugin.getCopyDestSession());
+	}
 }


This was sent by the SourceForge.net collaborative development platform, the world's largest Open Source development site.