def test_rm_twice(self):
pwds = Accounts()
with self.assertRaises(KeyError):
pwds.rm('a', 'b')
pwds.add('a', 'b', 'c')
pwds.rm('a', 'b')
with self.assertRaises(KeyError):
pwds.rm('a', 'b')
def get_all_passwords(datapath):
accounts = Accounts()
for root, _, files in os.walk(datapath):
for fil in files:
if fil.endswith('.gpg'):
(host, user) = path_to_hostuser(root, fil)
accounts.add(host, user, os.path.join(root, fil))
return accounts
def test_add_rm(self):
pwds = Accounts()
self.assertFalse(pwds.exists('host', 'user'))
pwds.add('host', 'user', 'pwd')
self.assertTrue(pwds.exists('host', 'user'))
pwds.rm('host', 'user')
self.assertFalse(pwds.exists('host', 'user'))
def test_iter(self):
tests = [
('host1', 'user1', 'path1'),
('host2', 'user1', 'path2'),
('host2', 'user2', 'path3'),
('host3', 'user1', 'path4'),
('host3', 'user2', 'path5'),
('host3', 'user3', 'path6'),
]
pwds = Accounts()
for (h, u, p) in tests:
pwds.add(h, u, p)
r = []
for a in pwds.iterate():
r.append(a)
self.assertCountEqual(r, tests)
def test_search(self):
pwds = Accounts()
pwds.add('host', 'user', 'path1')
pwds.add('hostx', 'userx', 'path2')
pwds.add('1host1', '1user1', 'path5')
pwds.add('1host', '1user', 'path4')
pwds.add('host1', 'user1', 'path3')
pwds.add('xhost', 'xuser', 'path6')
pwds.add('xhostx', 'xuserx', 'path7')
pwds.add('xhoxstx', 'user', 'path8')
pwds.add('host', 'usxer', 'path9')
l = pwds.search('host', 'user')
# Use assertEqual and not assertItemsEqual here because pwds.search()
# should return matching hosts and users sorted in alphabetical order
self.assertEqual(l, [
('1host', '1user'),
('1host1', '1user1'),
('host', 'user'),
('host1', 'user1'),
('hostx', 'userx'),
('xhost', 'xuser'),
('xhostx', 'xuserx'),
])
def test_search_empty(self):
pwds = Accounts()
l = pwds.search('host', 'user')
self.assertEqual(l, [])
def test_rm_nonexisting(self):
pwds = Accounts()
with self.assertRaises(KeyError):
pwds.rm('a', 'b')
def test_capitalize(self):
pwds = Accounts()
pwds.add('a', 'b', 'c')
self.assertTrue(pwds.exists('a', 'b'))
self.assertTrue(pwds.exists('A', 'b'))
self.assertTrue(pwds.exists('a', 'B'))
self.assertTrue(pwds.exists('A', 'B'))
with self.assertRaises(KeyError):
pwds.add('a', 'b', 'c')
with self.assertRaises(KeyError):
pwds.add('A', 'b', 'c')
with self.assertRaises(KeyError):
pwds.add('a', 'B', 'c')
with self.assertRaises(KeyError):
pwds.add('A', 'B', 'c')
pwds.rm('A', 'B')
self.assertFalse(pwds.exists('a', 'b'))
self.assertFalse(pwds.exists('A', 'b'))
self.assertFalse(pwds.exists('a', 'B'))
self.assertFalse(pwds.exists('A', 'B'))
pwds.add('A', 'B', 'c')
pwds.add('a', 'c', 'd')
pwds.add('A', 'D', 'e')
self.assertTrue(pwds.exists('a', 'b'))
self.assertTrue(pwds.exists('A', 'b'))
self.assertTrue(pwds.exists('a', 'B'))
self.assertTrue(pwds.exists('A', 'B'))
with self.assertRaises(KeyError):
pwds.add('a', 'b', 'c')
with self.assertRaises(KeyError):
pwds.add('A', 'b', 'c')
with self.assertRaises(KeyError):
pwds.add('a', 'B', 'c')
with self.assertRaises(KeyError):
pwds.add('A', 'B', 'c')
self.assertEqual(pwds.get('A', 'b'), ('A', 'B', 'c'))
self.assertEqual(pwds.get('a', 'B'), ('A', 'B', 'c'))
self.assertEqual(pwds.get('A', 'B'), ('A', 'B', 'c'))
self.assertEqual(pwds.get('a', 'c'), ('a', 'c', 'd'))
self.assertEqual(pwds.get('A', 'D'), ('A', 'D', 'e'))