def test_it_should_get_most_recent_version_for_a_existent_label_in_database(self):
self.execute_returns = {
"select count(*) from __db_version__;": 0,
"select version from __db_version__ where label = 'xxx' order by id desc": {"version": "vesion"},
}
mssql = MSSQL(self.config_mock, self.db_driver_mock)
ret = mssql.get_version_number_from_label("xxx")
self.assertEqual("vesion", ret)
expected_query_calls = [
call(
"if not exists ( select 1 from sysdatabases where name = 'migration_test' ) create database migration_test;"
),
call(
"if not exists ( select 1 from sysobjects where name = '__db_version__' and type = 'u' ) create table __db_version__ ( id INT IDENTITY(1,1) NOT NULL PRIMARY KEY, version varchar(20) NOT NULL default '0', label varchar(255), name varchar(255), sql_up NTEXT, sql_down NTEXT)"
),
call("insert into __db_version__ (version) values ('0')"),
]
self.assertEqual(expected_query_calls, self.db_mock.execute_non_query.mock_calls)
self.db_mock.select_db.assert_called_with("migration_test")
self.assertEqual(5, self.db_mock.close.call_count)
expected_execute_calls = [call("select count(*) from __db_version__;")]
self.assertEqual(expected_execute_calls, self.db_mock.execute_scalar.mock_calls)
expected_execute_calls = [call("select version from __db_version__ where label = 'xxx' order by id desc")]
self.assertEqual(expected_execute_calls, self.db_mock.execute_row.mock_calls)
def test_it_should_stop_process_when_an_error_occur_during_log_schema_version(self):
self.execute_returns['insert into __db_version__ (version, label, name, sql_up, sql_down) values (%s, %s, %s, %s, %s);'] = Exception("invalid sql")
try:
mssql = MSSQL(self.config_mock, self.db_driver_mock)
mssql.change("create table spam();", "20090212112104", "20090212112104_test_it_should_execute_migration_down_and_update_schema_version.migration", "create table spam();", "drop table spam;", label_version="label")
except Exception as e:
self.assertEqual('error logging migration: invalid sql\n\n[ERROR DETAILS] SQL command was:\n20090212112104_test_it_should_execute_migration_down_and_update_schema_version.migration', str(e))
self.assertTrue(isinstance(e, simple_db_migrate.core.exceptions.MigrationException))
expected_query_calls = [
call("if not exists ( select 1 from sysdatabases where name = 'migration_test' ) create database migration_test;"),
call("if not exists ( select 1 from sysobjects where name = '__db_version__' and type = 'u' ) create table __db_version__ ( id INT IDENTITY(1,1) NOT NULL PRIMARY KEY, version varchar(20) NOT NULL default '0', label varchar(255), name varchar(255), sql_up NTEXT, sql_down NTEXT)"),
call("insert into __db_version__ (version) values ('0')"),
call('create table spam()'),
call('insert into __db_version__ (version, label, name, sql_up, sql_down) values (%s, %s, %s, %s, %s);', ('20090212112104', 'label', '20090212112104_test_it_should_execute_migration_down_and_update_schema_version.migration', 'create table spam();', 'drop table spam;'))
]
self.assertEqual(expected_query_calls, self.db_mock.execute_non_query.mock_calls)
self.db_mock.select_db.assert_called_with('migration_test')
self.assertEqual(1, self.db_mock.cancel.call_count)
self.assertEqual(6, self.db_mock.close.call_count)
expected_execute_calls = [
call('select count(*) from __db_version__;'),
]
self.assertEqual(expected_execute_calls, self.db_mock.execute_scalar.mock_calls)
def test_it_should_get_current_schema_version(self):
self.execute_returns = {
'select count(*) from __db_version__;': 0,
'select top 1 version from __db_version__ order by id desc': "0"
}
mssql = MSSQL(self.config_mock, self.db_driver_mock)
self.assertEqual("0", mssql.get_current_schema_version())
expected_query_calls = [
call(
"if not exists ( select 1 from sysdatabases where name = 'migration_test' ) create database migration_test;"
),
call(
"if not exists ( select 1 from sysobjects where name = '__db_version__' and type = 'u' ) create table __db_version__ ( id INT IDENTITY(1,1) NOT NULL PRIMARY KEY, version varchar(20) NOT NULL default '0', label varchar(255), name varchar(255), sql_up NTEXT, sql_down NTEXT)"
),
call("insert into __db_version__ (version) values ('0')")
]
self.assertEqual(expected_query_calls,
self.db_mock.execute_non_query.mock_calls)
self.db_mock.select_db.assert_called_with('migration_test')
self.assertEqual(5, self.db_mock.close.call_count)
expected_execute_calls = [
call('select count(*) from __db_version__;'),
call("select top 1 version from __db_version__ order by id desc")
]
self.assertEqual(expected_execute_calls,
self.db_mock.execute_scalar.mock_calls)
def test_it_should_use_label_version_when_updating_schema_version(self):
mssql = MSSQL(self.config_mock, self.db_driver_mock)
mssql.change(
"create table spam();",
"20090212112104",
"20090212112104_test_it_should_execute_migration_down_and_update_schema_version.migration",
"create table spam();",
"drop table spam;",
label_version="label")
expected_query_calls = [
call(
"if not exists ( select 1 from sysdatabases where name = 'migration_test' ) create database migration_test;"
),
call(
"if not exists ( select 1 from sysobjects where name = '__db_version__' and type = 'u' ) create table __db_version__ ( id INT IDENTITY(1,1) NOT NULL PRIMARY KEY, version varchar(20) NOT NULL default '0', label varchar(255), name varchar(255), sql_up NTEXT, sql_down NTEXT)"
),
call("insert into __db_version__ (version) values ('0')"),
call('create table spam()'),
call(
'insert into __db_version__ (version, label, name, sql_up, sql_down) values (%s, %s, %s, %s, %s);',
('20090212112104', 'label',
'20090212112104_test_it_should_execute_migration_down_and_update_schema_version.migration',
'create table spam();', 'drop table spam;'))
]
self.assertEqual(expected_query_calls,
self.db_mock.execute_non_query.mock_calls)
self.db_mock.select_db.assert_called_with('migration_test')
self.assertEqual(6, self.db_mock.close.call_count)
expected_execute_calls = [call('select count(*) from __db_version__;')]
self.assertEqual(expected_execute_calls,
self.db_mock.execute_scalar.mock_calls)
def test_it_should_get_none_for_a_non_existent_label_in_database(self):
mssql = MSSQL(self.config_mock, self.db_driver_mock)
ret = mssql.get_version_number_from_label('xxx')
self.assertEqual(None, ret)
expected_query_calls = [
call(
"if not exists ( select 1 from sysdatabases where name = 'migration_test' ) create database migration_test;"
),
call(
"if not exists ( select 1 from sysobjects where name = '__db_version__' and type = 'u' ) create table __db_version__ ( id INT IDENTITY(1,1) NOT NULL PRIMARY KEY, version varchar(20) NOT NULL default '0', label varchar(255), name varchar(255), sql_up NTEXT, sql_down NTEXT)"
),
call("insert into __db_version__ (version) values ('0')")
]
self.assertEqual(expected_query_calls,
self.db_mock.execute_non_query.mock_calls)
self.db_mock.select_db.assert_called_with('migration_test')
self.assertEqual(5, self.db_mock.close.call_count)
expected_execute_calls = [
call('select count(*) from __db_version__;'),
]
self.assertEqual(expected_execute_calls,
self.db_mock.execute_scalar.mock_calls)
expected_execute_calls = [
call(
"select version from __db_version__ where label = 'xxx' order by id desc"
)
]
self.assertEqual(expected_execute_calls,
self.db_mock.execute_row.mock_calls)
def test_it_should_execute_migration_down_and_update_schema_version(self):
mssql = MSSQL(self.config_mock, self.db_driver_mock)
mssql.change(
"drop table spam;",
"20090212112104",
"20090212112104_test_it_should_execute_migration_down_and_update_schema_version.migration",
"create table spam();",
"drop table spam;",
False,
)
expected_query_calls = [
call(
"if not exists ( select 1 from sysdatabases where name = 'migration_test' ) create database migration_test;"
),
call(
"if not exists ( select 1 from sysobjects where name = '__db_version__' and type = 'u' ) create table __db_version__ ( id INT IDENTITY(1,1) NOT NULL PRIMARY KEY, version varchar(20) NOT NULL default '0', label varchar(255), name varchar(255), sql_up NTEXT, sql_down NTEXT)"
),
call("insert into __db_version__ (version) values ('0')"),
call("drop table spam"),
call("delete from __db_version__ where version = %s;", ("20090212112104",)),
]
self.assertEqual(expected_query_calls, self.db_mock.execute_non_query.mock_calls)
self.db_mock.select_db.assert_called_with("migration_test")
self.assertEqual(6, self.db_mock.close.call_count)
expected_execute_calls = [call("select count(*) from __db_version__;")]
self.assertEqual(expected_execute_calls, self.db_mock.execute_scalar.mock_calls)
def test_it_should_get_current_schema_version(self):
self.execute_returns = {
"select count(*) from __db_version__;": 0,
"select top 1 version from __db_version__ order by id desc": "0",
}
mssql = MSSQL(self.config_mock, self.db_driver_mock)
self.assertEquals("0", mssql.get_current_schema_version())
expected_query_calls = [
call(
"if not exists ( select 1 from sysdatabases where name = 'migration_test' ) create database migration_test;"
),
call(
"if not exists ( select 1 from sysobjects where name = '__db_version__' and type = 'u' ) create table __db_version__ ( id INT IDENTITY(1,1) NOT NULL PRIMARY KEY, version varchar(20) NOT NULL default '0', label varchar(255), name varchar(255), sql_up NTEXT, sql_down NTEXT)"
),
call("insert into __db_version__ (version) values ('0')"),
]
self.assertEqual(expected_query_calls, self.db_mock.execute_non_query.mock_calls)
self.db_mock.select_db.assert_called_with("migration_test")
self.assertEqual(5, self.db_mock.close.call_count)
expected_execute_calls = [
call("select count(*) from __db_version__;"),
call("select top 1 version from __db_version__ order by id desc"),
]
self.assertEqual(expected_execute_calls, self.db_mock.execute_scalar.mock_calls)
def test_it_should_get_all_schema_migrations(self):
expected_versions = []
expected_versions.append([1, "0", None, None, None, None])
expected_versions.append([2, "20090211120001", "label", "20090211120001_name", "sql_up", "sql_down"])
db_versions = [
{
"id": db_version[0],
"version": db_version[1],
"label": db_version[2],
"name": db_version[3],
"sql_up": db_version[4],
"sql_down": db_version[5],
}
for db_version in expected_versions
]
self.execute_returns = {
"select count(*) from __db_version__;": 0,
"select id, version, label, name, cast(sql_up as text) as sql_up, cast(sql_down as text) as sql_down from __db_version__ order by id;": db_versions,
}
mssql = MSSQL(self.config_mock, self.db_driver_mock)
schema_migrations = mssql.get_all_schema_migrations()
self.assertEquals(len(expected_versions), len(schema_migrations))
for index, migration in enumerate(schema_migrations):
self.assertEqual(migration.id, expected_versions[index][0])
self.assertEqual(migration.version, expected_versions[index][1])
self.assertEqual(migration.label, expected_versions[index][2])
self.assertEqual(migration.file_name, expected_versions[index][3])
self.assertEqual(migration.sql_up, expected_versions[index][4] and expected_versions[index][4] or "")
self.assertEqual(migration.sql_down, expected_versions[index][5] and expected_versions[index][5] or "")
expected_query_calls = [
call(
"if not exists ( select 1 from sysdatabases where name = 'migration_test' ) create database migration_test;"
),
call(
"if not exists ( select 1 from sysobjects where name = '__db_version__' and type = 'u' ) create table __db_version__ ( id INT IDENTITY(1,1) NOT NULL PRIMARY KEY, version varchar(20) NOT NULL default '0', label varchar(255), name varchar(255), sql_up NTEXT, sql_down NTEXT)"
),
call("insert into __db_version__ (version) values ('0')"),
]
self.assertEqual(expected_query_calls, self.db_mock.execute_non_query.mock_calls)
self.db_mock.select_db.assert_called_with("migration_test")
self.assertEqual(5, self.db_mock.close.call_count)
expected_execute_calls = [call("select count(*) from __db_version__;")]
self.assertEqual(expected_execute_calls, self.db_mock.execute_scalar.mock_calls)
expected_execute_calls = [
call(
"select id, version, label, name, cast(sql_up as text) as sql_up, cast(sql_down as text) as sql_down from __db_version__ order by id;"
)
]
self.assertEqual(expected_execute_calls, self.db_mock.execute_query.mock_calls)
def test_it_should_log_execution_when_a_function_is_given_when_updating_schema_version(self):
execution_log_mock = Mock()
mssql = MSSQL(self.config_mock, self.db_driver_mock)
mssql.change("create table spam();", "20090212112104", "20090212112104_test_it_should_execute_migration_down_and_update_schema_version.migration", "create table spam();", "drop table spam;", execution_log=execution_log_mock)
expected_execution_log_calls = [
call('create table spam()\n-- 1 row(s) affected\n'),
call('migration 20090212112104_test_it_should_execute_migration_down_and_update_schema_version.migration registered\n')
]
self.assertEqual(expected_execution_log_calls, execution_log_mock.mock_calls)
def test_it_should_not_execute_migration_if_fake_execution_is_set(self):
mssql = MSSQL(self.config_mock, self.db_driver_mock)
mssql.change("create table spam();", "20090212112104", "20090212112104_test_it_should_execute_migration_down_and_update_schema_version.migration", "create table spam();", "drop table spam;", fake_execution=True)
expected_execute_calls = [
call("if not exists ( select 1 from sysdatabases where name = 'migration_test' ) create database migration_test;"),
call("if not exists ( select 1 from sysobjects where name = '__db_version__' and type = 'u' ) create table __db_version__ ( id INT IDENTITY(1,1) NOT NULL PRIMARY KEY, version varchar(20) NOT NULL default '0', label varchar(255), name varchar(255), sql_up NTEXT, sql_down NTEXT)"),
call("insert into __db_version__ (version) values ('0')"),
call('insert into __db_version__ (version, label, name, sql_up, sql_down) values (%s, %s, %s, %s, %s);', ('20090212112104', None, '20090212112104_test_it_should_execute_migration_down_and_update_schema_version.migration', 'create table spam();', 'drop table spam;'))
]
self.assertEqual(expected_execute_calls, self.db_mock.execute_non_query.mock_calls)
self.assertEqual(5, self.db_mock.close.call_count)
def test_it_should_stop_process_when_an_error_occur_during_log_schema_version(
self):
self.execute_returns[
'insert into __db_version__ (version, label, name, sql_up, sql_down) values (%s, %s, %s, %s, %s);'] = Exception(
"invalid sql")
try:
mssql = MSSQL(self.config_mock, self.db_driver_mock)
mssql.change(
"create table spam();",
"20090212112104",
"20090212112104_test_it_should_execute_migration_down_and_update_schema_version.migration",
"create table spam();",
"drop table spam;",
label_version="label")
except Exception as e:
self.assertEqual(
'error logging migration: invalid sql\n\n[ERROR DETAILS] SQL command was:\n20090212112104_test_it_should_execute_migration_down_and_update_schema_version.migration',
str(e))
self.assertTrue(
isinstance(
e, simple_db_migrate.core.exceptions.MigrationException))
expected_query_calls = [
call(
"if not exists ( select 1 from sysdatabases where name = 'migration_test' ) create database migration_test;"
),
call(
"if not exists ( select 1 from sysobjects where name = '__db_version__' and type = 'u' ) create table __db_version__ ( id INT IDENTITY(1,1) NOT NULL PRIMARY KEY, version varchar(20) NOT NULL default '0', label varchar(255), name varchar(255), sql_up NTEXT, sql_down NTEXT)"
),
call("insert into __db_version__ (version) values ('0')"),
call('create table spam()'),
call(
'insert into __db_version__ (version, label, name, sql_up, sql_down) values (%s, %s, %s, %s, %s);',
('20090212112104', 'label',
'20090212112104_test_it_should_execute_migration_down_and_update_schema_version.migration',
'create table spam();', 'drop table spam;'))
]
self.assertEqual(expected_query_calls,
self.db_mock.execute_non_query.mock_calls)
self.db_mock.select_db.assert_called_with('migration_test')
self.assertEqual(1, self.db_mock.cancel.call_count)
self.assertEqual(6, self.db_mock.close.call_count)
expected_execute_calls = [
call('select count(*) from __db_version__;'),
]
self.assertEqual(expected_execute_calls,
self.db_mock.execute_scalar.mock_calls)
def test_it_should_parse_sql_statements(self):
statements = MSSQL._parse_sql_statements(
'; ; create table eggs; drop table spam; ; ;')
self.assertEqual(2, len(statements))
self.assertEqual('create table eggs', statements[0])
self.assertEqual('drop table spam', statements[1])
def test_it_should_drop_database_on_init_if_its_asked(self):
self.config_dict["drop_db_first"] = True
MSSQL(self.config_mock, self.db_driver_mock)
expected_query_calls = [
call(
"if exists ( select 1 from sysdatabases where name = 'migration_test' ) drop database migration_test;"
),
call(
"if not exists ( select 1 from sysdatabases where name = 'migration_test' ) create database migration_test;"
),
call(
"if not exists ( select 1 from sysobjects where name = '__db_version__' and type = 'u' ) create table __db_version__ ( id INT IDENTITY(1,1) NOT NULL PRIMARY KEY, version varchar(20) NOT NULL default '0', label varchar(255), name varchar(255), sql_up NTEXT, sql_down NTEXT)"
),
call("insert into __db_version__ (version) values ('0')")
]
self.assertEqual(expected_query_calls,
self.db_mock.execute_non_query.mock_calls)
self.db_mock.select_db.assert_called_with('migration_test')
self.assertEqual(5, self.db_mock.close.call_count)
expected_execute_calls = [call('select count(*) from __db_version__;')]
self.assertEqual(expected_execute_calls,
self.db_mock.execute_scalar.mock_calls)
def test_it_should_use_default_port(self):
MSSQL(self.config_mock)
self.assertEqual(
call(user='******',
password='******',
charset='utf8',
port=1433,
server='somehost'), sys.modules['_mssql'].connect.call_args)
def test_it_should_use_given_configuration(self):
self.config_dict['database_port'] = 9876
MSSQL(self.config_mock)
self.assertEqual(
call(user='******',
password='******',
charset='utf8',
port=9876,
server='somehost'), sys.modules['_mssql'].connect.call_args)
def test_it_should_get_all_schema_versions(self):
expected_versions = []
expected_versions.append("0")
expected_versions.append("20090211120001")
expected_versions.append("20090211120002")
expected_versions.append("20090211120003")
db_versions = [{'version': version} for version in expected_versions]
self.execute_returns = {
'select count(*) from __db_version__;': 0,
'select version from __db_version__ order by id;': db_versions
}
mssql = MSSQL(self.config_mock, self.db_driver_mock)
schema_versions = mssql.get_all_schema_versions()
self.assertEqual(len(expected_versions), len(schema_versions))
for version in schema_versions:
self.assertTrue(version in expected_versions)
expected_query_calls = [
call(
"if not exists ( select 1 from sysdatabases where name = 'migration_test' ) create database migration_test;"
),
call(
"if not exists ( select 1 from sysobjects where name = '__db_version__' and type = 'u' ) create table __db_version__ ( id INT IDENTITY(1,1) NOT NULL PRIMARY KEY, version varchar(20) NOT NULL default '0', label varchar(255), name varchar(255), sql_up NTEXT, sql_down NTEXT)"
),
call("insert into __db_version__ (version) values ('0')")
]
self.assertEqual(expected_query_calls,
self.db_mock.execute_non_query.mock_calls)
self.db_mock.select_db.assert_called_with('migration_test')
self.assertEqual(5, self.db_mock.close.call_count)
expected_execute_calls = [call('select count(*) from __db_version__;')]
self.assertEqual(expected_execute_calls,
self.db_mock.execute_scalar.mock_calls)
expected_execute_calls = [
call("select version from __db_version__ order by id;")
]
self.assertEqual(expected_execute_calls,
self.db_mock.execute_query.mock_calls)
def test_it_should_log_execution_when_a_function_is_given_when_updating_schema_version(
self):
execution_log_mock = Mock()
mssql = MSSQL(self.config_mock, self.db_driver_mock)
mssql.change(
"create table spam();",
"20090212112104",
"20090212112104_test_it_should_execute_migration_down_and_update_schema_version.migration",
"create table spam();",
"drop table spam;",
execution_log=execution_log_mock)
expected_execution_log_calls = [
call('create table spam()\n-- 1 row(s) affected\n'),
call(
'migration 20090212112104_test_it_should_execute_migration_down_and_update_schema_version.migration registered\n'
)
]
self.assertEqual(expected_execution_log_calls,
execution_log_mock.mock_calls)
def test_it_should_raise_whem_migration_sql_has_a_syntax_error(self):
mssql = MSSQL(self.config_mock, self.db_driver_mock)
self.assertRaisesWithMessage(
Exception,
"error executing migration: invalid sql syntax 'create table foo(); create table spam());'",
mssql.change,
"create table foo(); create table spam());",
"20090212112104",
"20090212112104_test_it_should_execute_migration_down_and_update_schema_version.migration",
"create table spam());",
"drop table spam;",
label_version="label")
def test_it_should_get_all_schema_versions(self):
expected_versions = []
expected_versions.append("0")
expected_versions.append("20090211120001")
expected_versions.append("20090211120002")
expected_versions.append("20090211120003")
db_versions = [{"version": version} for version in expected_versions]
self.execute_returns = {
"select count(*) from __db_version__;": 0,
"select version from __db_version__ order by id;": db_versions,
}
mssql = MSSQL(self.config_mock, self.db_driver_mock)
schema_versions = mssql.get_all_schema_versions()
self.assertEquals(len(expected_versions), len(schema_versions))
for version in schema_versions:
self.assertTrue(version in expected_versions)
expected_query_calls = [
call(
"if not exists ( select 1 from sysdatabases where name = 'migration_test' ) create database migration_test;"
),
call(
"if not exists ( select 1 from sysobjects where name = '__db_version__' and type = 'u' ) create table __db_version__ ( id INT IDENTITY(1,1) NOT NULL PRIMARY KEY, version varchar(20) NOT NULL default '0', label varchar(255), name varchar(255), sql_up NTEXT, sql_down NTEXT)"
),
call("insert into __db_version__ (version) values ('0')"),
]
self.assertEqual(expected_query_calls, self.db_mock.execute_non_query.mock_calls)
self.db_mock.select_db.assert_called_with("migration_test")
self.assertEqual(5, self.db_mock.close.call_count)
expected_execute_calls = [call("select count(*) from __db_version__;")]
self.assertEqual(expected_execute_calls, self.db_mock.execute_scalar.mock_calls)
expected_execute_calls = [call("select version from __db_version__ order by id;")]
self.assertEqual(expected_execute_calls, self.db_mock.execute_query.mock_calls)
def test_it_should_raise_whem_migration_sql_has_a_syntax_error_sql_with_codec_error(
self):
mssql = MSSQL(self.config_mock, self.db_driver_mock)
expected_raised_message = u"error executing migration: invalid sql syntax 'create table foo(); create table spam()); -- ônibus'".encode(
"utf-8")
self.assertRaisesWithMessage(
Exception,
expected_raised_message,
mssql.change,
u"create table foo(); create table spam()); -- ônibus",
"20090212112104",
"20090212112104_test_it_should_execute_migration_down_and_update_schema_version.migration",
"create table foo(); create table spam());",
"drop table spam;",
label_version="label")
def test_it_should_stop_process_when_an_error_occur_during_connect_database(
self):
self.db_driver_mock.connect.side_effect = Exception(
"error when connecting")
try:
MSSQL(self.config_mock, self.db_driver_mock)
self.fail("it should not get here")
except Exception as e:
self.assertEqual(
"could not connect to database: error when connecting", str(e))
self.assertEqual(0, self.db_mock.close.call_count)
self.assertEqual(0, self.db_mock.execute_scalar.call_count)
self.assertEqual(0, self.db_mock.execute_non_query.call_count)
self.assertEqual(0, self.db_mock.execute_query.call_count)
def test_it_should_stop_process_when_an_error_occur_during_drop_database(
self):
self.config_dict["drop_db_first"] = True
self.db_mock.execute_non_query.side_effect = Exception(
"error when dropping")
try:
MSSQL(self.config_mock, self.db_driver_mock)
self.fail("it should not get here")
except Exception as e:
self.assertEqual(
"can't drop database 'migration_test'; \nerror when dropping",
str(e))
expected_query_calls = [
call(
"if exists ( select 1 from sysdatabases where name = 'migration_test' ) drop database migration_test;"
)
]
self.assertEqual(expected_query_calls,
self.db_mock.execute_non_query.mock_calls)
self.assertEqual(1, self.db_mock.close.call_count)
def test_it_should_parse_sql_statements_with_html_inside(self):
sql = u"""
create table eggs;
INSERT INTO widget_parameter_domain (widget_parameter_id, label, value)
VALUES ((SELECT MAX(widget_parameter_id)
FROM widget_parameter), "Carros", '<div class="box-zap-geral">
<div class="box-zap box-zap-autos">
<a class="logo" target="_blank" title="ZAP" href="http://www.zap.com.br/Parceiros/g1/RedirG1.aspx?CodParceriaLink=42&URL=http://www.zap.com.br">');
drop table spam;
"""
statements = MSSQL._parse_sql_statements(sql)
expected_sql_with_html = """INSERT INTO widget_parameter_domain (widget_parameter_id, label, value)
VALUES ((SELECT MAX(widget_parameter_id)
FROM widget_parameter), "Carros", '<div class="box-zap-geral">
<div class="box-zap box-zap-autos">
<a class="logo" target="_blank" title="ZAP" href="http://www.zap.com.br/Parceiros/g1/RedirG1.aspx?CodParceriaLink=42&URL=http://www.zap.com.br">')"""
self.assertEqual(3, len(statements))
self.assertEqual('create table eggs', statements[0])
self.assertEqual(expected_sql_with_html, statements[1])
self.assertEqual('drop table spam', statements[2])
def test_it_should_parse_sql_statements_with_html_inside(self):
sql = u"""
create table eggs;
INSERT INTO widget_parameter_domain (widget_parameter_id, label, value)
VALUES ((SELECT MAX(widget_parameter_id)
FROM widget_parameter), "Carros", '<div class="box-zap-geral">
<div class="box-zap box-zap-autos">
<a class="logo" target="_blank" title="ZAP" href="http://www.zap.com.br/Parceiros/g1/RedirG1.aspx?CodParceriaLink=42&URL=http://www.zap.com.br">');
drop table spam;
"""
statements = MSSQL._parse_sql_statements(sql)
expected_sql_with_html = """INSERT INTO widget_parameter_domain (widget_parameter_id, label, value)
VALUES ((SELECT MAX(widget_parameter_id)
FROM widget_parameter), "Carros", '<div class="box-zap-geral">
<div class="box-zap box-zap-autos">
<a class="logo" target="_blank" title="ZAP" href="http://www.zap.com.br/Parceiros/g1/RedirG1.aspx?CodParceriaLink=42&URL=http://www.zap.com.br">')"""
self.assertEqual(3, len(statements))
self.assertEqual("create table eggs", statements[0])
self.assertEqual(expected_sql_with_html, statements[1])
self.assertEqual("drop table spam", statements[2])
def test_it_should_parse_sql_statements(self):
statements = MSSQL._parse_sql_statements("; ; create table eggs; drop table spam; ; ;")
self.assertEqual(2, len(statements))
self.assertEqual("create table eggs", statements[0])
self.assertEqual("drop table spam", statements[1])
def test_it_should_use_mssql_as_driver(self):
MSSQL(self.config_mock)
self.assertNotEqual(0, sys.modules['_mssql'].connect.call_count)
def test_it_should_get_all_schema_migrations(self):
expected_versions = []
expected_versions.append([1, "0", None, None, None, None])
expected_versions.append([
2, "20090211120001", "label", "20090211120001_name", "sql_up",
"sql_down"
])
db_versions = [{
'id': db_version[0],
'version': db_version[1],
'label': db_version[2],
'name': db_version[3],
'sql_up': db_version[4],
'sql_down': db_version[5]
} for db_version in expected_versions]
self.execute_returns = {
'select count(*) from __db_version__;':
0,
'select id, version, label, name, cast(sql_up as text) as sql_up, cast(sql_down as text) as sql_down from __db_version__ order by id;':
db_versions
}
mssql = MSSQL(self.config_mock, self.db_driver_mock)
schema_migrations = mssql.get_all_schema_migrations()
self.assertEqual(len(expected_versions), len(schema_migrations))
for index, migration in enumerate(schema_migrations):
self.assertEqual(migration.id, expected_versions[index][0])
self.assertEqual(migration.version, expected_versions[index][1])
self.assertEqual(migration.label, expected_versions[index][2])
self.assertEqual(migration.file_name, expected_versions[index][3])
self.assertEqual(
migration.sql_up,
expected_versions[index][4] and expected_versions[index][4]
or "")
self.assertEqual(
migration.sql_down,
expected_versions[index][5] and expected_versions[index][5]
or "")
expected_query_calls = [
call(
"if not exists ( select 1 from sysdatabases where name = 'migration_test' ) create database migration_test;"
),
call(
"if not exists ( select 1 from sysobjects where name = '__db_version__' and type = 'u' ) create table __db_version__ ( id INT IDENTITY(1,1) NOT NULL PRIMARY KEY, version varchar(20) NOT NULL default '0', label varchar(255), name varchar(255), sql_up NTEXT, sql_down NTEXT)"
),
call("insert into __db_version__ (version) values ('0')")
]
self.assertEqual(expected_query_calls,
self.db_mock.execute_non_query.mock_calls)
self.db_mock.select_db.assert_called_with('migration_test')
self.assertEqual(5, self.db_mock.close.call_count)
expected_execute_calls = [
call('select count(*) from __db_version__;'),
]
self.assertEqual(expected_execute_calls,
self.db_mock.execute_scalar.mock_calls)
expected_execute_calls = [
call(
'select id, version, label, name, cast(sql_up as text) as sql_up, cast(sql_down as text) as sql_down from __db_version__ order by id;'
)
]
self.assertEqual(expected_execute_calls,
self.db_mock.execute_query.mock_calls)
def test_it_should_parse_sql_statements(self):
statements = MSSQL._parse_sql_statements('; ; create table eggs; drop table spam; ; ;')
self.assertEqual(2, len(statements))
self.assertEqual('create table eggs', statements[0])
self.assertEqual('drop table spam', statements[1])
