def setUp(self): self.db = Database()
class BasicTestCase(unittest.TestCase): def setUp(self): self.db = Database() def test_case1(self): self.db.set("ex", 10) self.assertEqual(self.db.get("ex"), 10) self.db.unset("ex") self.assertEqual(self.db.get("ex"), "NULL") def test_case2(self): self.db.set("a", 10) self.db.set("b", 10) self.assertEqual(self.db.numequalto(10), 2) self.assertEqual(self.db.numequalto(20), 0) self.db.set("b", 30) self.assertEqual(self.db.numequalto(10), 1) def test_case3(self): self.db.begin() self.db.set("a", 10) self.assertEqual(self.db.get("a"), 10) self.db.begin() self.db.set("a", 20) self.db.rollback() self.assertEqual(self.db.get("a"), 10) self.db.rollback() self.assertEqual(self.db.get("a"), "NULL") def test_case4(self): self.db.begin() self.db.set("a", 30) self.db.begin() self.db.set("a", 40) self.db.commit() self.assertEqual(self.db.get("a"), 40) self.assertEqual(self.db.rollback(), "NO TRANSACTION") def test_case5(self): self.db.set("a", 50) self.db.begin() self.assertEqual(self.db.get("a"), 50) self.db.set("a", 60) self.db.begin() self.db.unset("a") self.assertEqual(self.db.get("a"), "NULL") self.db.rollback() self.assertEqual(self.db.get("a"), 60) self.db.commit() self.assertEqual(self.db.get("a"), 60) def test_case6(self): self.db.set("a", 10) self.db.begin() self.assertEqual(self.db.numequalto(10), 1) self.db.begin() self.db.unset("a") self.assertEqual(self.db.numequalto(10), 0) self.db.rollback() self.assertEqual(self.db.numequalto(10), 1) self.db.commit()