def test_add_all_tables(self):
db_uri = 'mysql+pymysql://root:@localhost:3306/MyDatabase'
dbm = Manager.DBManager(db_uri)
dbm.dropAllTables()
before = dbm.engine.table_names()
dbm.addAllTables()
after = dbm.engine.table_names()
#this is a list of all tables in CI.py
metadatatables = dbm.print_tables_from_metadata
expected = metadatatables.copy()
expected.update(before)
print(expected)
print(after)
self.assertEqual(after, expected)
def test_drop_single_table(self):
db_uri = 'mysql+pymysql://root:@localhost:3306/MyDatabase'
dbm = Manager.DBManager(db_uri)
dbm.dropSingleTable(Course.__tablename__)
result = dbm.engine.table_names()
self.assertEqual(result, emptylist)
def test_add_single_table(self):
db_uri = 'mysql+pymysql://root:@localhost:3306/MyDatabase'
dbm = Manager.DBManager(db_uri)
dbm.addSingleTable(Course.__table__)
listtables = dbm.engine.table_names()
self.assertIn(Course.__tablename__, listtables)
def test_drop_all_tables(self):
db_uri = 'mysql+pymysql://root:@localhost:3306/MyDatabase'
dbm = Manager.DBManager(db_uri)
dbm.dropAllTables()
result = dbm.engine.table_names()
self.assertEqual(result, emptylist)
