118 lines
5 KiB
Python
118 lines
5 KiB
Python
import uuid
|
|
|
|
import pytest
|
|
from django.contrib.auth import get_user_model
|
|
from django.core.cache import cache
|
|
|
|
from infrasynth.features.models import FeatureFlag, FeatureFlagOverride
|
|
from infrasynth.features.registry import FeatureRegistry
|
|
from infrasynth.features.services import FeatureService
|
|
|
|
UserModel = get_user_model()
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def clean_registry(clean_feature_registry):
|
|
cache.clear()
|
|
cache.clear()
|
|
yield
|
|
|
|
|
|
@pytest.fixture
|
|
def user(db):
|
|
return UserModel.objects.create_user(username="testuser", password="pass123")
|
|
|
|
|
|
@pytest.fixture
|
|
def service():
|
|
return FeatureService()
|
|
|
|
|
|
class TestIsEnabled:
|
|
def test_registry_default_when_no_db_flag(self, service, db):
|
|
FeatureRegistry.register("my.flag", default=True)
|
|
assert service.is_enabled("my.flag") is True
|
|
|
|
def test_registry_default_false_when_no_db_flag(self, service, db):
|
|
FeatureRegistry.register("my.flag", default=False)
|
|
assert service.is_enabled("my.flag") is False
|
|
|
|
def test_fallback_default_when_not_registered(self, service, db):
|
|
assert service.is_enabled("unknown.flag") is False
|
|
|
|
def test_fallback_default_explicit(self, service, db):
|
|
assert service.is_enabled("unknown.flag", default=True) is True
|
|
|
|
def test_db_flag_overrides_registry_default(self, service, db):
|
|
FeatureRegistry.register("my.flag", default=False)
|
|
FeatureFlag.objects.create(slug="my.flag", is_active=True)
|
|
assert service.is_enabled("my.flag") is True
|
|
|
|
def test_db_flag_false_overrides_registry(self, service, db):
|
|
FeatureRegistry.register("my.flag", default=True)
|
|
FeatureFlag.objects.create(slug="my.flag", is_active=False)
|
|
assert service.is_enabled("my.flag") is False
|
|
|
|
def test_user_override_enables(self, service, user, db):
|
|
FeatureRegistry.register("my.flag", default=False)
|
|
flag = FeatureFlag.objects.create(slug="my.flag", is_active=False)
|
|
FeatureFlagOverride.objects.create(flag=flag, user=user, is_enabled=True)
|
|
assert service.is_enabled("my.flag", user=user) is True
|
|
|
|
def test_user_override_disables(self, service, user, db):
|
|
FeatureRegistry.register("my.flag", default=True)
|
|
flag = FeatureFlag.objects.create(slug="my.flag", is_active=True)
|
|
FeatureFlagOverride.objects.create(flag=flag, user=user, is_enabled=False)
|
|
assert service.is_enabled("my.flag", user=user) is False
|
|
|
|
def test_user_override_wins_over_db(self, service, user, db):
|
|
FeatureRegistry.register("my.flag", default=False)
|
|
flag = FeatureFlag.objects.create(slug="my.flag", is_active=True)
|
|
FeatureFlagOverride.objects.create(flag=flag, user=user, is_enabled=False)
|
|
assert service.is_enabled("my.flag", user=user) is False
|
|
|
|
def test_unauthenticated_user_ignores_override(self, service, db):
|
|
FeatureRegistry.register("my.flag", default=True)
|
|
anon = UserModel.objects.create(username="anon", password="x")
|
|
assert service.is_enabled("my.flag", user=anon) is True
|
|
|
|
def test_tenant_id_scope(self, service, db):
|
|
FeatureRegistry.register("tenant.flag", default=False)
|
|
tid = uuid.UUID("00000000-0000-0000-0000-000000000001")
|
|
FeatureFlag.objects.create(slug="tenant.flag", is_active=True, tenant_id=tid)
|
|
assert service.is_enabled("tenant.flag", tenant_id=tid) is True
|
|
|
|
def test_global_flag_used_when_no_tenant(self, service, db):
|
|
FeatureRegistry.register("g.flag", default=False)
|
|
FeatureFlag.objects.create(slug="g.flag", is_active=True, tenant_id=None)
|
|
tid = uuid.UUID("00000000-0000-0000-0000-000000000002")
|
|
assert service.is_enabled("g.flag", tenant_id=tid) is False
|
|
|
|
def test_tenant_specific_flag(self, service, db):
|
|
FeatureRegistry.register("t.flag", default=False)
|
|
tid = uuid.UUID("00000000-0000-0000-0000-000000000003")
|
|
FeatureFlag.objects.create(slug="t.flag", is_active=True, tenant_id=tid)
|
|
assert service.is_enabled("t.flag", tenant_id=tid) is True
|
|
other_tid = uuid.UUID("00000000-0000-0000-0000-000000000004")
|
|
assert service.is_enabled("t.flag", tenant_id=other_tid) is False
|
|
|
|
|
|
class TestGetActiveFlags:
|
|
def test_returns_registry_flags_when_no_db(self, service, db):
|
|
FeatureRegistry.register("flag.a", default=True)
|
|
FeatureRegistry.register("flag.b", default=False)
|
|
FeatureRegistry.register("flag.c", default=True)
|
|
result = service.get_active_flags()
|
|
assert result == {"flag.a": True, "flag.b": False, "flag.c": True}
|
|
|
|
def test_includes_db_only_flags(self, service, db):
|
|
FeatureFlag.objects.create(slug="db_only", is_active=True)
|
|
result = service.get_active_flags()
|
|
assert "db_only" in result
|
|
|
|
def test_respects_user_overrides(self, service, user, db):
|
|
FeatureRegistry.register("my.flag", default=False)
|
|
flag = FeatureFlag.objects.create(slug="my.flag", is_active=False)
|
|
FeatureFlagOverride.objects.create(flag=flag, user=user, is_enabled=True)
|
|
result = service.get_active_flags(user=user)
|
|
assert result["my.flag"] is True
|