Party/tests/test_message.py

28 lines
842 B
Python

import pytest
from party import message as m
from tests import message
class TestMessage:
def test_setId(self, message):
assert message.id == 0
message.setId(1)
assert message.id == 1
def test_toJson(self, message):
json = message.toJson()
assert json['author'] == 'author'
assert json['content'] == 'content'
assert json['timestamp'] == 'timestamp'
assert json['id'] == 0
def test_fromJson(self, message):
json = message.toJson()
message2 = m.Message.fromJson(json)
assert message2.author == 'author'
assert message2.content == 'content'
assert message2.timestamp == 'timestamp'
def test_equals(self, message):
message2 = m.Message('author', 'content', 'timestamp', 0)
assert message.equals(message2)