def test_iterate_over_tree(self): """ Test the iteration over a tree. """ tree = create_set() expected = '"thirdfolder": { "description": "thirdLevel", '\ '"accounts": [ { "name": "p4", "password": "******"} ], '\ '"folders": [{ }] }' output = iterate_over_tree(tree.folders[0].folders[0], '') self.assertEqual(output, expected)
def test_tree_to_json(self): """ Test the conversion of a tree to a json string. """ tree = create_set() expected = '{"rootfolder": { "description": "firstLevel", '\ '"accounts": [ { "name": "p1", "password": "******"} , '\ '{ "name": "p2", "password": "******"} ], "folders": '\ '[{"secfolder": { "description": "secLevel", "accounts": '\ '[ { "name": "p3", "password": "******"} ], "folders": '\ '[{"thirdfolder": { "description": "thirdLevel", "accounts": '\ '[ { "name": "p4", "password": "******"} ], "folders": '\ '[{ }] } }] } }] }}' self.assertEqual(tree.dump(), expected)
def test_load_account(self): """ Test the load_account function. """ tree = create_set() name = 'test_name' passw = 'test_password' url = 'http://example.org' desc = 'This is a test account' account = {'name': name, 'password': passw, 'url': url, 'description': desc } output = load_account(account) self.assertEqual(output.name, name) self.assertEqual(output.extras['description'], desc) self.assertEqual(output.extras['url'], url) self.assertEqual(output.password, passw)
def test_load_account(self): """ Test the load_account function. """ tree = create_set() name = 'test_name' passw = 'test_password' url = 'http://example.org' desc = 'This is a test account' account = { 'name': name, 'password': passw, 'url': url, 'description': desc } output = load_account(account) self.assertEqual(output.name, name) self.assertEqual(output.extras['description'], desc) self.assertEqual(output.extras['url'], url) self.assertEqual(output.password, passw)
def test_create_set(self): """ Test the generation of a tree. """ tree = create_set() self.assertEqual(tree.name, 'rootfolder') self.assertEqual(tree.folders[0].name, 'secfolder') self.assertEqual(tree.folders[0].folders[0].name, 'thirdfolder')