class TestNode(unittest.TestCase): def setUp(self): self.node = Node(username=b'foo', password=b's3kr3t', url=b'example.com', notes=b'just a reminder to self', tags=[b'baz', b'baz']) def test_do_encdict(self): ce = CryptoEngine.get() for k, v in self.node.to_encdict().items(): if k == 'user': self.assertEqual(ce.decrypt(v).decode(), getattr(self.node, 'username')) elif k != 'tags': self.assertEqual(ce.decrypt(v).decode(), getattr(self.node, k)) def test_setters(self): new_node = {'username': b'baz', 'password': b'n3ws3k43t', 'notes': b'i have changed the password', 'url': b'newexample.com', 'tags': [b'tag1', b'tag2']} for k in new_node: setattr(self.node, k, new_node[k]) for attribute in ['username', 'password', 'url', 'notes']: self.assertEqual(bytearray(getattr(self.node, attribute), 'utf-8'), new_node[attribute]) self.assertEqual(bytearray(getattr(self.node, 'username'), 'utf-8'), new_node['username']) self.assertEqual(bytearray(getattr(self.node, 'password'), 'utf-8'), new_node['password']) self.assertEqual(getattr(self.node, 'tags'), new_node['tags'])
class TestNode(unittest.TestCase): def setUp(self): self.node = Node(username=b'foo', password=b's3kr3t', url=b'example.com', notes=b'just a reminder to self', tags=[b'baz', b'baz']) def test_do_encdict(self): ce = CryptoEngine.get() for k, v in self.node.to_encdict().items(): if k == 'user': self.assertEqual( ce.decrypt(v).decode(), getattr(self.node, 'username')) elif k != 'tags': self.assertEqual(ce.decrypt(v).decode(), getattr(self.node, k)) def test_setters(self): new_node = { 'username': b'baz', 'password': b'n3ws3k43t', 'notes': b'i have changed the password', 'url': b'newexample.com', 'tags': [b'tag1', b'tag2'] } for k in new_node: setattr(self.node, k, new_node[k]) for attribute in ['username', 'password', 'url', 'notes']: self.assertEqual(bytearray(getattr(self.node, attribute), 'utf-8'), new_node[attribute]) self.assertEqual(bytearray(getattr(self.node, 'username'), 'utf-8'), new_node['username']) self.assertEqual(bytearray(getattr(self.node, 'password'), 'utf-8'), new_node['password']) self.assertEqual(getattr(self.node, 'tags'), new_node['tags'])