def test_save(self):
""" Test creating and saving a database. """
db = Database()
i_group = db.create_default_group()
e_group = db.create_group(title="eMail")
e1 = i_group.create_entry(title="FirstEntry", username="******", password="******", url="http://example.com")
e2 = i_group.create_entry(title="SecondEntry", username="******", password="******", url="http://example.com")
e3 = e_group.create_entry(title="ThirdEntry", username="******", password="******", url="http://example.com")
ser = db.to_dict(hierarchy=True, hide_passwords=True)
with self.assertRaisesRegexp(ValueError, r"Unable to save without target file."):
db.save(password='******')
stream = BytesIO()
db.save(dbfile=stream, password='******')
stream.seek(0)
with self.assertRaises(exc.AuthenticationError):
db.load(dbfile=stream, password='******')
stream.seek(0)
db.load(dbfile=stream, password='******')
self.maxDiff = None
self.assertEquals(ser, db.to_dict(hierarchy=True, hide_passwords=True))
def test_load_file(self):
"""
Test loading from file path.
"""
db = Database()
kdb = os.path.join(RESOURCES_DIR, 'example.kdb')
with self.assertRaisesRegexp(ValueError, r'Password and/or keyfile is required.'):
db.load(kdb)
db.load(kdb, password='******')
self.assertEquals(kdb, db.filepath)
def test_load_file(self):
"""
Test loading from file path.
"""
db = Database()
kdb = os.path.join(RESOURCES_DIR, 'example.kdb')
with self.assertRaisesRegexp(ValueError,
r'Password and/or keyfile is required.'):
db.load(kdb)
db.load(kdb, password='******')
self.assertEquals(kdb, db.filepath)
def test_load_stream(self):
"""
Test loading from stream.
"""
db = Database()
kdb = os.path.join(RESOURCES_DIR, 'example.kdb')
with open(kdb, 'rb') as fp:
stream = BytesIO(fp.read())
stream.seek(0)
with self.assertRaisesRegexp(ValueError, r'Password and/or keyfile is required.'):
db.load(stream)
stream.seek(0)
db.load(stream, password='******')
def test_load(self):
""" Test loading database """
db = Database()
kdb = os.path.join(RESOURCES_DIR, 'example.kdb')
db.load(kdb, password='******')
#print(db.groups)
# Make assertions about the structure.
top_groups = [g.title for g in db.root.children]
self.assertEquals(['Internet', 'eMail', 'Backup'], top_groups)
self.assertEquals(['A1', 'B1', 'C1'], [g.title for g in db.root.children[0].children])
self.assertEquals(set(['AEntry1', 'AEntry2', 'AEntry3']), set([e.title for e in db.root.children[0].children[0].entries]))
self.assertEquals(['A2'], [g.title for g in db.root.children[0].children[0].children])
def test_load_stream(self):
"""
Test loading from stream.
"""
db = Database()
kdb = os.path.join(RESOURCES_DIR, 'example.kdb')
with open(kdb, 'rb') as fp:
stream = BytesIO(fp.read())
stream.seek(0)
with self.assertRaisesRegexp(
ValueError, r'Password and/or keyfile is required.'):
db.load(stream)
stream.seek(0)
db.load(stream, password='******')
def test_load(self):
""" Test loading database """
db = Database()
kdb = os.path.join(RESOURCES_DIR, 'example.kdb')
db.load(kdb, password='******')
#print(db.groups)
# Make assertions about the structure.
top_groups = [g.title for g in db.root.children]
self.assertEquals(['Internet', 'eMail', 'Backup'], top_groups)
self.assertEquals(['A1', 'B1', 'C1'],
[g.title for g in db.root.children[0].children])
self.assertEquals(
set(['AEntry1', 'AEntry2', 'AEntry3']),
set([e.title for e in db.root.children[0].children[0].entries]))
self.assertEquals(
['A2'],
[g.title for g in db.root.children[0].children[0].children])
def test_save(self):
""" Test creating and saving a database. """
db = Database()
i_group = db.create_default_group()
e_group = db.create_group(title="eMail")
e1 = i_group.create_entry(title="FirstEntry",
username="******",
password="******",
url="http://example.com")
e2 = i_group.create_entry(title="SecondEntry",
username="******",
password="******",
url="http://example.com")
e3 = e_group.create_entry(title="ThirdEntry",
username="******",
password="******",
url="http://example.com")
ser = db.to_dict(hierarchy=True, hide_passwords=True)
with self.assertRaisesRegexp(ValueError,
r"Unable to save without target file."):
db.save(password='******')
stream = BytesIO()
db.save(dbfile=stream, password='******')
stream.seek(0)
with self.assertRaises(exc.AuthenticationError):
db.load(dbfile=stream, password='******')
stream.seek(0)
db.load(dbfile=stream, password='******')
self.maxDiff = None
self.assertEquals(ser, db.to_dict(hierarchy=True, hide_passwords=True))
