def test_closing_socket_by_client(self):
self.socket.recv.return_value = "\xFF\x00"
ws = WebSocket(self.socket, self.protocol)
self.assertFalse(ws.closed)
self.assertEquals(ws.wait(), None)
self.assertTrue(ws.closed)
self.assertEquals(self.socket.shutdown.call_count, 0)
self.assertEquals(self.socket.close.call_count, 0)
def test_closing_socket_by_client(self):
self.socket.recv.return_value = '\xFF\x00'
ws = WebSocket(self.socket, self.protocol)
self.assertFalse(ws.closed)
self.assertEquals(ws.wait(), None)
self.assertTrue(ws.closed)
self.assertEquals(self.socket.shutdown.call_count, 0)
self.assertEquals(self.socket.close.call_count, 0)
def test_message_receiving(self):
ws = WebSocket(self.socket, self.protocol)
self.assertFalse(ws.closed)
results = ["\x00spam & eggs\xFF", "\x00K\xc3\xbcss die Hand sch\xc3\xb6ne Frau\xFF", "\xFF\x00"][::-1]
def return_results(*args, **kwargs):
return results.pop()
self.socket.recv.side_effect = return_results
self.assertEquals(ws.wait(), u"spam & eggs")
self.assertEquals(ws.wait(), u"Küss die Hand schöne Frau")
def test_message_receiving(self):
ws = WebSocket(self.socket, self.protocol)
self.assertFalse(ws.closed)
results = [
'\x00spam & eggs\xFF',
'\x00K\xc3\xbcss die Hand sch\xc3\xb6ne Frau\xFF',
'\xFF\x00'][::-1]
def return_results(*args, **kwargs):
return results.pop()
self.socket.recv.side_effect = return_results
self.assertEquals(ws.wait(), u'spam & eggs')
self.assertEquals(ws.wait(), u'Küss die Hand schöne Frau')
