"""Test for helper's module from test_utils."""

from unittest.mock import MagicMock

from royalty_common.test_utils import helpers


def test_get_json_body():
    """Test get_json_body method."""
    mock_response = MagicMock()
    mock_response.data.decode.return_value = '{"result": "ok"}'
    result = helpers.get_json_body(mock_response)
    assert result == {'result': 'ok'}


def test_get_message():
    """Test get_message method."""
    mock_response = MagicMock()
    mock_response.data.decode.return_value = '{"message": "awesome"}'
    result = helpers.get_message(mock_response)
    assert result == 'awesome'