# tests/fixtures/auth_fixtures.py """ Authentication-related test fixtures. Note: Fixtures should NOT use db.expunge() as it breaks lazy loading. See tests/conftest.py for details on fixture best practices. """ import uuid import pytest from middleware.auth import AuthManager from models.database.user import User @pytest.fixture(scope="session") def auth_manager(): """Create auth manager instance (session scope since it's stateless).""" return AuthManager() @pytest.fixture def test_user(db, auth_manager): """Create a test user with unique username.""" unique_id = str(uuid.uuid4())[:8] hashed_password = auth_manager.hash_password("testpass123") user = User( email=f"test_{unique_id}@example.com", username=f"testuser_{unique_id}", hashed_password=hashed_password, role="user", is_active=True, ) db.add(user) db.commit() db.refresh(user) return user @pytest.fixture def test_admin(db, auth_manager): """Create a test admin user with unique username.""" unique_id = str(uuid.uuid4())[:8] hashed_password = auth_manager.hash_password("adminpass123") admin = User( email=f"admin_{unique_id}@example.com", username=f"admin_{unique_id}", hashed_password=hashed_password, role="admin", is_active=True, ) db.add(admin) db.commit() db.refresh(admin) return admin @pytest.fixture def another_admin(db, auth_manager): """Create another test admin user for testing admin-to-admin interactions.""" unique_id = str(uuid.uuid4())[:8] hashed_password = auth_manager.hash_password("anotheradminpass123") admin = User( email=f"another_admin_{unique_id}@example.com", username=f"another_admin_{unique_id}", hashed_password=hashed_password, role="admin", is_active=True, ) db.add(admin) db.commit() db.refresh(admin) return admin @pytest.fixture def other_user(db, auth_manager): """Create a different user for testing access controls.""" unique_id = str(uuid.uuid4())[:8] hashed_password = auth_manager.hash_password("otherpass123") user = User( email=f"other_{unique_id}@example.com", username=f"otheruser_{unique_id}", hashed_password=hashed_password, role="user", is_active=True, ) db.add(user) db.commit() db.refresh(user) return user @pytest.fixture def auth_headers(test_user, auth_manager): """Get authentication headers for test user (non-admin). Uses direct JWT generation to avoid vendor context requirement of shop login. This is used for testing non-admin access to admin endpoints. """ token_data = auth_manager.create_access_token(user=test_user) return {"Authorization": f"Bearer {token_data['access_token']}"} @pytest.fixture def admin_headers(client, test_admin): """Get authentication headers for admin user""" response = client.post( "/api/v1/admin/auth/login", json={"email_or_username": test_admin.username, "password": "adminpass123"}, ) assert response.status_code == 200, f"Admin login failed: {response.text}" token = response.json()["access_token"] return {"Authorization": f"Bearer {token}"} @pytest.fixture def test_vendor_user(db, auth_manager): """Create a test vendor user with unique username.""" unique_id = str(uuid.uuid4())[:8] hashed_password = auth_manager.hash_password("vendorpass123") user = User( email=f"vendor_{unique_id}@example.com", username=f"vendoruser_{unique_id}", hashed_password=hashed_password, role="vendor", is_active=True, ) db.add(user) db.commit() db.refresh(user) return user @pytest.fixture def vendor_user_headers(client, test_vendor_user, test_vendor_with_vendor_user): """Get authentication headers for vendor user (uses get_current_vendor_api). Depends on test_vendor_with_vendor_user to ensure VendorUser association exists. """ response = client.post( "/api/v1/vendor/auth/login", json={ "email_or_username": test_vendor_user.username, "password": "vendorpass123", }, ) assert response.status_code == 200, f"Vendor login failed: {response.text}" token = response.json()["access_token"] return {"Authorization": f"Bearer {token}"}