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)