// // $Id$ // // SPTableCopyTest.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 "SPTableCopy.h" #import "SPTableCopyTest.h" @implementation SPTableCopyTest - (id) getMockConnection { id mockConnection = [OCMockObject niceMockForClass:[MCPConnection class]]; return [mockConnection autorelease]; } - (id) getMockResult { id mockResult = [OCMockObject niceMockForClass:[MCPResult class]]; return mockResult; } - (SPTableCopy *) getTableCopyFixture { SPTableCopy *tableCopy = [[SPTableCopy alloc] init]; return [tableCopy autorelease]; } - (void)testCopyTableFromToWithData { id mockResult = [self getMockResult]; unsigned long long varOne = 1; NSValue *valueOne = [NSValue value:&varOne withObjCType:@encode(__typeof__(varOne))]; BOOL varNo = NO; NSValue *valueNo = [NSValue value:&varNo withObjCType:@encode(BOOL)]; NSArray *resultArray = [[NSArray alloc] initWithObjects:@"", @"CREATE TABLE `table_name` ()", nil]; [[[mockResult expect] andReturnValue:valueOne] numOfRows]; [[[mockResult expect] andReturn:resultArray] fetchRowAsArray]; id mockConnection = [self getMockConnection]; [[[mockConnection expect] andReturn:mockResult] queryString:@"SHOW CREATE TABLE `source_db`.`table_name`"]; [[mockConnection expect] queryString:@"CREATE TABLE `target_db`.`table_name` ()"]; [[mockConnection expect] queryString:@"INSERT INTO `target_db`.`table_name` SELECT * FROM `source_db`.`table_name`"]; [[[mockConnection stub] andReturnValue:valueNo] queryErrored]; id tableCopy = [self getTableCopyFixture]; [tableCopy setConnection:mockConnection]; [tableCopy copyTable: @"table_name" from: @"source_db" to: @"target_db" withContent: YES]; [mockResult verify]; [mockConnection verify]; [resultArray release]; } @end