206 lines
6.2 KiB
Python
206 lines
6.2 KiB
Python
from datetime import date, timedelta
|
|
from decimal import Decimal
|
|
|
|
import pytest
|
|
from rest_framework.test import APIClient
|
|
|
|
from apps.clients.models import Client
|
|
from apps.projects.models import Project
|
|
from apps.tags.models import Tag
|
|
from apps.time_entries.models import TimeEntry
|
|
from apps.users.models import User
|
|
from apps.workspaces.models import Workspace, WorkspaceMembership
|
|
|
|
|
|
@pytest.fixture()
|
|
def api_client():
|
|
return APIClient()
|
|
|
|
|
|
@pytest.fixture()
|
|
def owner(db):
|
|
return User.objects.create_user(mobile="09128880001", password="secret123", first_name="Owner")
|
|
|
|
|
|
@pytest.fixture()
|
|
def admin(db):
|
|
return User.objects.create_user(mobile="09128880002", password="secret123", first_name="Admin")
|
|
|
|
|
|
@pytest.fixture()
|
|
def member(db):
|
|
return User.objects.create_user(mobile="09128880003", password="secret123", first_name="Member")
|
|
|
|
|
|
@pytest.fixture()
|
|
def workspace(owner, admin, member):
|
|
workspace = Workspace.objects.create(name="Reports", owner=owner)
|
|
WorkspaceMembership.objects.create(workspace=workspace, user=admin, role=WorkspaceMembership.Role.ADMIN, is_active=True)
|
|
WorkspaceMembership.objects.create(workspace=workspace, user=member, role=WorkspaceMembership.Role.MEMBER, is_active=True)
|
|
return workspace
|
|
|
|
|
|
@pytest.fixture()
|
|
def client(workspace):
|
|
return Client.objects.create(workspace=workspace, name="Acme")
|
|
|
|
|
|
@pytest.fixture()
|
|
def project(workspace, client):
|
|
return Project.objects.create(workspace=workspace, name="Website", client=client)
|
|
|
|
|
|
@pytest.fixture()
|
|
def tag(workspace):
|
|
return Tag.objects.create(workspace=workspace, name="Design", color="#ffffff")
|
|
|
|
|
|
@pytest.fixture()
|
|
def time_entries(workspace, owner, member, project, tag):
|
|
entry_owner = TimeEntry.objects.create(
|
|
workspace=workspace,
|
|
user=owner,
|
|
project=project,
|
|
description="Owner work",
|
|
start_time="2026-04-10T08:00:00+03:30",
|
|
end_time="2026-04-10T10:00:00+03:30",
|
|
duration=timedelta(hours=2),
|
|
is_billable=True,
|
|
hourly_rate=Decimal("25.00"),
|
|
currency="USD",
|
|
)
|
|
entry_owner.tags.add(tag)
|
|
entry_member = TimeEntry.objects.create(
|
|
workspace=workspace,
|
|
user=member,
|
|
project=project,
|
|
description="Member work",
|
|
start_time="2026-04-11T09:00:00+03:30",
|
|
end_time="2026-04-11T10:00:00+03:30",
|
|
duration=timedelta(hours=1),
|
|
is_billable=False,
|
|
currency="USD",
|
|
)
|
|
entry_member.tags.add(tag)
|
|
return [entry_owner, entry_member]
|
|
|
|
|
|
def test_member_only_sees_own_chart_report(api_client, member, workspace, time_entries):
|
|
api_client.force_authenticate(user=member)
|
|
|
|
response = api_client.get(
|
|
"/api/reports/chart/",
|
|
{"workspace": str(workspace.id), "period": "this_month"},
|
|
)
|
|
|
|
assert response.status_code == 200
|
|
assert response.data["summary"]["total_duration"] == "01:00:00"
|
|
|
|
|
|
def test_admin_can_request_combined_table_report(api_client, admin, workspace, time_entries):
|
|
api_client.force_authenticate(user=admin)
|
|
|
|
response = api_client.get(
|
|
"/api/reports/table/",
|
|
{"workspace": str(workspace.id), "period": "this_month"},
|
|
)
|
|
|
|
assert response.status_code == 200
|
|
assert response.data["summary"]["total_duration"] == "03:00:00"
|
|
assert len(response.data["days"]) == 2
|
|
assert response.data["days"][0]["latest_hourly_rate"] is None
|
|
assert response.data["days"][1]["latest_hourly_rate"] is None
|
|
|
|
|
|
def test_daily_rate_uses_latest_billable_entry_snapshot(api_client, owner, workspace, project):
|
|
api_client.force_authenticate(user=owner)
|
|
|
|
TimeEntry.objects.create(
|
|
workspace=workspace,
|
|
user=owner,
|
|
project=project,
|
|
description="Morning work",
|
|
start_time="2026-04-15T08:00:00+03:30",
|
|
end_time="2026-04-15T09:00:00+03:30",
|
|
duration=timedelta(hours=1),
|
|
is_billable=True,
|
|
hourly_rate=Decimal("20.00"),
|
|
currency="USD",
|
|
)
|
|
TimeEntry.objects.create(
|
|
workspace=workspace,
|
|
user=owner,
|
|
project=project,
|
|
description="Later work",
|
|
start_time="2026-04-15T13:00:00+03:30",
|
|
end_time="2026-04-15T15:00:00+03:30",
|
|
duration=timedelta(hours=2),
|
|
is_billable=True,
|
|
hourly_rate=Decimal("35.00"),
|
|
currency="USD",
|
|
)
|
|
|
|
response = api_client.get(
|
|
"/api/reports/table/",
|
|
{"workspace": str(workspace.id), "period": "this_month", "user": str(owner.id)},
|
|
)
|
|
|
|
assert response.status_code == 200
|
|
assert response.data["days"][0]["latest_hourly_rate"] == {
|
|
"amount": "35.00",
|
|
"currency": "USD",
|
|
}
|
|
|
|
|
|
def test_custom_period_longer_than_31_days_is_rejected(api_client, owner, workspace):
|
|
api_client.force_authenticate(user=owner)
|
|
|
|
response = api_client.get(
|
|
"/api/reports/chart/",
|
|
{
|
|
"workspace": str(workspace.id),
|
|
"period": "period",
|
|
"from_date": "2026-01-01",
|
|
"to_date": "2026-02-15",
|
|
},
|
|
)
|
|
|
|
assert response.status_code == 400
|
|
|
|
|
|
def test_persian_this_month_uses_jalali_month_bounds(api_client, owner, workspace, project, monkeypatch):
|
|
api_client.force_authenticate(user=owner)
|
|
monkeypatch.setattr("apps.reports.services.aggregation.timezone.localdate", lambda: date(2026, 4, 27))
|
|
|
|
TimeEntry.objects.create(
|
|
workspace=workspace,
|
|
user=owner,
|
|
project=project,
|
|
description="Previous jalali month",
|
|
start_time="2026-04-20T08:00:00+03:30",
|
|
end_time="2026-04-20T09:00:00+03:30",
|
|
duration=timedelta(hours=1),
|
|
is_billable=False,
|
|
currency="USD",
|
|
)
|
|
TimeEntry.objects.create(
|
|
workspace=workspace,
|
|
user=owner,
|
|
project=project,
|
|
description="Current jalali month",
|
|
start_time="2026-04-21T08:00:00+03:30",
|
|
end_time="2026-04-21T10:00:00+03:30",
|
|
duration=timedelta(hours=2),
|
|
is_billable=False,
|
|
currency="USD",
|
|
)
|
|
|
|
response = api_client.get(
|
|
"/api/reports/table/",
|
|
{"workspace": str(workspace.id), "period": "this_month", "language": "fa"},
|
|
)
|
|
|
|
assert response.status_code == 200
|
|
assert response.data["summary"]["total_duration"] == "02:00:00"
|
|
assert response.data["scope"]["from_date"] == "2026-04-21"
|