orion/tests/fixtures/module_fixtures.py

# tests/fixtures/module_fixtures.py
"""
Module system test fixtures.

Provides fixtures for testing platform module enablement, configuration,
and access control.
"""

import uuid
from datetime import UTC, datetime

import pytest

from app.modules.tenancy.models import Platform, PlatformModule


@pytest.fixture
def platform_with_modules(db, test_super_admin):
    """Create a test platform with specific modules enabled."""
    unique_id = str(uuid.uuid4())[:8]
    platform = Platform(
        code=f"modtest_{unique_id}",
        name=f"Module Test Platform {unique_id}",
        description="A test platform with module configuration",
        path_prefix=f"modtest{unique_id}",
        is_active=True,
        is_public=True,
        default_language="en",
        supported_languages=["en", "fr"],
    )
    db.add(platform)
    db.flush()

    # Enable specific modules via junction table
    enabled_modules = ["billing", "inventory", "orders"]
    for module_code in enabled_modules:
        pm = PlatformModule(
            platform_id=platform.id,
            module_code=module_code,
            is_enabled=True,
            enabled_at=datetime.now(UTC),
            enabled_by_user_id=test_super_admin.id,
            config={},
        )
        db.add(pm)

    # Add a disabled module
    pm_disabled = PlatformModule(
        platform_id=platform.id,
        module_code="marketplace",
        is_enabled=False,
        disabled_at=datetime.now(UTC),
        disabled_by_user_id=test_super_admin.id,
        config={},
    )
    db.add(pm_disabled)

    db.commit()
    db.refresh(platform)
    return platform


@pytest.fixture
def platform_with_config(db, test_super_admin):
    """Create a test platform with module configuration."""
    unique_id = str(uuid.uuid4())[:8]
    platform = Platform(
        code=f"cfgtest_{unique_id}",
        name=f"Config Test Platform {unique_id}",
        description="A test platform with module config",
        path_prefix=f"cfgtest{unique_id}",
        is_active=True,
        is_public=True,
        default_language="en",
        supported_languages=["en"],
    )
    db.add(platform)
    db.flush()

    # Add module with configuration
    pm = PlatformModule(
        platform_id=platform.id,
        module_code="billing",
        is_enabled=True,
        enabled_at=datetime.now(UTC),
        enabled_by_user_id=test_super_admin.id,
        config={
            "stripe_mode": "test",
            "default_trial_days": 30,
            "allow_free_tier": True,
        },
    )
    db.add(pm)

    # Add inventory module with config
    pm_inv = PlatformModule(
        platform_id=platform.id,
        module_code="inventory",
        is_enabled=True,
        enabled_at=datetime.now(UTC),
        enabled_by_user_id=test_super_admin.id,
        config={
            "low_stock_threshold": 5,
            "enable_locations": True,
        },
    )
    db.add(pm_inv)

    db.commit()
    db.refresh(platform)
    return platform


@pytest.fixture
def platform_all_modules_disabled(db, test_super_admin):
    """Create a test platform with all optional modules disabled."""
    unique_id = str(uuid.uuid4())[:8]
    platform = Platform(
        code=f"nomod_{unique_id}",
        name=f"No Modules Platform {unique_id}",
        description="A test platform with minimal modules",
        path_prefix=f"nomod{unique_id}",
        is_active=True,
        is_public=True,
        default_language="en",
        supported_languages=["en"],
        settings={"enabled_modules": []},  # Legacy format for testing
    )
    db.add(platform)
    db.commit()
    db.refresh(platform)
    return platform


@pytest.fixture
def module_factory(db, test_super_admin):
    """Factory for creating PlatformModule records."""
    def _create_module(
        platform_id: int,
        module_code: str,
        is_enabled: bool = True,
        config: dict | None = None,
    ):
        pm = PlatformModule(
            platform_id=platform_id,
            module_code=module_code,
            is_enabled=is_enabled,
            enabled_at=datetime.now(UTC) if is_enabled else None,
            enabled_by_user_id=test_super_admin.id if is_enabled else None,
            disabled_at=None if is_enabled else datetime.now(UTC),
            disabled_by_user_id=None if is_enabled else test_super_admin.id,
            config=config or {},
        )
        db.add(pm)
        db.commit()
        db.refresh(pm)
        return pm
    return _create_module