mirror of
https://github.com/DRYTRIX/TimeTracker.git
synced 2026-02-09 05:38:55 -06:00
feat: add overtime tracking support with configurable working hours
Implement comprehensive overtime tracking feature that allows users to set their standard working hours per day and automatically calculates overtime for hours worked beyond that threshold. Core Features: - Add standard_hours_per_day field to User model (default: 8.0 hours) - Create Alembic migration (031_add_standard_hours_per_day.py) - Implement overtime calculation utilities (app/utils/overtime.py) * calculate_daily_overtime: per-day overtime calculation * calculate_period_overtime: multi-day overtime aggregation * get_daily_breakdown: detailed day-by-day analysis * get_weekly_overtime_summary: weekly overtime statistics * get_overtime_statistics: comprehensive overtime metrics User Interface: - Add "Overtime Settings" section to user settings page - Display overtime data in user reports (regular vs overtime hours) - Show "Days with Overtime" badge in reports - Add overtime analytics API endpoint (/api/analytics/overtime) - Improve input field styling with cleaner appearance (no spinners) Reports Enhancement: - Standardize form input styling across all report pages - Replace inline Tailwind classes with consistent form-input class - Add FontAwesome icons to form labels for better UX - Improve button hover states and transitions Testing: - Add comprehensive unit tests (tests/test_overtime.py) - Add smoke tests for quick validation (tests/test_overtime_smoke.py) - Test coverage for models, utilities, and various overtime scenarios Documentation: - OVERTIME_FEATURE_DOCUMENTATION.md: complete feature guide - OVERTIME_IMPLEMENTATION_SUMMARY.md: technical implementation details - docs/features/OVERTIME_TRACKING.md: quick start guide This change enables organizations to track employee overtime accurately based on individual working hour configurations, providing better insights into work patterns and resource allocation.
This commit is contained in:
458
tests/test_overtime.py
Normal file
458
tests/test_overtime.py
Normal file
@@ -0,0 +1,458 @@
|
||||
"""
|
||||
Tests for overtime calculation functionality
|
||||
"""
|
||||
|
||||
import pytest
|
||||
from datetime import datetime, timedelta, date
|
||||
from app import db
|
||||
from app.models import User, TimeEntry, Project, Client
|
||||
from app.utils.overtime import (
|
||||
calculate_daily_overtime,
|
||||
calculate_period_overtime,
|
||||
get_daily_breakdown,
|
||||
get_weekly_overtime_summary,
|
||||
get_overtime_statistics
|
||||
)
|
||||
|
||||
|
||||
class TestOvertimeCalculations:
|
||||
"""Test suite for overtime calculation utilities"""
|
||||
|
||||
def test_calculate_daily_overtime_no_overtime(self):
|
||||
"""Test that no overtime is calculated when hours are below standard"""
|
||||
result = calculate_daily_overtime(6.0, 8.0)
|
||||
assert result == 0.0
|
||||
|
||||
def test_calculate_daily_overtime_exact_standard(self):
|
||||
"""Test that no overtime is calculated when hours equal standard"""
|
||||
result = calculate_daily_overtime(8.0, 8.0)
|
||||
assert result == 0.0
|
||||
|
||||
def test_calculate_daily_overtime_with_overtime(self):
|
||||
"""Test overtime calculation when hours exceed standard"""
|
||||
result = calculate_daily_overtime(10.0, 8.0)
|
||||
assert result == 2.0
|
||||
|
||||
def test_calculate_daily_overtime_large_overtime(self):
|
||||
"""Test overtime calculation with significant overtime"""
|
||||
result = calculate_daily_overtime(14.5, 8.0)
|
||||
assert result == 6.5
|
||||
|
||||
|
||||
class TestPeriodOvertime:
|
||||
"""Test suite for period-based overtime calculations"""
|
||||
|
||||
@pytest.fixture
|
||||
def test_user(self, app):
|
||||
"""Create a test user with 8 hour standard day"""
|
||||
user = User(username='test_user_ot', role='user')
|
||||
user.standard_hours_per_day = 8.0
|
||||
db.session.add(user)
|
||||
db.session.commit()
|
||||
return user
|
||||
|
||||
@pytest.fixture
|
||||
def test_client_obj(self, app):
|
||||
"""Create a test client"""
|
||||
test_client = Client(name='Test Client OT')
|
||||
db.session.add(test_client)
|
||||
db.session.commit()
|
||||
return test_client
|
||||
|
||||
@pytest.fixture
|
||||
def test_project(self, app, test_client_obj):
|
||||
"""Create a test project"""
|
||||
project = Project(
|
||||
name='Test Project OT',
|
||||
client_id=test_client_obj.id
|
||||
)
|
||||
db.session.add(project)
|
||||
db.session.commit()
|
||||
return project
|
||||
|
||||
def test_period_overtime_no_entries(self, app, test_user):
|
||||
"""Test period overtime calculation with no time entries"""
|
||||
start_date = date.today() - timedelta(days=7)
|
||||
end_date = date.today()
|
||||
|
||||
result = calculate_period_overtime(test_user, start_date, end_date)
|
||||
|
||||
assert result['regular_hours'] == 0.0
|
||||
assert result['overtime_hours'] == 0.0
|
||||
assert result['total_hours'] == 0.0
|
||||
assert result['days_with_overtime'] == 0
|
||||
|
||||
def test_period_overtime_all_regular(self, app, test_user, test_project):
|
||||
"""Test period with all regular hours (no overtime)"""
|
||||
start_date = date.today() - timedelta(days=2)
|
||||
|
||||
# Create entries for 2 days with 7 hours each (below standard 8)
|
||||
for i in range(2):
|
||||
entry_date = start_date + timedelta(days=i)
|
||||
entry_start = datetime.combine(entry_date, datetime.min.time().replace(hour=9))
|
||||
entry_end = entry_start + timedelta(hours=7)
|
||||
|
||||
entry = TimeEntry(
|
||||
user_id=test_user.id,
|
||||
project_id=test_project.id,
|
||||
start_time=entry_start,
|
||||
end_time=entry_end,
|
||||
notes='Regular work'
|
||||
)
|
||||
db.session.add(entry)
|
||||
|
||||
db.session.commit()
|
||||
|
||||
result = calculate_period_overtime(test_user, start_date, date.today())
|
||||
|
||||
assert result['regular_hours'] == 14.0
|
||||
assert result['overtime_hours'] == 0.0
|
||||
assert result['total_hours'] == 14.0
|
||||
assert result['days_with_overtime'] == 0
|
||||
|
||||
def test_period_overtime_with_overtime(self, app, test_user, test_project):
|
||||
"""Test period with overtime hours"""
|
||||
start_date = date.today() - timedelta(days=2)
|
||||
|
||||
# Day 1: 10 hours (2 hours overtime)
|
||||
entry_date = start_date
|
||||
entry_start = datetime.combine(entry_date, datetime.min.time().replace(hour=9))
|
||||
entry_end = entry_start + timedelta(hours=10)
|
||||
|
||||
entry1 = TimeEntry(
|
||||
user_id=test_user.id,
|
||||
project_id=test_project.id,
|
||||
start_time=entry_start,
|
||||
end_time=entry_end,
|
||||
notes='Long day'
|
||||
)
|
||||
db.session.add(entry1)
|
||||
|
||||
# Day 2: 6 hours (no overtime)
|
||||
entry_date2 = start_date + timedelta(days=1)
|
||||
entry_start2 = datetime.combine(entry_date2, datetime.min.time().replace(hour=9))
|
||||
entry_end2 = entry_start2 + timedelta(hours=6)
|
||||
|
||||
entry2 = TimeEntry(
|
||||
user_id=test_user.id,
|
||||
project_id=test_project.id,
|
||||
start_time=entry_start2,
|
||||
end_time=entry_end2,
|
||||
notes='Short day'
|
||||
)
|
||||
db.session.add(entry2)
|
||||
|
||||
db.session.commit()
|
||||
|
||||
result = calculate_period_overtime(test_user, start_date, date.today())
|
||||
|
||||
assert result['regular_hours'] == 14.0 # 8 + 6
|
||||
assert result['overtime_hours'] == 2.0
|
||||
assert result['total_hours'] == 16.0
|
||||
assert result['days_with_overtime'] == 1
|
||||
|
||||
def test_period_overtime_multiple_entries_same_day(self, app, test_user, test_project):
|
||||
"""Test overtime calculation with multiple entries on the same day"""
|
||||
entry_date = date.today()
|
||||
|
||||
# Create 3 entries totaling 10 hours (2 hours overtime)
|
||||
for i, hours in enumerate([4, 3, 3]):
|
||||
entry_start = datetime.combine(entry_date, datetime.min.time().replace(hour=9 + i * 3))
|
||||
entry_end = entry_start + timedelta(hours=hours)
|
||||
|
||||
entry = TimeEntry(
|
||||
user_id=test_user.id,
|
||||
project_id=test_project.id,
|
||||
start_time=entry_start,
|
||||
end_time=entry_end,
|
||||
notes=f'Entry {i+1}'
|
||||
)
|
||||
db.session.add(entry)
|
||||
|
||||
db.session.commit()
|
||||
|
||||
result = calculate_period_overtime(test_user, entry_date, entry_date)
|
||||
|
||||
assert result['regular_hours'] == 8.0
|
||||
assert result['overtime_hours'] == 2.0
|
||||
assert result['total_hours'] == 10.0
|
||||
assert result['days_with_overtime'] == 1
|
||||
|
||||
|
||||
class TestDailyBreakdown:
|
||||
"""Test suite for daily overtime breakdown"""
|
||||
|
||||
@pytest.fixture
|
||||
def test_user_daily(self, app):
|
||||
"""Create a test user"""
|
||||
user = User(username='test_user_daily', role='user')
|
||||
user.standard_hours_per_day = 8.0
|
||||
db.session.add(user)
|
||||
db.session.commit()
|
||||
return user
|
||||
|
||||
@pytest.fixture
|
||||
def test_project_daily(self, app, test_client_obj):
|
||||
"""Create a test project"""
|
||||
project = Project(
|
||||
name='Test Project Daily',
|
||||
client_id=test_client_obj.id
|
||||
)
|
||||
db.session.add(project)
|
||||
db.session.commit()
|
||||
return project
|
||||
|
||||
@pytest.fixture
|
||||
def test_client_obj(self, app):
|
||||
"""Create a test client"""
|
||||
test_client = Client(name='Test Client Daily')
|
||||
db.session.add(test_client)
|
||||
db.session.commit()
|
||||
return test_client
|
||||
|
||||
def test_daily_breakdown_empty(self, app, test_user_daily):
|
||||
"""Test daily breakdown with no entries"""
|
||||
start_date = date.today() - timedelta(days=7)
|
||||
end_date = date.today()
|
||||
|
||||
result = get_daily_breakdown(test_user_daily, start_date, end_date)
|
||||
|
||||
assert len(result) == 0
|
||||
|
||||
def test_daily_breakdown_with_entries(self, app, test_user_daily, test_project_daily):
|
||||
"""Test daily breakdown with various entries"""
|
||||
start_date = date.today() - timedelta(days=2)
|
||||
|
||||
# Day 1: 9 hours (1 hour overtime)
|
||||
entry1_start = datetime.combine(start_date, datetime.min.time().replace(hour=9))
|
||||
entry1_end = entry1_start + timedelta(hours=9)
|
||||
entry1 = TimeEntry(
|
||||
user_id=test_user_daily.id,
|
||||
project_id=test_project_daily.id,
|
||||
start_time=entry1_start,
|
||||
end_time=entry1_end
|
||||
)
|
||||
db.session.add(entry1)
|
||||
|
||||
# Day 2: 6 hours (no overtime)
|
||||
entry2_start = datetime.combine(start_date + timedelta(days=1), datetime.min.time().replace(hour=9))
|
||||
entry2_end = entry2_start + timedelta(hours=6)
|
||||
entry2 = TimeEntry(
|
||||
user_id=test_user_daily.id,
|
||||
project_id=test_project_daily.id,
|
||||
start_time=entry2_start,
|
||||
end_time=entry2_end
|
||||
)
|
||||
db.session.add(entry2)
|
||||
|
||||
db.session.commit()
|
||||
|
||||
result = get_daily_breakdown(test_user_daily, start_date, date.today())
|
||||
|
||||
assert len(result) == 2
|
||||
|
||||
# Check day 1
|
||||
day1 = result[0]
|
||||
assert day1['total_hours'] == 9.0
|
||||
assert day1['regular_hours'] == 8.0
|
||||
assert day1['overtime_hours'] == 1.0
|
||||
assert day1['is_overtime'] is True
|
||||
|
||||
# Check day 2
|
||||
day2 = result[1]
|
||||
assert day2['total_hours'] == 6.0
|
||||
assert day2['regular_hours'] == 6.0
|
||||
assert day2['overtime_hours'] == 0.0
|
||||
assert day2['is_overtime'] is False
|
||||
|
||||
|
||||
class TestOvertimeStatistics:
|
||||
"""Test suite for comprehensive overtime statistics"""
|
||||
|
||||
@pytest.fixture
|
||||
def test_user_stats(self, app):
|
||||
"""Create a test user"""
|
||||
user = User(username='test_user_stats', role='user')
|
||||
user.standard_hours_per_day = 8.0
|
||||
db.session.add(user)
|
||||
db.session.commit()
|
||||
return user
|
||||
|
||||
@pytest.fixture
|
||||
def test_project_stats(self, app, test_client_obj):
|
||||
"""Create a test project"""
|
||||
project = Project(
|
||||
name='Test Project Stats',
|
||||
client_id=test_client_obj.id
|
||||
)
|
||||
db.session.add(project)
|
||||
db.session.commit()
|
||||
return project
|
||||
|
||||
@pytest.fixture
|
||||
def test_client_obj(self, app):
|
||||
"""Create a test client"""
|
||||
test_client = Client(name='Test Client Stats')
|
||||
db.session.add(test_client)
|
||||
db.session.commit()
|
||||
return test_client
|
||||
|
||||
def test_overtime_statistics_comprehensive(self, app, test_user_stats, test_project_stats):
|
||||
"""Test comprehensive overtime statistics"""
|
||||
start_date = date.today() - timedelta(days=4)
|
||||
|
||||
# Create entries for multiple days with varying hours
|
||||
hours_per_day = [10, 7, 9, 6, 11] # 5 days
|
||||
|
||||
for i, hours in enumerate(hours_per_day):
|
||||
entry_date = start_date + timedelta(days=i)
|
||||
entry_start = datetime.combine(entry_date, datetime.min.time().replace(hour=9))
|
||||
entry_end = entry_start + timedelta(hours=hours)
|
||||
|
||||
entry = TimeEntry(
|
||||
user_id=test_user_stats.id,
|
||||
project_id=test_project_stats.id,
|
||||
start_time=entry_start,
|
||||
end_time=entry_end
|
||||
)
|
||||
db.session.add(entry)
|
||||
|
||||
db.session.commit()
|
||||
|
||||
result = get_overtime_statistics(test_user_stats, start_date, date.today())
|
||||
|
||||
# Verify structure
|
||||
assert 'period' in result
|
||||
assert 'hours' in result
|
||||
assert 'days_statistics' in result
|
||||
assert 'averages' in result
|
||||
assert 'max_overtime' in result
|
||||
|
||||
# Verify calculations
|
||||
# Total hours: 10 + 7 + 9 + 6 + 11 = 43
|
||||
# Days with overtime: 10 (2 OT), 9 (1 OT), 11 (3 OT) = 3 days
|
||||
# Total overtime: 2 + 1 + 3 = 6 hours
|
||||
# Regular: 43 - 6 = 37 hours
|
||||
|
||||
assert result['hours']['total_hours'] == 43.0
|
||||
assert result['hours']['overtime_hours'] == 6.0
|
||||
assert result['hours']['regular_hours'] == 37.0
|
||||
assert result['days_statistics']['days_worked'] == 5
|
||||
assert result['days_statistics']['days_with_overtime'] == 3
|
||||
|
||||
# Max overtime should be 3 hours (from the 11-hour day)
|
||||
assert result['max_overtime']['hours'] == 3.0
|
||||
|
||||
|
||||
class TestUserModel:
|
||||
"""Test suite for User model overtime-related functionality"""
|
||||
|
||||
def test_user_has_standard_hours_field(self, app):
|
||||
"""Test that User model has standard_hours_per_day field"""
|
||||
user = User(username='test_user_field', role='user')
|
||||
db.session.add(user)
|
||||
db.session.commit()
|
||||
|
||||
# Check that field exists and has default value
|
||||
assert hasattr(user, 'standard_hours_per_day')
|
||||
assert user.standard_hours_per_day == 8.0
|
||||
|
||||
def test_user_can_set_custom_standard_hours(self, app):
|
||||
"""Test that standard hours can be customized"""
|
||||
user = User(username='test_user_custom', role='user')
|
||||
user.standard_hours_per_day = 7.5
|
||||
db.session.add(user)
|
||||
db.session.commit()
|
||||
|
||||
# Reload from database
|
||||
user_reloaded = User.query.filter_by(username='test_user_custom').first()
|
||||
assert user_reloaded.standard_hours_per_day == 7.5
|
||||
|
||||
def test_user_standard_hours_validation_min(self, app):
|
||||
"""Test that standard hours can be set to minimum value"""
|
||||
user = User(username='test_user_min', role='user')
|
||||
user.standard_hours_per_day = 0.5
|
||||
db.session.add(user)
|
||||
db.session.commit()
|
||||
|
||||
assert user.standard_hours_per_day == 0.5
|
||||
|
||||
def test_user_standard_hours_validation_max(self, app):
|
||||
"""Test that standard hours can be set to maximum value"""
|
||||
user = User(username='test_user_max', role='user')
|
||||
user.standard_hours_per_day = 24.0
|
||||
db.session.add(user)
|
||||
db.session.commit()
|
||||
|
||||
assert user.standard_hours_per_day == 24.0
|
||||
|
||||
|
||||
class TestWeeklyOvertimeSummary:
|
||||
"""Test suite for weekly overtime summaries"""
|
||||
|
||||
@pytest.fixture
|
||||
def test_user_weekly(self, app):
|
||||
"""Create a test user"""
|
||||
user = User(username='test_user_weekly', role='user')
|
||||
user.standard_hours_per_day = 8.0
|
||||
db.session.add(user)
|
||||
db.session.commit()
|
||||
return user
|
||||
|
||||
@pytest.fixture
|
||||
def test_project_weekly(self, app, test_client_obj):
|
||||
"""Create a test project"""
|
||||
project = Project(
|
||||
name='Test Project Weekly',
|
||||
client_id=test_client_obj.id
|
||||
)
|
||||
db.session.add(project)
|
||||
db.session.commit()
|
||||
return project
|
||||
|
||||
@pytest.fixture
|
||||
def test_client_obj(self, app):
|
||||
"""Create a test client"""
|
||||
test_client = Client(name='Test Client Weekly')
|
||||
db.session.add(test_client)
|
||||
db.session.commit()
|
||||
return test_client
|
||||
|
||||
def test_weekly_summary_empty(self, app, test_user_weekly):
|
||||
"""Test weekly summary with no entries"""
|
||||
result = get_weekly_overtime_summary(test_user_weekly, weeks=2)
|
||||
assert len(result) == 0
|
||||
|
||||
def test_weekly_summary_with_data(self, app, test_user_weekly, test_project_weekly):
|
||||
"""Test weekly summary with entries across multiple weeks"""
|
||||
# Create entries for the past 2 weeks
|
||||
for week in range(2):
|
||||
for day in range(5): # 5 working days
|
||||
entry_date = date.today() - timedelta(weeks=1-week, days=day)
|
||||
entry_start = datetime.combine(entry_date, datetime.min.time().replace(hour=9))
|
||||
entry_end = entry_start + timedelta(hours=9) # 9 hours per day (1 hour OT)
|
||||
|
||||
entry = TimeEntry(
|
||||
user_id=test_user_weekly.id,
|
||||
project_id=test_project_weekly.id,
|
||||
start_time=entry_start,
|
||||
end_time=entry_end
|
||||
)
|
||||
db.session.add(entry)
|
||||
|
||||
db.session.commit()
|
||||
|
||||
result = get_weekly_overtime_summary(test_user_weekly, weeks=2)
|
||||
|
||||
# Should have data for weeks with entries
|
||||
assert len(result) > 0
|
||||
|
||||
# Each week should have proper structure
|
||||
for week_data in result:
|
||||
assert 'week_start' in week_data
|
||||
assert 'week_end' in week_data
|
||||
assert 'regular_hours' in week_data
|
||||
assert 'overtime_hours' in week_data
|
||||
assert 'total_hours' in week_data
|
||||
assert 'days_worked' in week_data
|
||||
|
||||
268
tests/test_overtime_smoke.py
Normal file
268
tests/test_overtime_smoke.py
Normal file
@@ -0,0 +1,268 @@
|
||||
"""
|
||||
Smoke tests for overtime feature
|
||||
Quick tests to verify basic overtime functionality is working
|
||||
"""
|
||||
|
||||
import pytest
|
||||
from datetime import datetime, timedelta, date
|
||||
from app import db
|
||||
from app.models import User, TimeEntry, Project, Client
|
||||
from app.utils.overtime import calculate_daily_overtime, calculate_period_overtime
|
||||
|
||||
|
||||
class TestOvertimeSmoke:
|
||||
"""Smoke tests for overtime feature"""
|
||||
|
||||
def test_overtime_utils_import(self):
|
||||
"""Smoke test: verify overtime utilities can be imported"""
|
||||
from app.utils import overtime
|
||||
assert hasattr(overtime, 'calculate_daily_overtime')
|
||||
assert hasattr(overtime, 'calculate_period_overtime')
|
||||
assert hasattr(overtime, 'get_daily_breakdown')
|
||||
assert hasattr(overtime, 'get_weekly_overtime_summary')
|
||||
assert hasattr(overtime, 'get_overtime_statistics')
|
||||
|
||||
def test_user_model_has_standard_hours(self, app):
|
||||
"""Smoke test: verify User model has standard_hours_per_day field"""
|
||||
user = User(username='smoke_test_user', role='user')
|
||||
assert hasattr(user, 'standard_hours_per_day')
|
||||
assert user.standard_hours_per_day == 8.0 # Default value
|
||||
|
||||
def test_basic_overtime_calculation(self):
|
||||
"""Smoke test: verify basic overtime calculation works"""
|
||||
# 10 hours worked with 8 hour standard = 2 hours overtime
|
||||
overtime = calculate_daily_overtime(10.0, 8.0)
|
||||
assert overtime == 2.0
|
||||
|
||||
def test_no_overtime_calculation(self):
|
||||
"""Smoke test: verify no overtime when under standard hours"""
|
||||
overtime = calculate_daily_overtime(6.0, 8.0)
|
||||
assert overtime == 0.0
|
||||
|
||||
def test_period_overtime_basic(self, app):
|
||||
"""Smoke test: verify period overtime calculation doesn't crash"""
|
||||
# Create a test user
|
||||
user = User(username='smoke_period_user', role='user')
|
||||
user.standard_hours_per_day = 8.0
|
||||
db.session.add(user)
|
||||
db.session.commit()
|
||||
|
||||
# Calculate overtime for a period with no entries
|
||||
start_date = date.today() - timedelta(days=7)
|
||||
end_date = date.today()
|
||||
|
||||
result = calculate_period_overtime(user, start_date, end_date)
|
||||
|
||||
# Should return valid structure even with no data
|
||||
assert 'regular_hours' in result
|
||||
assert 'overtime_hours' in result
|
||||
assert 'total_hours' in result
|
||||
assert 'days_with_overtime' in result
|
||||
assert result['overtime_hours'] == 0.0
|
||||
|
||||
def test_settings_route_accessible(self, app):
|
||||
"""Smoke test: verify settings page is accessible"""
|
||||
from app.routes.user import settings
|
||||
# Just verify the route exists and is importable
|
||||
assert settings is not None
|
||||
|
||||
def test_user_report_route_exists(self, app):
|
||||
"""Smoke test: verify user report route exists"""
|
||||
from app.routes.reports import user_report
|
||||
assert user_report is not None
|
||||
|
||||
def test_analytics_overtime_route_exists(self, app):
|
||||
"""Smoke test: verify analytics overtime route exists"""
|
||||
from app.routes.analytics import overtime_analytics
|
||||
assert overtime_analytics is not None
|
||||
|
||||
def test_overtime_calculation_with_real_entry(self, app):
|
||||
"""Smoke test: verify overtime calculation with a real time entry"""
|
||||
# Create test data
|
||||
user = User(username='smoke_entry_user', role='user')
|
||||
user.standard_hours_per_day = 8.0
|
||||
db.session.add(user)
|
||||
|
||||
client_obj = Client(name='Smoke Test Client')
|
||||
db.session.add(client_obj)
|
||||
db.session.commit()
|
||||
|
||||
project = Project(name='Smoke Test Project', client_id=client_obj.id)
|
||||
db.session.add(project)
|
||||
db.session.commit()
|
||||
|
||||
# Create a 10-hour time entry (should result in 2 hours overtime)
|
||||
entry_date = date.today()
|
||||
entry_start = datetime.combine(entry_date, datetime.min.time().replace(hour=9))
|
||||
entry_end = entry_start + timedelta(hours=10)
|
||||
|
||||
entry = TimeEntry(
|
||||
user_id=user.id,
|
||||
project_id=project.id,
|
||||
start_time=entry_start,
|
||||
end_time=entry_end,
|
||||
notes='Smoke test entry'
|
||||
)
|
||||
db.session.add(entry)
|
||||
db.session.commit()
|
||||
|
||||
# Calculate overtime
|
||||
result = calculate_period_overtime(user, entry_date, entry_date)
|
||||
|
||||
assert result['total_hours'] == 10.0
|
||||
assert result['regular_hours'] == 8.0
|
||||
assert result['overtime_hours'] == 2.0
|
||||
assert result['days_with_overtime'] == 1
|
||||
|
||||
def test_migration_file_exists(self):
|
||||
"""Smoke test: verify migration file exists"""
|
||||
import os
|
||||
migration_path = 'migrations/versions/031_add_standard_hours_per_day.py'
|
||||
assert os.path.exists(migration_path), f"Migration file not found: {migration_path}"
|
||||
|
||||
def test_overtime_template_fields(self, app):
|
||||
"""Smoke test: verify settings template has overtime field"""
|
||||
import os
|
||||
template_path = 'app/templates/user/settings.html'
|
||||
assert os.path.exists(template_path)
|
||||
|
||||
with open(template_path, 'r', encoding='utf-8') as f:
|
||||
content = f.read()
|
||||
assert 'standard_hours_per_day' in content, "Settings template missing overtime field"
|
||||
assert 'Overtime Settings' in content, "Settings template missing overtime section"
|
||||
|
||||
|
||||
class TestOvertimeIntegration:
|
||||
"""Integration tests for overtime feature"""
|
||||
|
||||
def test_full_overtime_workflow(self, app):
|
||||
"""Integration test: full overtime calculation workflow"""
|
||||
# 1. Create user with custom standard hours
|
||||
user = User(username='integration_user', role='user')
|
||||
user.standard_hours_per_day = 7.5 # 7.5 hour workday
|
||||
db.session.add(user)
|
||||
|
||||
# 2. Create client and project
|
||||
client_obj = Client(name='Integration Client')
|
||||
db.session.add(client_obj)
|
||||
db.session.commit()
|
||||
|
||||
project = Project(name='Integration Project', client_id=client_obj.id)
|
||||
db.session.add(project)
|
||||
db.session.commit()
|
||||
|
||||
# 3. Create time entries over multiple days
|
||||
start_date = date.today() - timedelta(days=4)
|
||||
|
||||
# Day 1: 9 hours (1.5 hours overtime)
|
||||
entry1_start = datetime.combine(start_date, datetime.min.time().replace(hour=9))
|
||||
entry1_end = entry1_start + timedelta(hours=9)
|
||||
entry1 = TimeEntry(
|
||||
user_id=user.id,
|
||||
project_id=project.id,
|
||||
start_time=entry1_start,
|
||||
end_time=entry1_end
|
||||
)
|
||||
db.session.add(entry1)
|
||||
|
||||
# Day 2: 7 hours (no overtime)
|
||||
entry2_start = datetime.combine(start_date + timedelta(days=1), datetime.min.time().replace(hour=9))
|
||||
entry2_end = entry2_start + timedelta(hours=7)
|
||||
entry2 = TimeEntry(
|
||||
user_id=user.id,
|
||||
project_id=project.id,
|
||||
start_time=entry2_start,
|
||||
end_time=entry2_end
|
||||
)
|
||||
db.session.add(entry2)
|
||||
|
||||
# Day 3: 10 hours (2.5 hours overtime)
|
||||
entry3_start = datetime.combine(start_date + timedelta(days=2), datetime.min.time().replace(hour=9))
|
||||
entry3_end = entry3_start + timedelta(hours=10)
|
||||
entry3 = TimeEntry(
|
||||
user_id=user.id,
|
||||
project_id=project.id,
|
||||
start_time=entry3_start,
|
||||
end_time=entry3_end
|
||||
)
|
||||
db.session.add(entry3)
|
||||
|
||||
db.session.commit()
|
||||
|
||||
# 4. Calculate period overtime
|
||||
result = calculate_period_overtime(user, start_date, date.today())
|
||||
|
||||
# 5. Verify results
|
||||
# Total: 9 + 7 + 10 = 26 hours
|
||||
# Overtime: 1.5 + 0 + 2.5 = 4 hours
|
||||
# Regular: 26 - 4 = 22 hours
|
||||
assert result['total_hours'] == 26.0
|
||||
assert result['overtime_hours'] == 4.0
|
||||
assert result['regular_hours'] == 22.0
|
||||
assert result['days_with_overtime'] == 2
|
||||
|
||||
# 6. Verify daily breakdown
|
||||
from app.utils.overtime import get_daily_breakdown
|
||||
breakdown = get_daily_breakdown(user, start_date, date.today())
|
||||
|
||||
assert len(breakdown) == 3
|
||||
assert breakdown[0]['overtime_hours'] == 1.5 # Day 1
|
||||
assert breakdown[1]['overtime_hours'] == 0.0 # Day 2
|
||||
assert breakdown[2]['overtime_hours'] == 2.5 # Day 3
|
||||
|
||||
def test_different_standard_hours_between_users(self, app):
|
||||
"""Integration test: different users with different standard hours"""
|
||||
# User 1: 8 hour standard
|
||||
user1 = User(username='user_8h', role='user')
|
||||
user1.standard_hours_per_day = 8.0
|
||||
db.session.add(user1)
|
||||
|
||||
# User 2: 6 hour standard (part-time)
|
||||
user2 = User(username='user_6h', role='user')
|
||||
user2.standard_hours_per_day = 6.0
|
||||
db.session.add(user2)
|
||||
|
||||
# Create client and project
|
||||
client_obj = Client(name='Multi User Client')
|
||||
db.session.add(client_obj)
|
||||
db.session.commit()
|
||||
|
||||
project = Project(name='Multi User Project', client_id=client_obj.id)
|
||||
db.session.add(project)
|
||||
db.session.commit()
|
||||
|
||||
# Both users work 7 hours today
|
||||
today = date.today()
|
||||
entry_start = datetime.combine(today, datetime.min.time().replace(hour=9))
|
||||
entry_end = entry_start + timedelta(hours=7)
|
||||
|
||||
entry1 = TimeEntry(
|
||||
user_id=user1.id,
|
||||
project_id=project.id,
|
||||
start_time=entry_start,
|
||||
end_time=entry_end
|
||||
)
|
||||
db.session.add(entry1)
|
||||
|
||||
entry2 = TimeEntry(
|
||||
user_id=user2.id,
|
||||
project_id=project.id,
|
||||
start_time=entry_start,
|
||||
end_time=entry_end
|
||||
)
|
||||
db.session.add(entry2)
|
||||
|
||||
db.session.commit()
|
||||
|
||||
# Calculate overtime for both users
|
||||
result1 = calculate_period_overtime(user1, today, today)
|
||||
result2 = calculate_period_overtime(user2, today, today)
|
||||
|
||||
# User 1: 7 hours, no overtime (under 8)
|
||||
assert result1['overtime_hours'] == 0.0
|
||||
assert result1['regular_hours'] == 7.0
|
||||
|
||||
# User 2: 7 hours, 1 hour overtime (over 6)
|
||||
assert result2['overtime_hours'] == 1.0
|
||||
assert result2['regular_hours'] == 6.0
|
||||
|
||||
Reference in New Issue
Block a user