def test_create(self): table_name = 'password_db' table_model = DataModel(table_name, PasswordData) table_model.new_table() vo = PasswordData("your", "your_password") res = table_model.create(vo) self.assertEqual(res, 1) os.remove("./password_db.db")
def test_new_table(self): table_name = 'password_db' table_model = DataModel(table_name, PasswordData) res = table_model.new_table() self.assertTrue(res) res = table_model.new_table() self.assertFalse(res) os.remove("./password_db.db")
def test_create_batch_and_retrieve_limit(self): table_name = 'password_db' table_model = DataModel(table_name, PasswordData) table_model.new_table() l = [] for x in range(0, 5): l.append(PasswordData('111' + str(x), '222')) table_model.create_batch(l) self.assertEqual(table_model.retrieve("name", "1110"), PasswordData('1110', '222')) self.assertEqual(table_model.retrieve("name", "1114"), PasswordData('1114', '222')) self.assertEqual(len(table_model.retrieve_limit(1, 3)), 3) os.remove("./password_db.db")
def test_update(self): table_name = 'password_db' table_model = DataModel(table_name, PasswordData) table_model.new_table() l = [] for x in range(1, 6): l.append(PasswordData('No' + str(x), str(x))) table_model.create_batch(l) row_size = table_model.update("name", "No3", "password", "new_password") self.assertEqual(row_size, 1) row_size = table_model.update("name", "No8", "password", "new_password") self.assertEqual(row_size, 0) os.remove("./password_db.db")
def test_retrieve(self): table_name = 'password_db' table_model = DataModel(table_name, PasswordData) table_model.new_table() vo1 = PasswordData("your", "your_password") vo2 = PasswordData("me", "hello") vo3 = PasswordData("Jack", "lsjdf23dfs") table_model.create(vo1) table_model.create(vo2) table_model.create(vo3) res = table_model.retrieve("name", "Jack") obj = PasswordData("Jack", "lsjdf23dfs") self.assertEqual(res, obj) res = table_model.retrieve("name", "not_this_name") self.assertEqual(res, None) os.remove("./password_db.db")