def test_it_should_get_current_schema_version(self): self.fetchone_returns = { 'select count(*) from __db_version__;': [0], 'select version from __db_version__ order by id desc limit 0,1;': ["0"] } mysql = MySQL(self.config_mock, self.db_driver_mock) self.assertEqual("0", mysql.get_current_schema_version()) expected_query_calls = [ call('create database if not exists `migration_test`;') ] self.assertEqual(expected_query_calls, self.db_mock.query.mock_calls) self.db_mock.select_db.assert_called_with('migration_test') self.assertEqual(2, self.db_mock.commit.call_count) self.assertEqual(5, self.db_mock.close.call_count) expected_execute_calls = [ call( 'create table if not exists __db_version__ ( id int(11) NOT NULL AUTO_INCREMENT, version varchar(20) NOT NULL default "0", label varchar(255), name varchar(255), sql_up LONGTEXT, sql_down LONGTEXT, PRIMARY KEY (id))' ), call('select count(*) from __db_version__;'), call('insert into __db_version__ (version) values ("0")'), call( 'select version from __db_version__ order by id desc limit 0,1;' ) ] self.assertEqual(expected_execute_calls, self.cursor_mock.execute.mock_calls) self.assertEqual(4, self.cursor_mock.close.call_count)
def test_it_should_get_current_schema_version(self): self.fetchone_returns = {'select count(*) from __db_version__;': [0], 'select version from __db_version__ order by id desc limit 0,1;': ["0"]} mysql = MySQL(self.config_mock, self.db_driver_mock) self.assertEqual("0", mysql.get_current_schema_version()) expected_query_calls = [ call('create database if not exists `migration_test`;') ] self.assertEqual(expected_query_calls, self.db_mock.query.mock_calls) self.db_mock.select_db.assert_called_with('migration_test') self.assertEqual(2, self.db_mock.commit.call_count) self.assertEqual(5, self.db_mock.close.call_count) expected_execute_calls = [ call('create table if not exists __db_version__ ( id int(11) NOT NULL AUTO_INCREMENT, version varchar(20) NOT NULL default "0", label varchar(255), name varchar(255), sql_up LONGTEXT, sql_down LONGTEXT, PRIMARY KEY (id))'), call('select count(*) from __db_version__;'), call('insert into __db_version__ (version) values ("0")'), call('select version from __db_version__ order by id desc limit 0,1;') ] self.assertEqual(expected_execute_calls, self.cursor_mock.execute.mock_calls) self.assertEqual(4, self.cursor_mock.close.call_count)