def test_resolve_top_level_list_attribute(self):
ws_message = WSMessage().with_attribute('colour', '${/0/colour}')
response = [
{'colour': 'red'},
{'colour': 'green'},
{'colour': 'blue'}
]
expected_value = '{"colour": "red"}'
ws_message = ws_message.resolve(response)
self.assertEqual(expected_value, str(ws_message))
def test_add_key_value_message(self):
message = WSMessage().with_attribute('test', 123)
ws_tester = (WSTest('wss://example.com').with_message(message))
self.assertEqual(1, len(ws_tester.messages))
self.assertEqual(message, ws_tester.messages[0])
async def test_websocket_senfing_message_with_no_delay(
self, mock_websockets, mock_sleep):
ws_tester = (WSTest('ws://example.com').with_message(
WSMessage().with_attribute('test', 123)))
mock_socket = MagicMock()
mock_socket.close = MagicMock(return_value=asyncio.Future())
mock_socket.close.return_value.set_result(MagicMock())
send_future = asyncio.Future()
send_future.set_result({})
mock_socket.send = MagicMock(return_value=send_future)
mock_socket.recv = MagicMock(return_value=asyncio.Future())
mock_socket.recv.return_value.set_result(MagicMock())
mock_websockets.return_value = asyncio.Future()
mock_websockets.return_value.set_result(mock_socket)
await ws_tester.run()
self.assertTrue(ws_tester.is_complete())
mock_socket.send.assert_called_once_with('{"test": 123}')
mock_sleep.assert_not_called()
mock_socket.close.assert_called_once()
async def test_websocket_test_receive_response_with_unresolved_trigger(
self, mock_ssl, mock_websockets):
ws_tester = (WSTest('wss://example.com').with_response(
WSResponse().with_attribute('type').with_trigger(
WSMessage().with_attribute('test', '${body}'))))
mock_socket = MagicMock()
mock_socket.close = MagicMock(return_value=asyncio.Future())
mock_socket.close.return_value.set_result(MagicMock())
send_future = asyncio.Future()
send_future.set_result({})
mock_socket.send = MagicMock(return_value=send_future)
receive_future = asyncio.Future()
receive_future.set_result(json.dumps({'type': 'Hello, world!'}))
mock_socket.recv = MagicMock(
side_effect=[receive_future, asyncio.Future()])
mock_websockets.return_value = asyncio.Future()
mock_websockets.return_value.set_result(mock_socket)
ssl_context = MagicMock()
mock_ssl.return_value = ssl_context
await ws_tester.run()
self.assertTrue(ws_tester.is_complete())
mock_socket.send.assert_called_once_with('{"test": "${body}"}')
mock_socket.close.assert_called_once()
async def test_websocket_test_send_single_message(self, mock_ssl,
mock_websockets):
message = WSMessage().with_attribute('test', 123)
ws_tester = WSTest('wss://example.com').with_message(message)
mock_socket = MagicMock()
mock_socket.close = MagicMock(return_value=asyncio.Future())
mock_socket.close.return_value.set_result(MagicMock())
future = asyncio.Future()
mock_socket.send = MagicMock(return_value=future)
future.set_result({})
mock_websockets.return_value = asyncio.Future()
mock_websockets.return_value.set_result(mock_socket)
ssl_context = MagicMock()
mock_ssl.return_value = ssl_context
await ws_tester.run()
self.assertFalse(ws_tester.messages)
self.assertEqual(1, len(ws_tester.sent_messages))
mock_socket.send.assert_called_once_with('{"test": 123}')
mock_socket.close.assert_called_once()
def test_with_delay(self):
ws_message = (
WSMessage()
.with_attribute('test', 123)
.with_delay(0.1)
)
self.assertEqual(0.1, ws_message.delay)
def test_resolve_attribute(self):
ws_message = (
WSMessage()
.with_attribute('test', 123)
.with_attribute('example', '${body/example}')
)
response = {
'type': 'message',
'body': {
'example': 456
}
}
expected_value = '{"test": 123, "example": 456}'
ws_message = ws_message.resolve(response)
self.assertEqual(expected_value, str(ws_message))
async def test_websocket_message_timeout(self, mock_ssl, mock_websockets):
ws_tester = (
WSTest('wss://example.com').with_message_timeout(0.1).with_message(
WSMessage().with_attribute('test', 123)))
mock_socket = MagicMock()
mock_socket.close = MagicMock(return_value=asyncio.Future())
mock_socket.close.return_value.set_result(MagicMock())
mock_socket.send = MagicMock(return_value=asyncio.Future())
mock_websockets.return_value = asyncio.Future()
mock_websockets.return_value.set_result(mock_socket)
ssl_context = MagicMock()
mock_ssl.return_value = ssl_context
self.assertEqual(ws_tester.message_timeout, 0.1)
with self.assertRaises(WSTimeoutError):
await ws_tester.run()
mock_socket.close.assert_called_once()
def test_with_trigger(self):
message = WSMessage().with_attribute('test', 123)
ws_response = WSResponse().with_trigger(message)
self.assertEqual(1, len(ws_response.triggers))
self.assertEqual(message, ws_response.triggers[0])
def test_create_ws_message(self):
ws_message = WSMessage()
self.assertDictEqual({}, ws_message.attributes)
def test_stringify_with_attribute(self):
ws_message = WSMessage().with_attribute('test', 123)
expected_value = '{"test": 123}'
self.assertEqual(expected_value, str(ws_message))
def test_with_attribute(self):
ws_message = WSMessage().with_attribute('test', 123)
self.assertIn('test', ws_message.attributes)
self.assertEqual(1, len(ws_message.attributes))