// // $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