2c10fa583b
# What this PR does ## Which issue(s) this PR fixes ## Checklist - [ ] Tests updated - [ ] Documentation added - [ ] `CHANGELOG.md` updated --------- Co-authored-by: Vadim Stepanov <vadimkerr@gmail.com> Co-authored-by: Joey Orlando <joey.orlando@grafana.com>
108 lines
3.4 KiB
Python
108 lines
3.4 KiB
Python
import pytest
|
|
from django.urls import reverse
|
|
from rest_framework import status
|
|
from rest_framework.test import APIClient
|
|
|
|
from apps.api.views.features import (
|
|
FEATURE_GRAFANA_CLOUD_CONNECTION,
|
|
FEATURE_GRAFANA_CLOUD_NOTIFICATIONS,
|
|
FEATURE_LIVE_SETTINGS,
|
|
FEATURE_SLACK,
|
|
FEATURE_TELEGRAM,
|
|
FEATURE_WEB_SCHEDULES,
|
|
)
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_features_view(
|
|
make_organization_and_user_with_plugin_token,
|
|
make_user_auth_headers,
|
|
):
|
|
"""
|
|
Test access to features without credentials
|
|
"""
|
|
organization, user, token = make_organization_and_user_with_plugin_token()
|
|
client = APIClient()
|
|
url = reverse("api-internal:features")
|
|
response = client.get(url, format="json", **make_user_auth_headers(user, token))
|
|
|
|
assert response.status_code == status.HTTP_200_OK
|
|
assert isinstance(response.json(), list)
|
|
|
|
|
|
@pytest.mark.django_db
|
|
@pytest.mark.parametrize(
|
|
"feature_attr,expected_feature",
|
|
[
|
|
("FEATURE_SLACK_INTEGRATION_ENABLED", FEATURE_SLACK),
|
|
("FEATURE_TELEGRAM_INTEGRATION_ENABLED", FEATURE_TELEGRAM),
|
|
("FEATURE_LIVE_SETTINGS_ENABLED", FEATURE_LIVE_SETTINGS),
|
|
("FEATURE_WEB_SCHEDULES_ENABLED", FEATURE_WEB_SCHEDULES),
|
|
],
|
|
)
|
|
def test_core_features_switch(
|
|
settings,
|
|
make_organization_and_user_with_plugin_token,
|
|
make_user_auth_headers,
|
|
feature_attr,
|
|
expected_feature,
|
|
):
|
|
_, user, token = make_organization_and_user_with_plugin_token()
|
|
setattr(settings, feature_attr, True)
|
|
|
|
client = APIClient()
|
|
url = reverse("api-internal:features")
|
|
response = client.get(url, format="json", **make_user_auth_headers(user, token))
|
|
|
|
assert response.status_code == status.HTTP_200_OK
|
|
assert expected_feature in response.json()
|
|
|
|
setattr(settings, feature_attr, False)
|
|
response = client.get(url, format="json", **make_user_auth_headers(user, token))
|
|
|
|
assert response.status_code == status.HTTP_200_OK
|
|
assert expected_feature not in response.json()
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_oss_features_enabled_in_oss_installation_by_default(
|
|
make_organization_and_user_with_plugin_token,
|
|
make_user_auth_headers,
|
|
):
|
|
_, user, token = make_organization_and_user_with_plugin_token()
|
|
client = APIClient()
|
|
url = reverse("api-internal:features")
|
|
response = client.get(url, format="json", **make_user_auth_headers(user, token))
|
|
|
|
assert response.status_code == status.HTTP_200_OK
|
|
assert FEATURE_GRAFANA_CLOUD_CONNECTION in response.json()
|
|
assert FEATURE_GRAFANA_CLOUD_NOTIFICATIONS in response.json()
|
|
|
|
|
|
@pytest.mark.django_db
|
|
@pytest.mark.parametrize(
|
|
"feature_attr,expected_feature",
|
|
[
|
|
("GRAFANA_CLOUD_NOTIFICATIONS_ENABLED", FEATURE_GRAFANA_CLOUD_NOTIFICATIONS),
|
|
],
|
|
)
|
|
def test_oss_features_switch(
|
|
settings,
|
|
make_organization_and_user_with_plugin_token,
|
|
make_user_auth_headers,
|
|
feature_attr,
|
|
expected_feature,
|
|
):
|
|
_, user, token = make_organization_and_user_with_plugin_token()
|
|
setattr(settings, feature_attr, True)
|
|
|
|
client = APIClient()
|
|
url = reverse("api-internal:features")
|
|
response = client.get(url, format="json", **make_user_auth_headers(user, token))
|
|
|
|
assert response.status_code == status.HTTP_200_OK
|
|
assert expected_feature in response.json()
|
|
|
|
setattr(settings, feature_attr, False)
|
|
response = client.get(url, format="json", **make_user_auth_headers(user, token))
|
|
assert expected_feature not in response.json()
|