"""Tests for message_extractor module."""

import json
from unittest import mock

import pytest

from app_name.utils import message_extractor


@pytest.fixture
def message_mock():
    """Fixture for Message mocking."""
    message_body = {
        'message_content': {
            'foo': 'bar'
        },
        'message_id': 1
    }

    json_encoded_message = json.dumps({
        'Message': json.dumps(message_body)
    })
    message = mock.MagicMock(body=json_encoded_message)
    return {
        'input_message': message,
        'expected_body': message_body
    }


def test_get_message_body(message_mock):
    """Test for getting topic from order data."""
    # test functional call
    result = message_extractor.get_message_body(
        message_mock['input_message'])
    # checking
    assert result == message_mock['expected_body']