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