1 files changed, 100 insertions, 0 deletions
diff --git a/UnitTests/SPDatabaseCopyTest.m b/UnitTests/SPDatabaseCopyTest.m
new file mode 100644
index 00000000..bb90d35d
--- /dev/null
+++ b/UnitTests/SPDatabaseCopyTest.m
@@ -0,0 +1,100 @@
+//
+//  $Id: $
+//
+//  DatabaseCopyTest.m
+//  sequel-pro
+//
+//  Created by David Rekowski
+//  Copyright (c) 2010 David Rekowski. All rights reserved.
+//
+//  This program is free software; you can redistribute it and/or modify
+//  it under the terms of the GNU General Public License as published by
+//  the Free Software Foundation; either version 2 of the License, or
+//  (at your option) any later version.
+//
+//  This program is distributed in the hope that it will be useful,
+//  but WITHOUT ANY WARRANTY; without even the implied warranty of
+//  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+//  GNU General Public License for more details.
+//
+//  You should have received a copy of the GNU General Public License
+//  along with this program; if not, write to the Free Software
+//  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+//
+//  More info at <http://code.google.com/p/sequel-pro/>
+
+#import <OCMock/OCMock.h>
+#import "SPAlertSheets.h"
+#import "SPDatabaseCopyTest.h"
+#import "SPDatabaseCopy.h"
+#import "SPTableCopy.h"
+#import "MCPConnection.h"
+#import "SPDatabaseInfo.h"
+#import "SPStringAdditions.h"
+#import "SPLogger.h"
+
+@implementation SPDatabaseCopyTest
+
+- (SPDatabaseCopy *) getDatabaseCopyFixture {
+    SPDatabaseCopy *dbCopy = [[SPDatabaseCopy alloc] init];
+	return [dbCopy autorelease];
+}
+
+- (SPTableCopy *) getTableCopyFixture {
+    SPTableCopy *tableCopy = [[SPTableCopy alloc] init];
+	return [tableCopy autorelease];
+}
+
+- (id) getMockConnection {
+	id mockConnection = [OCMockObject niceMockForClass:[MCPConnection class]];
+	return mockConnection;
+}
+
+- (id) getMockDBInfo {
+	id mockDBInfo = [OCMockObject niceMockForClass:[SPDatabaseInfo class]];
+	return mockDBInfo;
+}
+
+- (void) testCopyDatabase {
+
+	SPDatabaseCopy *dbCopy = [self getDatabaseCopyFixture];
+	id mockConnection = [self getMockConnection];
+	[[mockConnection expect] queryString:@"CREATE DATABASE `target_name`"];
+	[[mockConnection expect] listTablesFromDB:@"source_name"];
+	[[[mockConnection stub] andReturn:[[NSArray alloc] init]] listTablesFromDB:@"source_name"];
+	[dbCopy setConnection:mockConnection];
+	
+	id mockDBInfo = [self getMockDBInfo];
+
+	BOOL varNo = NO;
+	BOOL varYes = YES;
+	[[[mockDBInfo expect] andReturnValue:[NSValue value:&varYes withObjCType:@encode(BOOL)]] databaseExists:@"source_name"];
+
+	[[[mockDBInfo expect] andReturnValue:[NSValue value:&varNo withObjCType:@encode(BOOL)]] databaseExists:@"target_name"];
+
+	[dbCopy setDbInfo:mockDBInfo];
+	
+	NSString *source = [[NSString alloc] initWithString:@"source_name"];
+	NSString *target = [[NSString alloc] initWithString:@"target_name"];
+	[dbCopy copyDatabaseFrom:source to:target withContent:YES];
+	
+	[mockConnection verify];
+	[source release];
+	[target release];
+}
+
+- (void) testCopyDatabaseTables {
+	SPDatabaseCopy *dbCopy = [self getDatabaseCopyFixture];
+	SPTableCopy *tableCopy = [self getTableCopyFixture];
+	STAssertTrue([tableCopy copyTable:@"table_one" from: @"source_db" to: @"target_db"], 
+				 @"Should have copied database table table_one.");
+	STAssertTrue([tableCopy copyTable:@"table_two" from: @"source_db" to: @"target_db"], 
+				 @"Should have copied database table table_two.");
+}
+	 
+- (void) testCreateDatabase {
+	SPDatabaseCopy *dbCopy = [self getDatabaseCopyFixture];
+	// test missing :)
+}
+
+@end