replypier/backend/tests/repo/test_organization_repo.py

"""
Copyright 2024-2026 ReplyPier

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
"""

import pytest
from app.models.organization import Organization
from app.repositories.organization import OrganizationRepository

@pytest.fixture
def org_repo(db):
    """Create an organization repository instance"""
    return OrganizationRepository(db)

def test_create_organization(org_repo):
    """Test creating a new organization"""
    name = "Test Organization"
    domain = "test.com"
    timezone = "UTC"
    business_hours = {
        "monday": {"start": "09:00", "end": "17:00"},
        "tuesday": {"start": "09:00", "end": "17:00"}
    }

    org = org_repo.create_organization(
        name=name,
        domain=domain,
        timezone=timezone,
        business_hours=business_hours
    )

    assert org.name == name
    assert org.domain == domain
    assert org.timezone == timezone
    assert org.business_hours == business_hours
    assert org.is_active is True

def test_get_organization(org_repo):
    """Test retrieving an organization by ID"""
    # Create test organization
    org = org_repo.create_organization(
        name="Test Organization",
        domain="test.com",
        timezone="UTC"
    )

    # Retrieve organization
    retrieved_org = org_repo.get_organization(org.id)
    assert retrieved_org.id == org.id
    assert retrieved_org.name == org.name

def test_get_organization_by_domain(org_repo):
    """Test retrieving an organization by domain"""
    domain = "test.com"
    org = org_repo.create_organization(
        name="Test Organization",
        domain=domain,
        timezone="UTC"
    )

    retrieved_org = org_repo.get_organization_by_domain(domain)
    assert retrieved_org.id == org.id
    assert retrieved_org.domain == domain

def test_is_domain_taken(org_repo):
    """Domain clashes ignore case and the org being updated"""
    org = org_repo.create_organization(name="Test Organization", domain="test.com", timezone="UTC")

    assert org_repo.is_domain_taken("TEST.com") is True
    assert org_repo.is_domain_taken("test.com", exclude_org_id=org.id) is False
    assert org_repo.is_domain_taken("other.com") is False

def test_get_multi(org_repo):
    """Test retrieving multiple organizations with pagination"""
    # Create multiple organizations
    orgs = []
    for i in range(5):
        org = org_repo.create_organization(
            name=f"Organization {i}",
            domain=f"org{i}.com",
            timezone="UTC"
        )
        orgs.append(org)

    # Test pagination
    active_orgs = org_repo.get_active_organizations()
    assert len(active_orgs) == 5

def test_update_organization(org_repo):
    """Test updating an organization"""
    # Create test organization
    org = org_repo.create_organization(
        name="Test Organization",
        domain="test.com",
        timezone="UTC"
    )

    # Update organization
    new_name = "Updated Organization"
    new_settings = {"new_setting": "new_value"}
    updated_org = org_repo.update_organization(
        org_id=org.id,
        name=new_name,
        settings=new_settings
    )

    assert updated_org.name == new_name
    assert updated_org.settings == new_settings
    assert updated_org.id == org.id

def test_delete_organization(org_repo):
    """Test deactivating an organization"""
    # Create test organization
    org = org_repo.create_organization(
        name="Test Organization",
        domain="test.com",
        timezone="UTC"
    )

    # Deactivate organization
    success = org_repo.deactivate_organization(org.id)
    assert success is True

    # Verify organization is deactivated
    deactivated_org = org_repo.get_organization(org.id)
    assert deactivated_org is not None
    assert deactivated_org.is_active is False

def test_get_active_organizations(org_repo):
    """Test retrieving only active organizations"""
    # Create active and inactive organizations
    active_org = org_repo.create_organization(
        name="Active Organization",
        domain="active.com",
        timezone="UTC"
    )
    inactive_org = org_repo.create_organization(
        name="Inactive Organization",
        domain="inactive.com",
        timezone="UTC"
    )
    org_repo.deactivate_organization(inactive_org.id)

    # Get active organizations
    active_orgs = org_repo.get_active_organizations()
    assert len(active_orgs) == 1
    assert active_orgs[0].id == active_org.id
    assert active_orgs[0].is_active is True

def test_organization_with_business_hours(org_repo):
    """Test organization with business hours"""
    business_hours = {
        "monday": {"start": "09:00", "end": "17:00"},
        "tuesday": {"start": "09:00", "end": "17:00"},
        "wednesday": {"start": "09:00", "end": "17:00"},
        "thursday": {"start": "09:00", "end": "17:00"},
        "friday": {"start": "09:00", "end": "17:00"}
    }

    org = org_repo.create_organization(
        name="Test Organization",
        domain="test.com",
        timezone="UTC",
        business_hours=business_hours
    )

    retrieved_org = org_repo.get_organization(org.id)
    assert retrieved_org.business_hours == business_hours

def test_organization_with_settings(org_repo):
    """Test organization with custom settings"""
    settings = {
        "chat_timeout": 300,
        "max_sessions": 10,
        "welcome_message": "Welcome to our organization!"
    }

    org = org_repo.create_organization(
        name="Test Organization",
        domain="test.com",
        timezone="UTC"
    )

    # Update settings
    updated_org = org_repo.update_settings(org.id, settings)
    assert updated_org.settings == settings