[Squirrel-sql-commits] SF.net SVN: squirrel-sql:[6044] trunk/sql12/app/src/test/java/net/ sourceforge/squirrel_sql/client/session/mainpanel/objecttree/ AbstractINodeExpanderTest.java

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

Revision: 6044
          http://squirrel-sql.svn.sourceforge.net/squirrel-sql/?rev=6044&view=rev
Author:   manningr
Date:     2010-12-22 13:33:06 +0000 (Wed, 22 Dec 2010)

Log Message:
-----------
Add new mock object setup in support of object tree icons feature testing.

Modified Paths:
--------------
    trunk/sql12/app/src/test/java/net/sourceforge/squirrel_sql/client/session/mainpanel/objecttree/AbstractINodeExpanderTest.java

Modified: trunk/sql12/app/src/test/java/net/sourceforge/squirrel_sql/client/session/mainpanel/objecttree/AbstractINodeExpanderTest.java
===================================================================
--- trunk/sql12/app/src/test/java/net/sourceforge/squirrel_sql/client/session/mainpanel/objecttree/AbstractINodeExpanderTest.java	2010-12-22 13:32:12 UTC (rev 6043)
+++ trunk/sql12/app/src/test/java/net/sourceforge/squirrel_sql/client/session/mainpanel/objecttree/AbstractINodeExpanderTest.java	2010-12-22 13:33:06 UTC (rev 6044)
@@ -33,9 +33,11 @@
 import net.sourceforge.squirrel_sql.fw.id.IIdentifier;
 import net.sourceforge.squirrel_sql.fw.sql.DatabaseObjectType;
 import net.sourceforge.squirrel_sql.fw.sql.IDatabaseObjectInfo;
+import net.sourceforge.squirrel_sql.fw.sql.IObjectTypes;
 import net.sourceforge.squirrel_sql.fw.sql.ISQLConnection;
 import net.sourceforge.squirrel_sql.fw.sql.SQLDatabaseMetaData;
 
+
 import org.easymock.classextension.EasyMock;
 import org.junit.Test;
 
@@ -55,11 +57,12 @@
 	protected SessionProperties mockSessionProperties = mockHelper.createMock("mockSessionProperties", SessionProperties.class);
 	protected Class<?> clazz = null;
 	protected SchemaInfo mockSchemaInfo = mockHelper.createMock("mockSchemaInfo", SchemaInfo.class);
+	protected IObjectTypes mockObjectTypes = mockHelper.createMock("mockObjectTypes", IObjectTypes.class);
 
 	@Test
 	public void testCreateChildren() throws SQLException
 	{
-		setupMockDatabaseObjectInfo();
+		setupMockObjects();
 		expect(mockSession.getSQLConnection()).andStubReturn(mockSQLConnection);
 		expect(mockSession.getApplication()).andStubReturn(mockApplication);
 		expect(mockSession.getIdentifier()).andStubReturn(mockIdentifier);
@@ -99,13 +102,30 @@
 		mockHelper.resetAll();
 	}
 
-	protected void setupMockDatabaseObjectInfo()
+	protected void setupMockObjects()
 	{
 		expect(mockDatabaseObjectInfo.getSchemaName()).andStubReturn(TEST_SCHEMA_NAME);
 		expect(mockDatabaseObjectInfo.getCatalogName()).andStubReturn(TEST_CATALOG_NAME);
 		expect(mockDatabaseObjectInfo.getSimpleName()).andStubReturn(TEST_SIMPLE_NAME);
 		expect(mockDatabaseObjectInfo.getQualifiedName()).andStubReturn(TEST_QUALIFIED_NAME);
 		expect(mockDatabaseObjectInfo.getDatabaseObjectType()).andStubReturn(DatabaseObjectType.TABLE);
+		
+		
+		expect(mockObjectTypes.getConstraintParent()).andStubReturn(DatabaseObjectType.createNewDatabaseObjectType("TestConstraintParent"));
+		expect(mockObjectTypes.getConsumerGroup()).andStubReturn(DatabaseObjectType.createNewDatabaseObjectType("TestConsumerGroup"));
+		expect(mockObjectTypes.getConsumerGroupParent()).andStubReturn(DatabaseObjectType.createNewDatabaseObjectType("TestConsumerGroupParent"));
+		expect(mockObjectTypes.getFunctionParent()).andStubReturn(DatabaseObjectType.createNewDatabaseObjectType("TestFunctionParent"));
+		expect(mockObjectTypes.getIndexParent()).andStubReturn(DatabaseObjectType.createNewDatabaseObjectType("TestIndexParent"));
+		expect(mockObjectTypes.getInstance()).andStubReturn(DatabaseObjectType.createNewDatabaseObjectType("TestInstance"));
+		expect(mockObjectTypes.getLob()).andStubReturn(DatabaseObjectType.createNewDatabaseObjectType("TestLob"));
+		expect(mockObjectTypes.getLobParent()).andStubReturn(DatabaseObjectType.createNewDatabaseObjectType("TestLobParent"));
+		expect(mockObjectTypes.getPackage()).andStubReturn(DatabaseObjectType.createNewDatabaseObjectType("TestPackage"));	
+		expect(mockObjectTypes.getPackageParent()).andStubReturn(DatabaseObjectType.createNewDatabaseObjectType("TestPackageParent"));
+		expect(mockObjectTypes.getSession()).andStubReturn(DatabaseObjectType.createNewDatabaseObjectType("TestSession"));
+		expect(mockObjectTypes.getSequenceParent()).andStubReturn(DatabaseObjectType.createNewDatabaseObjectType("TestSequenceParent"));
+		expect(mockObjectTypes.getTriggerParent()).andStubReturn(DatabaseObjectType.createNewDatabaseObjectType("TestTriggerParent"));
+		expect(mockObjectTypes.getType()).andStubReturn(DatabaseObjectType.createNewDatabaseObjectType("TestType"));
+		expect(mockObjectTypes.getTypeParent()).andStubReturn(DatabaseObjectType.createNewDatabaseObjectType("TestTypeParent"));
 	}
 
 }
\ No newline at end of file


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