def test_get_branch_by_id(webapp, session): message = Message(data=b'test') message.calculate_key() session.register_message(message) r = webapp.get('/b/' + message.key) assert r.status_code == 200 assert r.content_type == 'text/uri-list' assert r.body.decode() == message.key
def test_get_message_by_id(webapp, session): message = Message(data=b'test') message.calculate_key() session.register_message(message) r = webapp.get('/m/' + message.key) assert r.status_code == 200 assert r.content_type == 'text/plain' assert r.body == message.data
def test_message_generate_key(): data = 'test'.encode('utf8') public_key = 'public key' message = Message(data=data, public_key=public_key) assert message.data == data assert message.public_key == public_key message.calculate_key() assert message.data == data assert message.public_key == public_key assert message.key is not None
def test_populate_branch_with_messages_in_the_wrong_order(session): messages = {} last = None for c in 'abcdefghij': message = Message(data=b'a') if last: message.in_reply_to = last.key message.calculate_key() last = message messages[c] = message for c in 'efbcagijhd': session.register_message(messages[c]) branch = session.get_branch(messages['a'].key) assert len(session.get_branches()) == 1 assert len(branch) == 10 assert branch.root == messages['a'].key tree = branch.to_flat_tree() assert tree == [ {'key': messages[x].key, 'level': 0} for x in 'abcdefghij' ]