def setUp(self): handle = tempfile.NamedTemporaryFile(delete=False) self.tempdb = handle.name handle.close() self.pwdb = PwDb(self.tempdb) key, self.plain = "thiskey", "thispwd" self.pwdb.set_key(key, self.plain, "pass")
class TestPwDb(unittest.TestCase): def setUp(self): handle = tempfile.NamedTemporaryFile(delete=False) self.tempdb = handle.name handle.close() self.pwdb = PwDb(self.tempdb) key, self.plain = "thiskey", "thispwd" self.pwdb.set_key(key, self.plain, "pass") def tearDown(self): os.remove(self.tempdb) def test_get_pwd(self): dpwd = self.pwdb.get_key("thiskey", "pass") self.assertEqual(self.plain, dpwd) def test_wrong_master(self): dpwd = self.pwdb.get_key("thiskey", "wrong") self.assertTrue(len(dpwd)==0) def test_wrong_key(self): with self.assertRaises(KeyError): self.pwdb.get_key("dontexists", "pass")