class ConnectMessageSerializerTestCase(unittest.TestCase):
def setUp(self):
self.serializer = ConnectMessageSerializer()
self.version = 'pre2'
self.support = ['pre2', 'pre1']
self.session = 'session'
def test_with_support_with_session(self):
message = ConnectMessage(self.version, support=self.support,
session=self.session)
pod = self.serializer.serialize(message)
expected = {'msg': 'connect', 'version': self.version,
'support': self.support, 'session': self.session}
self.assertEqual(pod, expected)
def test_with_support_without_session(self):
message = ConnectMessage(self.version, support=self.support)
expected = {'msg': 'connect', 'version': self.version,
'support': self.support}
pod = self.serializer.serialize(message)
self.assertEqual(pod, expected)
def test_without_support_with_session(self):
message = ConnectMessage(self.version, session=self.session)
expected = {'msg': 'connect', 'version': self.version,
'support': [self.version], 'session': self.session}
pod = self.serializer.serialize(message)
self.assertEqual(pod, expected)
def test_without_support_without_session(self):
message = ConnectMessage(self.version)
pod = self.serializer.serialize(message)
self.assertEqual(pod, {'msg': 'connect', 'version': self.version,
'support': [self.version]})
def setUp(self):
self.serializer = ConnectMessageSerializer()
self.version = 'pre2'
self.support = ['pre2', 'pre1']
self.session = 'session'