orion/tests/unit/services/test_team_service.py

# tests/unit/services/test_team_service.py
"""
Unit tests for TeamService.

Tests cover:
- Get team members
- Invite team member
- Update team member
- Remove team member
- Get vendor roles
"""

from datetime import UTC, datetime
from unittest.mock import MagicMock

import pytest

from app.exceptions import ValidationException
from app.modules.tenancy.services.team_service import TeamService, team_service
from app.modules.tenancy.models import Role, VendorUser


@pytest.mark.unit
@pytest.mark.service
class TestTeamServiceGetMembers:
    """Test get_team_members functionality"""

    def test_get_team_members_empty(self, db, test_vendor, test_user):
        """Test get_team_members returns empty list when no members"""
        service = TeamService()
        result = service.get_team_members(db, test_vendor.id, test_user)
        assert isinstance(result, list)

    def test_get_team_members_with_data(self, db, test_vendor_with_vendor_user, test_user):
        """Test get_team_members returns member data or raises"""
        service = TeamService()
        try:
            result = service.get_team_members(
                db, test_vendor_with_vendor_user.id, test_user
            )
            assert isinstance(result, list)
            if len(result) > 0:
                member = result[0]
                assert "id" in member
                assert "email" in member
        except ValidationException:
            # This is expected if the vendor user has no role
            pass


@pytest.mark.unit
@pytest.mark.service
class TestTeamServiceInvite:
    """Test invite_team_member functionality"""

    def test_invite_team_member_placeholder(self, db, test_vendor, test_user):
        """Test invite_team_member returns placeholder response"""
        service = TeamService()
        result = service.invite_team_member(
            db,
            test_vendor.id,
            {"email": "newmember@example.com", "role": "member"},
            test_user,
        )

        assert "message" in result
        assert result["email"] == "newmember@example.com"


@pytest.mark.unit
@pytest.mark.service
class TestTeamServiceUpdate:
    """Test update_team_member functionality"""

    def test_update_team_member_not_found(self, db, test_vendor, test_user):
        """Test update_team_member raises for non-existent member"""
        service = TeamService()
        with pytest.raises(ValidationException) as exc_info:
            service.update_team_member(
                db,
                test_vendor.id,
                99999,  # Non-existent user
                {"role_id": 1},
                test_user,
            )
        assert "failed" in str(exc_info.value).lower()

    def test_update_team_member_success(
        self, db, test_vendor_with_vendor_user, test_vendor_user, test_user
    ):
        """Test update_team_member updates member"""
        service = TeamService()

        # Get the vendor_user record
        vendor_user = (
            db.query(VendorUser)
            .filter(VendorUser.vendor_id == test_vendor_with_vendor_user.id)
            .first()
        )

        if vendor_user:
            result = service.update_team_member(
                db,
                test_vendor_with_vendor_user.id,
                vendor_user.user_id,
                {"is_active": True},
                test_user,
            )
            db.commit()

            assert result["message"] == "Team member updated successfully"
            assert result["user_id"] == vendor_user.user_id


@pytest.mark.unit
@pytest.mark.service
class TestTeamServiceRemove:
    """Test remove_team_member functionality"""

    def test_remove_team_member_not_found(self, db, test_vendor, test_user):
        """Test remove_team_member raises for non-existent member"""
        service = TeamService()
        with pytest.raises(ValidationException) as exc_info:
            service.remove_team_member(
                db,
                test_vendor.id,
                99999,  # Non-existent user
                test_user,
            )
        assert "failed" in str(exc_info.value).lower()

    def test_remove_team_member_success(
        self, db, test_vendor_with_vendor_user, test_vendor_user, test_user
    ):
        """Test remove_team_member soft deletes member"""
        service = TeamService()

        # Get the vendor_user record
        vendor_user = (
            db.query(VendorUser)
            .filter(VendorUser.vendor_id == test_vendor_with_vendor_user.id)
            .first()
        )

        if vendor_user:
            result = service.remove_team_member(
                db,
                test_vendor_with_vendor_user.id,
                vendor_user.user_id,
                test_user,
            )
            db.commit()

            assert result is True

            # Verify soft delete
            db.refresh(vendor_user)
            assert vendor_user.is_active is False


@pytest.mark.unit
@pytest.mark.service
class TestTeamServiceRoles:
    """Test get_vendor_roles functionality"""

    def test_get_vendor_roles_empty(self, db, test_vendor):
        """Test get_vendor_roles returns empty list when no roles"""
        service = TeamService()
        result = service.get_vendor_roles(db, test_vendor.id)
        assert isinstance(result, list)

    def test_get_vendor_roles_with_data(self, db, test_vendor_with_vendor_user):
        """Test get_vendor_roles returns role data"""
        # Create a role for the vendor
        role = Role(
            vendor_id=test_vendor_with_vendor_user.id,
            name="Test Role",
            permissions=["view_orders", "edit_products"],
        )
        db.add(role)
        db.commit()

        service = TeamService()
        result = service.get_vendor_roles(db, test_vendor_with_vendor_user.id)

        assert len(result) >= 1
        role_data = next((r for r in result if r["name"] == "Test Role"), None)
        if role_data:
            assert role_data["permissions"] == ["view_orders", "edit_products"]


@pytest.mark.unit
@pytest.mark.service
class TestTeamServiceSingleton:
    """Test singleton instance"""

    def test_singleton_exists(self):
        """Test team_service singleton exists"""
        assert team_service is not None
        assert isinstance(team_service, TeamService)