def test_vault(self): vault = Vault('password', ':test:') vault.add_entry('TestArgon2', 'username2', 'password') print(vault.get_entries()) del vault vault2 = Vault('password', 'test.db.enc') vault2.add_entry('Test1', 'username', 'password3') print(vault2.get_entries())
class TestSQLite3(unittest.TestCase): def setUp(self): self.vault = Vault(':memory:') self.vault.nuke_table() def test_1insert_entry(self): self.vault.add_entry(service1, username1, password1) entries = self.vault.get_entries() self.assertEqual( tuple, type(entries)) # Always tuple if only one entry, else list def test_2no_duplicate_inserts(self): self.vault.add_entry(service1, username1, password1) entries = self.vault.get_entries() self.assertEqual(tuple, type(entries)) def test_3delete_entry(self): self.vault.delete_entry(service1) entries = self.vault.get_entries() self.assertEqual(0, len(entries)) def test_4_nuke_table(self): self.vault.nuke_table() self.assertEqual(0, len(self.vault.get_entries()))