def test_init_new(self):
""" Test initializing new database. """
db = Database()
db.create_default_group()
exp_g = model.Group(title="Internet", icon=1, level=0, id=1, db=db, parent=db.root)
self.assertEquals(1, len(db.groups))
self.assertEquals(exp_g.__dict__, db.groups[0].__dict__)
self.assertEquals([], db.groups[0].entries)
def test_init_new(self):
""" Test initializing new database. """
db = Database()
db.create_default_group()
exp_g = model.Group(title="Internet",
icon=1,
level=0,
id=1,
db=db,
parent=db.root)
self.assertEquals(1, len(db.groups))
self.assertEquals(exp_g.__dict__, db.groups[0].__dict__)
self.assertEquals([], db.groups[0].entries)
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_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))
