예제 #1
0
def test_handle_invalid_message(mock_rabbit, mocker, capsys):
    # ARRANGE
    eventListener = EventListener()

    xml = "<broken></roken>"
    spy = mocker.spy(eventListener.rabbitClient, "send_message")
    channel_mock = mocker.MagicMock()
    method_mock = mocker.MagicMock()
    method_mock.delivery_tag = 1

    # ACT
    eventListener.handle_message(channel_mock, method_mock, None, xml)
    captured = capsys.readouterr()

    # ASSERT
    assert not spy.called
    assert "error" in captured.out
예제 #2
0
def test_handle_message(event, mock_rabbit, mocker):
    # ARRANGE
    eventListener = EventListener()
    xml = resources.load_xml_resource(event)

    spy = mocker.spy(eventListener.rabbitClient, "send_message")
    channel_mock = mocker.MagicMock()
    method_mock = mocker.MagicMock()
    method_mock.delivery_tag = 1
    method_mock.routing_key = "test_key"

    expected_key = f"test_key.{event}"

    # ACT
    eventListener.handle_message(channel_mock, method_mock, None, xml)

    # ASSERT
    spy.assert_called_with(xml, expected_key)