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))
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_executeResultsQuery(self): self.helper.setup_clean_db() my_sql_mixin = MySQLMixin() my_sql_mixin.executeUpdateQuery(""" insert into projects (projectname, description, currency) values ('test', 'a simple test', 'GBP')""") expected = [(2, u'test', None, None, u'a simple test', u'GBP')] query = 'select * from projects' self.assertEqual(expected, my_sql_mixin.executeResultsQuery(query))
def test_executeUpdateQueryWithParams(self): self.helper.setup_clean_db() my_sql_mixin = MySQLMixin() p = ['test3', '2012-06-04', 'description', 'LIR'] my_sql_mixin.executeUpdateQuery(""" insert into projects (projectname, startdate, enddate, description, currency) values (%s, %s, '2013-07-03', %s, %s)""", p) expected = [(2, u'test3', datetime.date(2012, 6, 4), datetime.date(2013, 7, 3), u'description', u'LIR')] query = 'select * from projects' self.assertEqual(expected, my_sql_mixin.executeResultsQuery(query))