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()))
