def test_query_setters(self):
query = Query()
query.set_select_expression(['country', 'first_name']),
query.set_aggregate_function('COUNT(*)'),
query.set_table('table_name'),
query.add_where_clause('', 'country', 'USA'),
query.set_result([3]),
self.assertEqual(query.get_select_expression(), ['country', 'first_name'])
self.assertEqual(query.get_aggregate_function(), 'COUNT(*)')
self.assertEqual(query.get_table(), 'table_name')
self.assertEqual(query.get_where_clause(), [('', 'country', 'USA')])
self.assertEqual(query.get_result(), [3])
def test_database_setters(self):
query = Query()
query.set_table('tada')
table = Table('table_name', [])
error_text = 'some error'
database = Database('database_name', query)
database.add_table(table)
database.set_error(error_text)
self.assertEqual(database.get_name(), 'database_name')
self.assertEqual(database.get_tables(), {'table_name': table})
self.assertEqual(database.get_error(), error_text)
self.assertEqual(database.has_error(), True)
self.assertEqual(database.get_query(), query)