def test_executeMultipleUpdateQueries(self):
self.helper.setup_clean_db()
my_sql_mixin = MySQLMixin()
queries = [
"""
insert into projects
(projectname, description, currency)
values
('test', 'a simple test', 'GBP')""",
"""
insert into projects
(projectname, description, currency)
values
('test2', 'simple test', 'AUS')""",
]
my_sql_mixin.executeMultipleUpdateQueries(queries)
expected = [
(u'test', None, None, u'a simple test', u'GBP'),
(u'test2', None, None, u'simple test', u'AUS')
]
query = """
select projectname, startdate, enddate, description, currency
from projects
"""
self.assertEqual(expected, my_sql_mixin.executeResultsQuery(query))
def test_executeMultipleResultsQueries(self):
self.helper.setup_clean_db()
mixin = MySQLMixin()
queries = [
"""
insert into projects
(projectname, description, currency)
values
('test', 'a simple test', 'GBP')""",
"""
insert into projects
(projectname, description, currency)
values
('test2', 'simple test', 'AUS')""",
]
mixin.executeMultipleUpdateQueries(queries)
exp = [
[(u'test', None, None, u'a simple test', u'GBP')],
[(u'test2', None, None, u'simple test', u'AUS')]
]
s_queries = [
"""
select projectname, startdate, enddate, description, currency
from projects limit 1
""",
"""
select projectname, startdate, enddate, description, currency
from projects limit 1 offset 1
""",
]
self.assertEqual(exp, mixin.executeMultipleResultsQueries(s_queries))
