66 lines
2.7 KiB
Python
66 lines
2.7 KiB
Python
import pytest
|
|
from django.contrib.auth import get_user_model
|
|
from django.core.cache import cache
|
|
from rest_framework import status
|
|
|
|
from infrasynth.features.models import FeatureFlag
|
|
from infrasynth.features.registry import FeatureRegistry
|
|
|
|
UserModel = get_user_model()
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def clean_registry(clean_feature_registry):
|
|
cache.clear()
|
|
cache.clear()
|
|
yield
|
|
cache.clear()
|
|
|
|
|
|
class TestActiveFlagsEndpoint:
|
|
def test_returns_registry_flags(self, authenticated_client, db):
|
|
FeatureRegistry.register("flag.x", default=True)
|
|
FeatureRegistry.register("flag.y", default=False)
|
|
resp = authenticated_client.get("/api/features/active/")
|
|
assert resp.status_code == status.HTTP_200_OK
|
|
data = resp.json()
|
|
assert "flags" in data
|
|
assert data["flags"]["flag.x"] is True
|
|
assert data["flags"]["flag.y"] is False
|
|
|
|
def test_requires_auth(self, api_client, db):
|
|
resp = api_client.get("/api/features/active/")
|
|
assert resp.status_code == status.HTTP_401_UNAUTHORIZED
|
|
|
|
def test_includes_db_flags(self, authenticated_client, db):
|
|
FeatureRegistry.register("mixed", default=False)
|
|
FeatureFlag.objects.create(slug="mixed", is_active=True)
|
|
resp = authenticated_client.get("/api/features/active/")
|
|
assert resp.json()["flags"]["mixed"] is True
|
|
|
|
def test_includes_user_overrides(self, authenticated_client, user, db):
|
|
FeatureRegistry.register("override_me", default=False)
|
|
flag = FeatureFlag.objects.create(slug="override_me", is_active=False)
|
|
from infrasynth.features.models import FeatureFlagOverride
|
|
|
|
FeatureFlagOverride.objects.create(flag=flag, user=user, is_enabled=True)
|
|
resp = authenticated_client.get("/api/features/active/")
|
|
assert resp.json()["flags"]["override_me"] is True
|
|
|
|
|
|
class TestCheckFlagEndpoint:
|
|
def test_check_enabled(self, authenticated_client, db):
|
|
FeatureRegistry.register("my.flag", default=True)
|
|
resp = authenticated_client.get("/api/features/check/my.flag/")
|
|
assert resp.status_code == status.HTTP_200_OK
|
|
assert resp.json() == {"slug": "my.flag", "is_enabled": True}
|
|
|
|
def test_check_disabled(self, authenticated_client, db):
|
|
FeatureRegistry.register("off.flag", default=False)
|
|
resp = authenticated_client.get("/api/features/check/off.flag/")
|
|
assert resp.status_code == status.HTTP_200_OK
|
|
assert resp.json() == {"slug": "off.flag", "is_enabled": False}
|
|
|
|
def test_check_requires_auth(self, api_client, db):
|
|
resp = api_client.get("/api/features/check/anything/")
|
|
assert resp.status_code == status.HTTP_401_UNAUTHORIZED
|