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

from unittest.mock import MagicMock

from abacus_common_logic.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'


def test_make_feature_flag_mock():
    """Test make_feature_flag_mock method."""
    is_feature_flag_enabled_mock = MagicMock()
    mock_feature = helpers.make_feature_flag_mock(is_feature_flag_enabled_mock)
    mock_feature('no-go', False)
    mock_feature('awyeah', True)

    assert not is_feature_flag_enabled_mock('missing flag')
    assert not is_feature_flag_enabled_mock('no-go')
    assert is_feature_flag_enabled_mock('awyeah')