def test_backend_connection_send_message_send_failed():
mock_serializer = mock.MagicMock()
mock_serializer.serialize = mock.MagicMock(return_value=mock.sentinel.SERIALIZED)
mock_socket = mock.MagicMock()
mock_socket.send = mock.MagicMock(NotImplementedError)
connection = BackendConnection(mock.sentinel.FRONTEND, mock.sentinel.SERVICE_CONFIG, [], serializer=mock_serializer)
connection._socket = mock_socket
connection.state = State.Connected
# no message is sent if serialization fails, but no exception surfaces either:
connection.send_message(mock.sentinel.MESSAGE)
def test_backend_connection_send_message_serialization_failed():
mock_serializer = mock.MagicMock()
mock_serializer.serialize = mock.MagicMock(side_effect=NotImplementedError)
mock_socket = mock.MagicMock()
mock_socket.send = mock.MagicMock()
connection = BackendConnection(mock.sentinel.FRONTEND, mock.sentinel.SERVICE_CONFIG, [], serializer=mock_serializer)
connection._socket = mock_socket
connection.state = State.Connected
# no message is sent if serialization fails:
connection.send_message(mock.sentinel.MESSAGE)
assert not mock_socket.send.called
def test_backend_connection_send_message_send_failed():
mock_serializer = mock.MagicMock()
mock_serializer.serialize = mock.MagicMock(
return_value=mock.sentinel.SERIALIZED)
mock_socket = mock.MagicMock()
mock_socket.send = mock.MagicMock(NotImplementedError)
connection = BackendConnection(mock.sentinel.FRONTEND,
mock.sentinel.SERVICE_CONFIG, [],
serializer=mock_serializer)
connection._socket = mock_socket
connection.state = State.Connected
# no message is sent if serialization fails, but no exception surfaces either:
connection.send_message(mock.sentinel.MESSAGE)
def test_backend_connection_send_message_serialization_failed():
mock_serializer = mock.MagicMock()
mock_serializer.serialize = mock.MagicMock(side_effect=NotImplementedError)
mock_socket = mock.MagicMock()
mock_socket.send = mock.MagicMock()
connection = BackendConnection(mock.sentinel.FRONTEND,
mock.sentinel.SERVICE_CONFIG, [],
serializer=mock_serializer)
connection._socket = mock_socket
connection.state = State.Connected
# no message is sent if serialization fails:
connection.send_message(mock.sentinel.MESSAGE)
assert not mock_socket.send.called
def test_backend_connection_send_message_wrong_state():
mock_serializer = mock.MagicMock()
mock_serializer.serialize = mock.MagicMock(return_value=mock.sentinel.SERIALIZED)
mock_socket = mock.MagicMock()
mock_socket.send = mock.MagicMock()
connection = BackendConnection(mock.sentinel.FRONTEND, mock.sentinel.SERVICE_CONFIG, [], serializer=mock_serializer)
connection._socket = mock_socket
# no message is sent in any state that is not connected:
for state in {State.Disconnected, State.Connecting, State.Disconnecting}:
connection.state = state
connection.send_message(mock.sentinel.MESSAGE)
assert not mock_serializer.serialize.called
assert not mock_socket.send.called
def test_backend_connection_send_message_ok():
mock_serializer = mock.MagicMock()
mock_serializer.serialize = mock.MagicMock(return_value=mock.sentinel.SERIALIZED)
mock_socket = mock.MagicMock()
mock_socket.send = mock.MagicMock()
connection = BackendConnection(mock.sentinel.FRONTEND, mock.sentinel.SERVICE_CONFIG, [], serializer=mock_serializer)
connection._socket = mock_socket
connection.state = State.Connected
# we have a socket and are connected, so sending the message must go through just fine:
connection.send_message(mock.sentinel.MESSAGE)
mock_serializer.serialize.assert_called_once_with(mock.sentinel.MESSAGE)
mock_socket.send.assert_called_once_with(mock.sentinel.SERIALIZED)
def test_backend_connection_send_message_wrong_state():
mock_serializer = mock.MagicMock()
mock_serializer.serialize = mock.MagicMock(
return_value=mock.sentinel.SERIALIZED)
mock_socket = mock.MagicMock()
mock_socket.send = mock.MagicMock()
connection = BackendConnection(mock.sentinel.FRONTEND,
mock.sentinel.SERVICE_CONFIG, [],
serializer=mock_serializer)
connection._socket = mock_socket
# no message is sent in any state that is not connected:
for state in {State.Disconnected, State.Connecting, State.Disconnecting}:
connection.state = state
connection.send_message(mock.sentinel.MESSAGE)
assert not mock_serializer.serialize.called
assert not mock_socket.send.called
def test_backend_connection_send_message_ok():
mock_serializer = mock.MagicMock()
mock_serializer.serialize = mock.MagicMock(
return_value=mock.sentinel.SERIALIZED)
mock_socket = mock.MagicMock()
mock_socket.send = mock.MagicMock()
connection = BackendConnection(mock.sentinel.FRONTEND,
mock.sentinel.SERVICE_CONFIG, [],
serializer=mock_serializer)
connection._socket = mock_socket
connection.state = State.Connected
# we have a socket and are connected, so sending the message must go through just fine:
connection.send_message(mock.sentinel.MESSAGE)
mock_serializer.serialize.assert_called_once_with(mock.sentinel.MESSAGE)
mock_socket.send.assert_called_once_with(mock.sentinel.SERIALIZED)
def test_backend_connection_request_message_no_token_attribute():
connection = BackendConnection(mock.sentinel.FRONTEND,
mock.sentinel.SERVICE_CONFIG, [])
connection.state = State.Connected
with pytest.raises(AttributeError):
connection.request_message(Shutdown(), mock.sentinel.DURATION)
def test_backend_connection_request_message_no_token_attribute():
connection = BackendConnection(mock.sentinel.FRONTEND, mock.sentinel.SERVICE_CONFIG, [])
connection.state = State.Connected
with pytest.raises(AttributeError):
connection.request_message(Shutdown(), mock.sentinel.DURATION)
