class PongMessageParserTestCase(unittest.TestCase): def setUp(self): self.parser = PongMessageParser() def test_with_id(self): id = 'test' message = self.parser.parse({'msg': 'pong', 'id': id}) self.assertEqual(message, PongMessage(id=id)) def test_without_id(self): message = self.parser.parse({'msg': 'pong'}) self.assertEqual(message, PongMessage())
def setUp(self): self.parser = PongMessageParser()