def test_player_receive_json_rpc_response_connection_closed(self):
"""Tests that Player.receive_json_rpc_response_connection_closed() returns None when websocket throws ConnectionClosed exception
"""
websocket = MagicMock()
recv = AsyncMock(side_effect=ConnectionClosed(0, 'Connection Closed'))
websocket.attach_mock(recv, 'recv')
player = Player(None, websocket)
(x, y, z) = asyncio.run(player.receive_json_rpc_response())
self.assertEqual((x, y, z), (
None, None, None
), '(None, None, None) expected for receive_json_rpc_response() when connection is closed'
)
def test_player_receive_json_rpc_response(self):
"""Tests that Player.receive_json_rpc_response() calls websocket recv()
"""
websocket = MagicMock()
recv = AsyncMock()
recv.return_value = '{"id":2323, "error":"noerror", "result":"nothing"}'
websocket.attach_mock(recv, 'recv')
player = Player(None, websocket)
(reqid, error,
result) = asyncio.run(player.receive_json_rpc_response())
self.assertEqual(
reqid, 2323,
"receive_json_rpc_response(() returned unexpected 'id'")
self.assertEqual(
error, 'noerror',
"receive_json_rpc_response(() returned unexpected 'error'")
self.assertEqual(
result, 'nothing',
"receive_json_rpc_response(() returned unexpected 'result''")