def __test_send_raises(self, type, payload, expected_exception):
     socket = FakeWritableSocket()
     with self.assertRaises(expected_exception):
         m2proto.send(socket, type, payload)
     self.assertEqual(socket.fake_data(), b'')
 def __test_send(self, type, payload, expected_data):
     socket = FakeWritableSocket()
     self.assertTrue(m2proto.send(socket, type, payload))
     self.assertEqual(socket.fake_data(), expected_data)
 def test_send_empty(self):
     socket = FakeWritableSocket()
     self.assertTrue(m1proto.send(socket, ''))
     self.assertEqual(socket.fake_data(), b'\0')
 def test_send_too_big(self):
     socket = FakeWritableSocket()
     with self.assertRaises(m1proto.PayloadTooBig):
         m1proto.send(socket, 'x' * 256)
     self.assertEqual(socket.fake_data(), b'')
 def test_send_max(self):
     socket = FakeWritableSocket()
     self.assertTrue(m1proto.send(socket, 'x' * 255))
     self.assertEqual(socket.fake_data(), b'\xFF' + (b'x' * 255))
 def test_send_unicode(self):
     socket = FakeWritableSocket()
     self.assertTrue(m1proto.send(socket, '\xA0'))
     self.assertEqual(socket.fake_data(), b'\x02\xC2\xA0')
 def test_send_ascii(self):
     socket = FakeWritableSocket()
     self.assertTrue(m1proto.send(socket, 'abc'))
     self.assertEqual(socket.fake_data(), b'\x03abc')