7bad073626
It's a duplicate of LICENSE env var **What this PR does**: Remove OSS_INSTALLATION env var in favour of LICENSE env var. Also, I refactored features tests a little. From my point of view it makes little sense to test if all features are disabled or enabled. Better to test specific use-case (e.g. oss installation). Also to test that all features are disabled it is needed to set LICENSE equals cloud license, which makes test confusing. **Checklist** - [x] Tests updated - [ ] Documentation added - [ ] `CHANGELOG.md` updated
110 lines
3.5 KiB
Python
110 lines
3.5 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_MOBILE_APP,
|
|
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),
|
|
("FEATURE_MOBILE_APP_INTEGRATION_ENABLED", FEATURE_MOBILE_APP),
|
|
],
|
|
)
|
|
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()
|