def set_up_db(db: DB) -> None:
db.insert_data(users)
db.insert_data(sleep)
db.insert_data(diet)
db.insert_data(fitness)
db.insert_data(goals)
def drop_all_tables(db: DB) -> None:
db.insert_data('drop table Sleep')
db.insert_data('drop table Diet')
db.insert_data('drop table Fitness')
db.insert_data('drop table Goals')
db.insert_data('drop table Users')
class TestDB(unittest.TestCase):
"""
Tests for db.py
"""
@classmethod
def setUpClass(self):
"""
Set up the unit test.
"""
self.db = DB()
self.table = "TestingTable"
cmd = (
"CREATE TABLE TestingTable (id INT(11) UNSIGNED AUTO_INCREMENT PRIMARY KEY, "
"name varchar(30), age INT(3));")
self.db.insert_data(cmd)
def test_1_insert_cmd_success(self):
"""
Test for checking successful database insertion.
"""
cmd = (f"insert into {self.table} (name, age) values ('Naveena', 21);")
self.db.insert_data(cmd)
# this just asserts that we made it this far without an exception being thrown
self.assertTrue(True)
def test_2_select_no_results(self):
"""
Test for ensuring that an incorrect query returns an empty tuple.
"""
query = (f" select * from {self.table} where age = 100;")
results = self.db.select_data(query)
self.assertEqual(results, ())
def test_3_select_with_results(self):
"""
Test for fetching from the database, and compare against the orcale output.
"""
query = (f" select * from {self.table} where name = 'Naveena';")
results = self.db.select_data(query)
self.assertEqual(len(results), 1)
expected = {'id': 1, 'name': 'Naveena', 'age': 21}
self.assertDictEqual(results[0], expected)
def test_4_select_fail(self):
"""
Test for query failure when the corresponding table does not exists.
"""
query = (f" select * from NotExistingTable;")
with self.assertRaises(ProgrammingError):
self.db.select_data(query)
def test_5_insert_cmd_fail(self):
"""
Test for insertion failure when the table columns are incorrect.
"""
cmd = (
f"insert into {self.table} (name, age) values ('Naveena', 'string');"
)
with self.assertRaises(DataError):
self.db.insert_data(cmd)
def test_6_insert_row(self):
"""
Test for inserting rows into the table using 'insert_row()'.
"""
data = {'name': "'Lily'", 'age': 1}
self.db.insert_row(self.table, data)
query = (f" select * from {self.table} where name = 'Lily';")
results = self.db.select_data(query)
self.assertEqual(len(results), 1)
expected = {'id': 2, 'name': 'Lily', 'age': 1}
self.assertDictEqual(results[0], expected)
def test_7_insert_row_1(self):
"""
Test for inserting a row in the table using 'insert_row_1()'
"""
data = {'name': 'ABC', 'age': 10}
self.db.insert_row_1(self.table, data)
query = (f" select * from {self.table} where name = 'ABC';")
results = self.db.select_data(query)
self.assertEqual(len(results), 1)
expected = {'id': 3, 'name': 'ABC', 'age': 10}
self.assertDictEqual(results[0], expected)
def test_8_insert_cmd_success_1(self):
"""
Test for checking successful database insertion using 'insert_data_1()'
"""
cmd = (f"insert into {self.table} (name, age) values (%s, %s);")
self.db.insert_data_1(cmd, ('DEF', 11))
self.assertTrue(True)
@classmethod
def tearDownClass(self):
"""
Tear down method.
"""
cmd = "drop table TestingTable;"
self.db.insert_data(cmd)