"""Test for feature flag utils."""

from unittest.mock import patch

from owsresponse import response

from podcast.utils import feature_flag_utils


@patch('pythonfeatures.pythonfeatures.get_single_feature')
@patch('podcast.utils.feature_flag_utils.g', spec=['request_context'])
def test_get_feature_flag_enabled(mock_g, mock_get_feature_flag):
    """Test for get_feature_flag if enabled."""
    mock_get_feature_flag.return_value = response.Response(message='enabled')
    result = feature_flag_utils.get_feature_flag('test_feature')

    assert result is True


@patch('pythonfeatures.pythonfeatures.get_single_feature', return_value='disabled')
@patch('podcast.utils.feature_flag_utils.g', spec=['request_context'])
def test_get_feature_flag_disabled(mock_g, mock_get_feature_flag):
    """Test for get_feature_flag if disabled."""
    mock_get_feature_flag.return_value = response.Response(message='disabled')
    result = feature_flag_utils.get_feature_flag('test_feature')

    assert result is False