def test_init_with_database(self, input):
query = Query()
query.get_aggregate_function = MagicMock(return_value='COUNT(*)')
query.get_result = MagicMock(return_value=[3])
table = Table('drivers', [
['id', 'first_name', 'last_name', 'country', 'year_of_birth', 'company'],
['1', 'John', 'Carmack', 'USA', '1970', 'id Software'],
['2', 'Jonn', 'Romero', 'USA', '1967', 'id Software'],
['3', 'Elon', 'Musk', 'USA', '1971', 'Tesla'],
['4', 'Damon', 'Hill', 'UK', '1960', 'Williams']])
database = Database('database_name', query)
database.add_table(table)
controller = Controller()
with patch('sys.stdout', new=StringIO()) as fake_out:
controller.init(database)
self.assertEqual(fake_out.getvalue(), 'Tables found: 1\ndrivers\n')
def test_database_proceed_filled_query_string_with_table_not_exist(self):
query = Query()
database = Database('database_name', query)
database.add_table(Table('drivers', []))
database.proceed_query('SELECT company, country FROM drivers1;')
self.assertEqual(database.get_error(), 'Table not exist')
def test_database_proceed_empty_query_string(self):
query = Query()
database = Database('database_name', query)
database.proceed_query('')
self.assertEqual(database.get_error(),
'Wrong query syntax, please try again')