def test_delete_object(self): db = DBConnection("../data_base/test_base.db") self.assertTrue(db.get_object_count() == 0) db.insert_object(1, "1") self.assertTrue(db.get_object_count() == 1) self.assertTrue(db.get_object_name(1) == "1") db.delete_object(1) self.assertTrue(db.get_object_count() == 0) db.clear()
def test_insert_object(self): db = DBConnection("../data_base/test_base.db") db.clear() self.assertRaises(DBConnection("RANDOM_BASE")) self.assertTrue(db.get_object_count() == 0) db.insert_object(1, "1") self.assertTrue(db.get_object_count() == 1) self.assertTrue(db.get_object_name(1) == "1") db.clear() self.assertTrue(db.get_object_count() == 0)