def test_delete_slide(self): db = DBConnection("../data_base/test_base.db") self.assertTrue(db.get_slide_count() == 0) db.insert_slide(1, "1") self.assertTrue(db.get_slide_count() == 1) self.assertTrue(db.get_slide_name(1) == "1") db.delete_slide(1) self.assertTrue(db.get_slide_count() == 0)
def test_get_slide_list(self): db = DBConnection("../data_base/test_base.db") self.assertTrue(db.get_slide_count() == 0) db.insert_slide(1, "1") db.insert_slide(2, "2") db.insert_slide(3, "3") db.insert_slide(4, "4") db.insert_slide(5, "5") db.insert_slide(6, "6") self.assertTrue(db.get_slide_count() == 6) test_list = db.get_slide_list() self.assertTrue(test_list[0][2] == "1") self.assertTrue(test_list[1][2] == "2") self.assertTrue(test_list[2][2] == "3") self.assertTrue(test_list[3][2] == "4") self.assertTrue(test_list[4][2] == "5") self.assertTrue(test_list[5][2] == "6") db.clear() self.assertTrue(db.get_object_count() == 0)