"""Helper methods for use in the tests."""

import json
from unittest.mock import MagicMock


def get_message(response):
    """Extract the message field from a JSON response."""
    return get_json_body(response).get('message')


def get_json_body(response):
    """Extract the JSON body from a response."""
    return json.loads(response.data.decode())


def make_feature_flag_mock(mocker: MagicMock):
    """Make a utility function for mocking pythonfeatures flags."""
    feature_map = {}

    def _mock(feature, value):
        feature_map[feature] = value

    def _side_effect(*args, **kwargs):
        feature = args[0]
        value = feature_map.get(feature, False)
        return value

    mocker.side_effect = _side_effect
    return _mock