def test_build_error_message_not_error_prone(self):
hello_message = wamp.HelloMessage()
result = wamp.build_error_message(
in_message=hello_message.json,
uri="bad.bed.bug",
description="don't let the bed bug bite you",
)
self.assertIsNone(result)
def test_connection_succeeds(self):
request = self.build_request()
ws = yield websocket_connect(request)
msg = messages.HelloMessage(realm="burger.friday")
ws.write_message(msg.json)
response = yield ws.read_message()
message = messages.Message.from_text(response)
self.assertIs(message.code, messages.Code.WELCOME)
self.assertFalse(getattr(ws, "close_code", False))
ws.close()
def test_connection_with_xforwardedfor(self):
request = self.build_request(headers={"X-Forwarded-For": "10.0.0.1"})
ws = yield websocket_connect(request)
msg = messages.HelloMessage(realm="burger.thursday")
ws.write_message(msg.json)
connection_ip = list(session.connections.values())[0]
self.assertTrue(connection_ip.peer.startswith("10.0.0.1"))
response = yield ws.read_message()
message = messages.Message.from_text(response)
self.assertIs(message.code, messages.Code.WELCOME)
self.assertFalse(getattr(ws, "close_code", False))
ws.close()
def test_hello_message(self):
hello_message = wamp.HelloMessage(realm="world", details={"roles": {}})
self.assertEqual(hello_message.code, Code.HELLO)
self.assertEqual(hello_message.realm, "world")
self.assertEqual(hello_message.details, {"roles": {}})
self.assertEqual(hello_message.json, '[1, "world", {"roles": {}}]')