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_it_fills_aggregate_function_field(self): query = Query() query.set_aggregate_function = MagicMock() database = Database('database_name', query) database.add_table(Table('drivers', [])) database.proceed_query('SELECT COUNT(*) FROM drivers;') query.set_aggregate_function.assert_called_once_with('COUNT(*)')