@Test
public void testCreateDataSet() throws Exception
{
IDataSet mockImportedKeysDataSet = mockHelper.createMock("mockImportedKeysDataSet", IDataSet.class);
IDataSet mockExportedKeysDataSet = mockHelper.createMock("mockExportedKeysDataSet", IDataSet.class);
expect(mockSQLDatabaseMetaData.getImportedKeysDataSet(mockTableInfo)).andStubReturn(mockImportedKeysDataSet);
expect(mockSQLDatabaseMetaData.getExportedKeysDataSet(mockTableInfo)).andStubReturn(mockExportedKeysDataSet);
mockHelper.replayAll();
classUnderTest.setTableInfo(mockTableInfo);
classUnderTest.setSession(mockSession);
classUnderTest.setDatabaseObjectInfo(mockTableInfo);
IDataSet result = classUnderTest.createDataSet();
assertNotNull(result);
mockHelper.verifyAll();
}