263 lines
9.3 KiB
Python
263 lines
9.3 KiB
Python
from datetime import date, timedelta
|
|
from decimal import Decimal
|
|
from unittest.mock import patch
|
|
|
|
from django.core.cache import cache
|
|
from rest_framework.test import APITestCase
|
|
|
|
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
|
|
|
|
|
|
class ReportViewTests(APITestCase):
|
|
@classmethod
|
|
def setUpTestData(cls):
|
|
cls.owner = User.objects.create_user(
|
|
mobile="09128880001",
|
|
password="secret123",
|
|
first_name="Owner",
|
|
)
|
|
cls.admin = User.objects.create_user(
|
|
mobile="09128880002",
|
|
password="secret123",
|
|
first_name="Admin",
|
|
)
|
|
cls.member = User.objects.create_user(
|
|
mobile="09128880003",
|
|
password="secret123",
|
|
first_name="Member",
|
|
)
|
|
cls.workspace = Workspace.objects.create(name="Reports", owner=cls.owner)
|
|
WorkspaceMembership.objects.create(
|
|
workspace=cls.workspace,
|
|
user=cls.admin,
|
|
role=WorkspaceMembership.Role.ADMIN,
|
|
is_active=True,
|
|
)
|
|
WorkspaceMembership.objects.create(
|
|
workspace=cls.workspace,
|
|
user=cls.member,
|
|
role=WorkspaceMembership.Role.MEMBER,
|
|
is_active=True,
|
|
)
|
|
cls.client_obj = Client.objects.create(workspace=cls.workspace, name="Acme")
|
|
cls.project = Project.objects.create(
|
|
workspace=cls.workspace,
|
|
name="Website",
|
|
client=cls.client_obj,
|
|
)
|
|
cls.tag = Tag.objects.create(
|
|
workspace=cls.workspace,
|
|
name="Design",
|
|
color="#ffffff",
|
|
)
|
|
|
|
entry_owner = TimeEntry.objects.create(
|
|
workspace=cls.workspace,
|
|
user=cls.owner,
|
|
project=cls.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(cls.tag)
|
|
|
|
entry_member = TimeEntry.objects.create(
|
|
workspace=cls.workspace,
|
|
user=cls.member,
|
|
project=cls.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(cls.tag)
|
|
|
|
def setUp(self):
|
|
cache.clear()
|
|
|
|
def test_member_only_sees_own_chart_report(self):
|
|
self.client.force_authenticate(user=self.member)
|
|
|
|
with patch(
|
|
"apps.reports.services.aggregation.timezone.localdate",
|
|
return_value=date(2026, 4, 20),
|
|
):
|
|
response = self.client.get(
|
|
"/api/reports/chart/",
|
|
{"workspace": str(self.workspace.id), "period": "this_month"},
|
|
)
|
|
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(response.data["summary"]["total_duration"], "01:00:00")
|
|
|
|
def test_admin_can_request_combined_table_report(self):
|
|
self.client.force_authenticate(user=self.admin)
|
|
|
|
with patch(
|
|
"apps.reports.services.aggregation.timezone.localdate",
|
|
return_value=date(2026, 4, 20),
|
|
):
|
|
response = self.client.get(
|
|
"/api/reports/table/",
|
|
{"workspace": str(self.workspace.id), "period": "this_month"},
|
|
)
|
|
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(response.data["summary"]["total_duration"], "03:00:00")
|
|
self.assertEqual(len(response.data["days"]), 2)
|
|
self.assertIsNone(response.data["days"][0]["latest_hourly_rate"])
|
|
self.assertIsNone(response.data["days"][1]["latest_hourly_rate"])
|
|
|
|
def test_daily_rate_uses_latest_billable_entry_snapshot(self):
|
|
self.client.force_authenticate(user=self.owner)
|
|
|
|
TimeEntry.objects.create(
|
|
workspace=self.workspace,
|
|
user=self.owner,
|
|
project=self.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=self.workspace,
|
|
user=self.owner,
|
|
project=self.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",
|
|
)
|
|
|
|
with patch(
|
|
"apps.reports.services.aggregation.timezone.localdate",
|
|
return_value=date(2026, 4, 20),
|
|
):
|
|
response = self.client.get(
|
|
"/api/reports/table/",
|
|
{
|
|
"workspace": str(self.workspace.id),
|
|
"period": "this_month",
|
|
"user": str(self.owner.id),
|
|
},
|
|
)
|
|
|
|
self.assertEqual(response.status_code, 200)
|
|
target_day = next(day for day in response.data["days"] if day["date"] == "2026-04-15")
|
|
self.assertEqual(
|
|
target_day["latest_hourly_rate"],
|
|
{"amount": "35.00", "currency": "USD"},
|
|
)
|
|
|
|
def test_custom_period_longer_than_31_days_is_rejected(self):
|
|
self.client.force_authenticate(user=self.owner)
|
|
|
|
response = self.client.get(
|
|
"/api/reports/chart/",
|
|
{
|
|
"workspace": str(self.workspace.id),
|
|
"period": "period",
|
|
"from_date": "2026-01-01",
|
|
"to_date": "2026-02-15",
|
|
},
|
|
)
|
|
|
|
self.assertEqual(response.status_code, 400)
|
|
|
|
def test_persian_this_month_uses_jalali_month_bounds(self):
|
|
self.client.force_authenticate(user=self.owner)
|
|
|
|
with patch(
|
|
"apps.reports.services.aggregation.timezone.localdate",
|
|
return_value=date(2026, 4, 27),
|
|
):
|
|
TimeEntry.objects.create(
|
|
workspace=self.workspace,
|
|
user=self.owner,
|
|
project=self.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=self.workspace,
|
|
user=self.owner,
|
|
project=self.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 = self.client.get(
|
|
"/api/reports/table/",
|
|
{
|
|
"workspace": str(self.workspace.id),
|
|
"period": "this_month",
|
|
"language": "fa",
|
|
},
|
|
)
|
|
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(response.data["summary"]["total_duration"], "02:00:00")
|
|
self.assertEqual(response.data["scope"]["from_date"], "2026-04-21")
|
|
|
|
def test_table_report_cache_stays_until_time_entry_invalidation(self):
|
|
self.client.force_authenticate(user=self.owner)
|
|
url = "/api/reports/table/"
|
|
params = {"workspace": str(self.workspace.id), "period": "this_month"}
|
|
|
|
with patch(
|
|
"apps.reports.services.aggregation.timezone.localdate",
|
|
return_value=date(2026, 4, 20),
|
|
):
|
|
first_response = self.client.get(url, params)
|
|
self.assertEqual(first_response.status_code, 200)
|
|
self.assertEqual(first_response.data["summary"]["total_duration"], "03:00:00")
|
|
|
|
member_entry = TimeEntry.objects.get(description="Member work")
|
|
TimeEntry.objects.filter(id=member_entry.id).update(duration=timedelta(hours=5))
|
|
|
|
with patch(
|
|
"apps.reports.services.aggregation.timezone.localdate",
|
|
return_value=date(2026, 4, 20),
|
|
):
|
|
cached_response = self.client.get(url, params)
|
|
self.assertEqual(cached_response.status_code, 200)
|
|
self.assertEqual(cached_response.data["summary"]["total_duration"], "03:00:00")
|
|
|
|
member_entry.refresh_from_db()
|
|
member_entry.description = "Member work updated"
|
|
member_entry.save(update_fields=["description"])
|
|
|
|
with patch(
|
|
"apps.reports.services.aggregation.timezone.localdate",
|
|
return_value=date(2026, 4, 20),
|
|
):
|
|
fresh_response = self.client.get(url, params)
|
|
self.assertEqual(fresh_response.status_code, 200)
|
|
self.assertEqual(fresh_response.data["summary"]["total_duration"], "07:00:00")
|