mirror of
https://github.com/DRYTRIX/TimeTracker.git
synced 2026-01-06 11:40:52 -06:00
357 lines
14 KiB
Python
357 lines
14 KiB
Python
"""
|
|
Comprehensive tests for Client Portal feature.
|
|
|
|
This module tests:
|
|
- User model client portal fields and properties
|
|
- Client portal routes and access control
|
|
- Client portal data retrieval
|
|
- Admin interface for enabling/disabling portal access
|
|
"""
|
|
|
|
import pytest
|
|
from datetime import datetime, timedelta
|
|
from decimal import Decimal
|
|
from app.models import User, Client, Project, Invoice, InvoiceItem, TimeEntry
|
|
from app import db
|
|
|
|
|
|
# ============================================================================
|
|
# Model Tests
|
|
# ============================================================================
|
|
|
|
@pytest.mark.models
|
|
@pytest.mark.unit
|
|
class TestClientPortalUserModel:
|
|
"""Test User model client portal functionality"""
|
|
|
|
def test_user_client_portal_enabled_field(self, app, user):
|
|
"""Test client_portal_enabled field defaults to False"""
|
|
with app.app_context():
|
|
assert user.client_portal_enabled is False
|
|
|
|
def test_user_client_id_field(self, app, user):
|
|
"""Test client_id field defaults to None"""
|
|
with app.app_context():
|
|
assert user.client_id is None
|
|
|
|
def test_is_client_portal_user_property(self, app, user, test_client):
|
|
"""Test is_client_portal_user property"""
|
|
with app.app_context():
|
|
# Initially False
|
|
assert user.is_client_portal_user is False
|
|
|
|
# Enable portal but no client assigned
|
|
user.client_portal_enabled = True
|
|
assert user.is_client_portal_user is False
|
|
|
|
# Assign client
|
|
user.client_id = test_client.id
|
|
assert user.is_client_portal_user is True
|
|
|
|
def test_get_client_portal_data(self, app, user, test_client):
|
|
"""Test get_client_portal_data method"""
|
|
with app.app_context():
|
|
# No portal access
|
|
assert user.get_client_portal_data() is None
|
|
|
|
# Enable portal and assign client
|
|
user.client_portal_enabled = True
|
|
user.client_id = test_client.id
|
|
db.session.commit()
|
|
|
|
# Should return data structure
|
|
data = user.get_client_portal_data()
|
|
assert data is not None
|
|
assert 'client' in data
|
|
assert 'projects' in data
|
|
assert 'invoices' in data
|
|
assert 'time_entries' in data
|
|
assert data['client'].id == test_client.id
|
|
|
|
def test_get_client_portal_data_with_projects(self, app, user, test_client):
|
|
"""Test get_client_portal_data includes projects"""
|
|
with app.app_context():
|
|
user.client_portal_enabled = True
|
|
user.client_id = test_client.id
|
|
|
|
# Create projects
|
|
project1 = Project(name="Project 1", client_id=test_client.id, status='active')
|
|
project2 = Project(name="Project 2", client_id=test_client.id, status='active')
|
|
project3 = Project(name="Project 3", client_id=test_client.id, status='inactive')
|
|
db.session.add_all([project1, project2, project3])
|
|
db.session.commit()
|
|
|
|
data = user.get_client_portal_data()
|
|
assert len(data['projects']) == 2 # Only active projects
|
|
assert project1 in data['projects']
|
|
assert project2 in data['projects']
|
|
assert project3 not in data['projects']
|
|
|
|
def test_get_client_portal_data_with_invoices(self, app, user, test_client):
|
|
"""Test get_client_portal_data includes invoices"""
|
|
with app.app_context():
|
|
user.client_portal_enabled = True
|
|
user.client_id = test_client.id
|
|
db.session.commit()
|
|
|
|
# Handle potential session issues from audit logging
|
|
try:
|
|
db.session.rollback()
|
|
except Exception:
|
|
pass
|
|
|
|
project = Project(name="Test Project", client_id=test_client.id)
|
|
db.session.add(project)
|
|
db.session.flush() # Flush to get project.id without committing
|
|
project_id = project.id
|
|
|
|
# Create invoices
|
|
invoice1 = Invoice(
|
|
invoice_number="INV-001",
|
|
project_id=project_id,
|
|
client_name=test_client.name,
|
|
client_id=test_client.id,
|
|
due_date=datetime.utcnow().date() + timedelta(days=30),
|
|
created_by=user.id,
|
|
total_amount=Decimal('100.00')
|
|
)
|
|
invoice2 = Invoice(
|
|
invoice_number="INV-002",
|
|
project_id=project_id,
|
|
client_name=test_client.name,
|
|
client_id=test_client.id,
|
|
due_date=datetime.utcnow().date() + timedelta(days=30),
|
|
created_by=user.id,
|
|
total_amount=Decimal('200.00')
|
|
)
|
|
db.session.add_all([invoice1, invoice2])
|
|
db.session.commit()
|
|
|
|
data = user.get_client_portal_data()
|
|
assert len(data['invoices']) == 2
|
|
assert invoice1 in data['invoices']
|
|
assert invoice2 in data['invoices']
|
|
|
|
def test_get_client_portal_data_with_time_entries(self, app, user, test_client):
|
|
"""Test get_client_portal_data includes time entries"""
|
|
with app.app_context():
|
|
user.client_portal_enabled = True
|
|
user.client_id = test_client.id
|
|
|
|
project = Project(name="Test Project", client_id=test_client.id)
|
|
db.session.add(project)
|
|
db.session.commit()
|
|
|
|
# Create time entries
|
|
entry1 = TimeEntry(
|
|
user_id=user.id,
|
|
project_id=project.id,
|
|
start_time=datetime.utcnow() - timedelta(hours=2),
|
|
end_time=datetime.utcnow(),
|
|
duration_seconds=7200
|
|
)
|
|
entry2 = TimeEntry(
|
|
user_id=user.id,
|
|
project_id=project.id,
|
|
start_time=datetime.utcnow() - timedelta(hours=1),
|
|
end_time=datetime.utcnow(),
|
|
duration_seconds=3600
|
|
)
|
|
db.session.add_all([entry1, entry2])
|
|
db.session.commit()
|
|
|
|
data = user.get_client_portal_data()
|
|
assert len(data['time_entries']) == 2
|
|
assert entry1 in data['time_entries']
|
|
assert entry2 in data['time_entries']
|
|
|
|
|
|
# ============================================================================
|
|
# Route Tests
|
|
# ============================================================================
|
|
|
|
@pytest.mark.routes
|
|
@pytest.mark.unit
|
|
class TestClientPortalRoutes:
|
|
"""Test client portal routes"""
|
|
|
|
def test_client_portal_dashboard_requires_access(self, app, client, user):
|
|
"""Test dashboard requires client portal access"""
|
|
with app.app_context():
|
|
# Login user without portal access
|
|
with client.session_transaction() as sess:
|
|
sess['_user_id'] = str(user.id)
|
|
|
|
response = client.get('/client-portal/dashboard')
|
|
assert response.status_code == 403
|
|
|
|
def test_client_portal_dashboard_with_access(self, app, client, user, test_client):
|
|
"""Test dashboard accessible with portal access"""
|
|
with app.app_context():
|
|
user.client_portal_enabled = True
|
|
user.client_id = test_client.id
|
|
db.session.commit()
|
|
|
|
with client.session_transaction() as sess:
|
|
sess['_user_id'] = str(user.id)
|
|
|
|
response = client.get('/client-portal/dashboard')
|
|
assert response.status_code == 200
|
|
assert b'Client Portal' in response.data
|
|
|
|
def test_client_portal_projects_route(self, app, client, user, test_client):
|
|
"""Test projects route"""
|
|
with app.app_context():
|
|
user.client_portal_enabled = True
|
|
user.client_id = test_client.id
|
|
db.session.commit()
|
|
|
|
with client.session_transaction() as sess:
|
|
sess['_user_id'] = str(user.id)
|
|
|
|
response = client.get('/client-portal/projects')
|
|
assert response.status_code == 200
|
|
|
|
def test_client_portal_invoices_route(self, app, client, user, test_client):
|
|
"""Test invoices route"""
|
|
with app.app_context():
|
|
user.client_portal_enabled = True
|
|
user.client_id = test_client.id
|
|
db.session.commit()
|
|
|
|
with client.session_transaction() as sess:
|
|
sess['_user_id'] = str(user.id)
|
|
|
|
response = client.get('/client-portal/invoices')
|
|
assert response.status_code == 200
|
|
|
|
def test_client_portal_time_entries_route(self, app, client, user, test_client):
|
|
"""Test time entries route"""
|
|
with app.app_context():
|
|
user.client_portal_enabled = True
|
|
user.client_id = test_client.id
|
|
db.session.commit()
|
|
|
|
with client.session_transaction() as sess:
|
|
sess['_user_id'] = str(user.id)
|
|
|
|
response = client.get('/client-portal/time-entries')
|
|
assert response.status_code == 200
|
|
|
|
def test_view_invoice_belongs_to_client(self, app, client, user, test_client):
|
|
"""Test viewing invoice requires it belongs to user's client"""
|
|
with app.app_context():
|
|
user.client_portal_enabled = True
|
|
user.client_id = test_client.id
|
|
|
|
# Create another client
|
|
other_client = Client(name="Other Client")
|
|
db.session.add(other_client)
|
|
|
|
project = Project(name="Test Project", client_id=test_client.id)
|
|
db.session.add(project)
|
|
db.session.commit()
|
|
|
|
# Create invoice for user's client
|
|
invoice = Invoice(
|
|
invoice_number="INV-001",
|
|
project_id=project.id,
|
|
client_name=test_client.name,
|
|
client_id=test_client.id,
|
|
due_date=datetime.utcnow().date() + timedelta(days=30),
|
|
created_by=user.id,
|
|
total_amount=Decimal('100.00')
|
|
)
|
|
db.session.add(invoice)
|
|
db.session.commit()
|
|
|
|
with client.session_transaction() as sess:
|
|
sess['_user_id'] = str(user.id)
|
|
|
|
# Should be able to view invoice
|
|
response = client.get(f'/client-portal/invoices/{invoice.id}')
|
|
assert response.status_code == 200
|
|
|
|
|
|
# ============================================================================
|
|
# Admin Interface Tests
|
|
# ============================================================================
|
|
|
|
@pytest.mark.routes
|
|
@pytest.mark.unit
|
|
class TestAdminClientPortalManagement:
|
|
"""Test admin interface for managing client portal access"""
|
|
|
|
def test_admin_can_enable_client_portal(self, app, admin_authenticated_client, user, test_client):
|
|
"""Test admin can enable client portal for user"""
|
|
with app.app_context():
|
|
response = admin_authenticated_client.post(
|
|
f'/admin/users/{user.id}/edit',
|
|
data={
|
|
'username': user.username,
|
|
'role': user.role,
|
|
'is_active': 'on' if user.is_active else '',
|
|
'client_portal_enabled': 'on',
|
|
'client_id': str(test_client.id),
|
|
'csrf_token': 'test-csrf-token'
|
|
},
|
|
follow_redirects=True
|
|
)
|
|
# Should redirect to users list
|
|
assert response.status_code == 200
|
|
|
|
# Verify user was updated
|
|
updated_user = User.query.get(user.id)
|
|
assert updated_user.client_portal_enabled is True
|
|
assert updated_user.client_id == test_client.id
|
|
|
|
def test_admin_can_disable_client_portal(self, app, admin_authenticated_client, user, test_client):
|
|
"""Test admin can disable client portal for user"""
|
|
with app.app_context():
|
|
# Enable portal first
|
|
user.client_portal_enabled = True
|
|
user.client_id = test_client.id
|
|
db.session.commit()
|
|
|
|
response = admin_authenticated_client.post(
|
|
f'/admin/users/{user.id}/edit',
|
|
data={
|
|
'username': user.username,
|
|
'role': user.role,
|
|
'is_active': 'on' if user.is_active else '',
|
|
'client_portal_enabled': '', # Not checked
|
|
'client_id': '',
|
|
'csrf_token': 'test-csrf-token'
|
|
},
|
|
follow_redirects=True
|
|
)
|
|
|
|
# Verify user was updated
|
|
updated_user = User.query.get(user.id)
|
|
assert updated_user.client_portal_enabled is False
|
|
assert updated_user.client_id is None
|
|
|
|
|
|
# ============================================================================
|
|
# Smoke Tests
|
|
# ============================================================================
|
|
|
|
@pytest.mark.smoke
|
|
@pytest.mark.unit
|
|
def test_client_portal_smoke(app, user, test_client):
|
|
"""Smoke test for client portal basic functionality"""
|
|
with app.app_context():
|
|
# Enable portal
|
|
user.client_portal_enabled = True
|
|
user.client_id = test_client.id
|
|
db.session.commit()
|
|
|
|
# Verify properties
|
|
assert user.is_client_portal_user is True
|
|
|
|
# Get portal data
|
|
data = user.get_client_portal_data()
|
|
assert data is not None
|
|
assert data['client'] == test_client
|
|
|