<reponame>walison17/unleash-client-python
import responses
from pytest import mark, param
from tests.utilities.mocks.mock_features import MOCK_FEATURE_RESPONSE, MOCK_FEATURE_RESPONSE_PROJECT
from tests.utilities.testing_constants import URL, APP_NAME, INSTANCE_ID, CUSTOM_HEADERS, CUSTOM_OPTIONS, PROJECT_URL, PROJECT_NAME, ETAG_VALUE
from UnleashClient.constants import FEATURES_URL
from UnleashClient.api import get_feature_toggles


FULL_FEATURE_URL = URL + FEATURES_URL


@responses.activate
@mark.parametrize("response,status,expected", (
    param(MOCK_FEATURE_RESPONSE, 200, lambda result: result["version"] == 1, id="success"),
    param(MOCK_FEATURE_RESPONSE, 202, lambda result: not result, id="failure"),
    param({}, 500, lambda result: not result, id="failure"),
))
def test_get_feature_toggle(response, status, expected):
    responses.add(responses.GET, FULL_FEATURE_URL, json=response, status=status, headers={'etag': ETAG_VALUE})

    (result, etag) = get_feature_toggles(URL,
                                         APP_NAME,
                                         INSTANCE_ID,
                                         CUSTOM_HEADERS,
                                         CUSTOM_OPTIONS)

    assert len(responses.calls) == 1
    assert expected(result)


@responses.activate
def test_get_feature_toggle_project():
    responses.add(responses.GET, PROJECT_URL, json=MOCK_FEATURE_RESPONSE_PROJECT, status=200, headers={'etag': ETAG_VALUE})

    (result, etag) = get_feature_toggles(URL,
                                         APP_NAME,
                                         INSTANCE_ID,
                                         CUSTOM_HEADERS,
                                         CUSTOM_OPTIONS,
                                         PROJECT_NAME)

    assert len(responses.calls) == 1
    assert len(result["features"]) == 1
    assert etag == ETAG_VALUE


@responses.activate
def test_get_feature_toggle_failed_etag():
    responses.add(responses.GET, PROJECT_URL, json={}, status=500, headers={'etag': ETAG_VALUE})

    (result, etag) = get_feature_toggles(URL,
                                         APP_NAME,
                                         INSTANCE_ID,
                                         CUSTOM_HEADERS,
                                         CUSTOM_OPTIONS,
                                         PROJECT_NAME)

    assert len(responses.calls) == 1
    assert etag == ''


@responses.activate
def test_get_feature_toggle_etag_present():
    responses.add(responses.GET, PROJECT_URL, json={}, status=304, headers={'etag': ETAG_VALUE})

    (result, etag) = get_feature_toggles(URL,
                                         APP_NAME,
                                         INSTANCE_ID,
                                         CUSTOM_HEADERS,
                                         CUSTOM_OPTIONS,
                                         PROJECT_NAME,
                                         ETAG_VALUE)

    assert len(responses.calls) == 1
    assert not result
    assert responses.calls[0].request.headers['If-None-Match'] == ETAG_VALUE
    assert etag == ETAG_VALUE

import responses
from pytest import mark, param
from tests.utilities.mocks.mock_features import MOCK_FEATURE_RESPONSE, MOCK_FEATURE_RESPONSE_PROJECT
from tests.utilities.testing_constants import URL, APP_NAME, INSTANCE_ID, CUSTOM_HEADERS, CUSTOM_OPTIONS, PROJECT_URL, PROJECT_NAME, ETAG_VALUE
from UnleashClient.constants import FEATURES_URL
from UnleashClient.api import get_feature_toggles


FULL_FEATURE_URL = URL + FEATURES_URL


@responses.activate
@mark.parametrize("response,status,expected", (
    param(MOCK_FEATURE_RESPONSE, 200, lambda result: result["version"] == 1, id="success"),
    param(MOCK_FEATURE_RESPONSE, 202, lambda result: not result, id="failure"),
    param({}, 500, lambda result: not result, id="failure"),
))
def test_get_feature_toggle(response, status, expected):
    responses.add(responses.GET, FULL_FEATURE_URL, json=response, status=status, headers={'etag': ETAG_VALUE})

    (result, etag) = get_feature_toggles(URL,
                                         APP_NAME,
                                         INSTANCE_ID,
                                         CUSTOM_HEADERS,
                                         CUSTOM_OPTIONS)

    assert len(responses.calls) == 1
    assert expected(result)


@responses.activate
def test_get_feature_toggle_project():
    responses.add(responses.GET, PROJECT_URL, json=MOCK_FEATURE_RESPONSE_PROJECT, status=200, headers={'etag': ETAG_VALUE})

    (result, etag) = get_feature_toggles(URL,
                                         APP_NAME,
                                         INSTANCE_ID,
                                         CUSTOM_HEADERS,
                                         CUSTOM_OPTIONS,
                                         PROJECT_NAME)

    assert len(responses.calls) == 1
    assert len(result["features"]) == 1
    assert etag == ETAG_VALUE


@responses.activate
def test_get_feature_toggle_failed_etag():
    responses.add(responses.GET, PROJECT_URL, json={}, status=500, headers={'etag': ETAG_VALUE})

    (result, etag) = get_feature_toggles(URL,
                                         APP_NAME,
                                         INSTANCE_ID,
                                         CUSTOM_HEADERS,
                                         CUSTOM_OPTIONS,
                                         PROJECT_NAME)

    assert len(responses.calls) == 1
    assert etag == ''


@responses.activate
def test_get_feature_toggle_etag_present():
    responses.add(responses.GET, PROJECT_URL, json={}, status=304, headers={'etag': ETAG_VALUE})

    (result, etag) = get_feature_toggles(URL,
                                         APP_NAME,
                                         INSTANCE_ID,
                                         CUSTOM_HEADERS,
                                         CUSTOM_OPTIONS,
                                         PROJECT_NAME,
                                         ETAG_VALUE)

    assert len(responses.calls) == 1
    assert not result
    assert responses.calls[0].request.headers['If-None-Match'] == ETAG_VALUE
    assert etag == ETAG_VALUE