from label_studio.core.feature_flags.base import flag_set
from label_studio.core.feature_flags.utils import get_user_repr_from_organization


def test_get_user_repr_from_organization_owner_email_and_org_id(django_user_model):
    # Create a minimal organization-like object
    class Org:
        def __init__(self, id, email):
            self.id = id

            class CreatedBy:
                def __init__(self, email):
                    self.email = email

            self.created_by = CreatedBy(email)

    org = Org(123, 'owner@example.com')

    ctx = get_user_repr_from_organization(org)

    assert ctx['key'] == 'owner@example.com'
    assert ctx['custom']['organization'] == 'owner@example.com'
    assert ctx['custom']['organization_id'] == 123


def test_flag_set_with_organization_context_env_override(monkeypatch, settings):
    # Ensure offline mode/env control for deterministic behavior
    settings.FEATURE_FLAGS_OFFLINE = True

    # Use env override path for flag resolution
    monkeypatch.setenv('fflag_feat_test_org_targeting', 'true')

    class Org:
        def __init__(self, id, email):
            self.id = id

            class CreatedBy:
                def __init__(self, email):
                    self.email = email

            self.created_by = CreatedBy(email)

    org = Org(42, 'owner@example.com')

    assert flag_set('fflag_feat_test_org_targeting', organization=org, override_system_default=False) is True

    # Unset env should fall back to override_system_default=False
    monkeypatch.delenv('fflag_feat_test_org_targeting', raising=False)
    assert flag_set('fflag_feat_test_org_targeting', organization=org, override_system_default=False) is False