replypier/backend/tests/repo/test_user_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.user import User
from app.repositories.user import UserRepository
from app.models.organization import Organization

@pytest.fixture
def user_repo(db):
    """Create a user repository instance"""
    return UserRepository(db)

def test_create_user(user_repo, test_organization_id):
    """Test creating a new user"""
    email = "test@example.com"
    password = "testpassword"
    full_name = "Test User"
    hashed_password = User.get_password_hash(password)

    user = user_repo.create_user(
        email=email,
        hashed_password=hashed_password,
        full_name=full_name,
        organization_id=test_organization_id
    )

    assert user.email == email
    assert user.full_name == full_name
    assert user.organization_id == test_organization_id
    assert user.verify_password(password)  # Verify the password works

def test_get_user_by_email(user_repo, test_organization_id):
    """Test retrieving a user by email"""
    # Create test user
    email = "test@example.com"
    hashed_password = User.get_password_hash("testpassword")
    user = user_repo.create_user(
        email=email,
        hashed_password=hashed_password,
        full_name="Test User",
        organization_id=test_organization_id
    )

    # Retrieve user
    retrieved_user = user_repo.get_user_by_email(email)
    assert retrieved_user.id == user.id
    assert retrieved_user.email == email

def test_get_user_by_id(user_repo, test_organization_id):
    """Test retrieving a user by ID"""
    # Create test user
    hashed_password = User.get_password_hash("testpassword")
    user = user_repo.create_user(
        email="test@example.com",
        hashed_password=hashed_password,
        full_name="Test User",
        organization_id=test_organization_id
    )

    # Retrieve user
    retrieved_user = user_repo.get_user(user.id)
    assert retrieved_user.id == user.id
    assert retrieved_user.email == user.email

def test_get_users_by_organization(user_repo, test_organization_id):
    """Test retrieving all users for an organization"""
    # Create multiple users
    hashed_password = User.get_password_hash("testpassword")
    user_repo.create_user(
        email="user1@example.com",
        hashed_password=hashed_password,
        full_name="User One",
        organization_id=test_organization_id
    )
    user_repo.create_user(
        email="user2@example.com",
        hashed_password=hashed_password,
        full_name="User Two",
        organization_id=test_organization_id
    )

    # Create another organization for testing
    other_org = Organization(
        name="Other Organization",
        domain="other.com",
        timezone="UTC"
    )
    user_repo.db.add(other_org)
    user_repo.db.commit()
    user_repo.db.refresh(other_org)

    # Create user in different organization
    user_repo.create_user(
        email="other@example.com",
        hashed_password=hashed_password,
        full_name="Other User",
        organization_id=other_org.id
    )

    # Get users for test organization
    users = user_repo.get_users_by_organization(test_organization_id)
    assert len(users) == 2
    assert all(u.organization_id == test_organization_id for u in users)

def test_update_user(user_repo, test_organization_id):
    """Test updating a user"""
    # Create test user
    hashed_password = User.get_password_hash("testpassword")
    user = user_repo.create_user(
        email="test@example.com",
        hashed_password=hashed_password,
        full_name="Test User",
        organization_id=test_organization_id
    )

    # Update user
    new_name = "Updated User"
    updated_user = user_repo.update_user(
        user_id=str(user.id),
        full_name=new_name
    )

    assert updated_user.full_name == new_name
    assert updated_user.id == user.id

def test_authenticate_user(user_repo, test_organization_id):
    """Test user authentication"""
    email = "test@example.com"
    password = "testpassword"
    hashed_password = User.get_password_hash(password)
    
    # Create test user
    user = user_repo.create_user(
        email=email,
        hashed_password=hashed_password,
        full_name="Test User",
        organization_id=test_organization_id
    )

    # Test valid password
    assert user.verify_password(password)

    # Test invalid password
    assert not user.verify_password("wrongpassword")

def test_get_multi(user_repo, test_organization_id):
    """Test retrieving multiple users with pagination"""
    # Create multiple users
    users = []
    hashed_password = User.get_password_hash("testpassword")
    for i in range(5):
        user = user_repo.create_user(
            email=f"user{i}@example.com",
            hashed_password=hashed_password,
            full_name=f"User {i}",
            organization_id=test_organization_id
        )
        users.append(user)

    # Test pagination
    first_page = user_repo.get_users_by_organization(test_organization_id)[:2]
    assert len(first_page) == 2

    second_page = user_repo.get_users_by_organization(test_organization_id)[2:4]
    assert len(second_page) == 2

    last_page = user_repo.get_users_by_organization(test_organization_id)[4:]
    assert len(last_page) == 1