Ali Asadi
4946ec8c8f
fetch_users/fetch_tickets were placeholder stubs returning empty lists, so the generated Excel was always empty. Wire them to the existing utils.accounts_client / utils.advertising_client integrations (same ones the bundled-zip report on master uses), and update UserReport/TicketsReport to the real Account/Ticket field names instead of the old mock dict shape. Drop the now-unused local OAuth helper in favor of each client's own client-credentials flow. Co-Authored-By: Claude Sonnet 5 <noreply@anthropic.com>
152 lines
4.8 KiB
Python
152 lines
4.8 KiB
Python
from datetime import datetime
|
|
from io import BytesIO
|
|
from unittest.mock import MagicMock, patch
|
|
from uuid import uuid4
|
|
|
|
from openpyxl import load_workbook
|
|
from rest_framework import status
|
|
from rest_framework.test import APITestCase
|
|
|
|
from utils.clients.gooyal_advertising_client.models.ticket_state_enum import TicketStateEnum
|
|
|
|
|
|
class GenerateReportTests(APITestCase):
|
|
def _build_ticket(self, user_uuid=None):
|
|
ticket = MagicMock()
|
|
ticket.uuid = uuid4()
|
|
ticket.user = user_uuid or uuid4()
|
|
ticket.title = "Login issue"
|
|
ticket.description = "Cannot login"
|
|
ticket.state = TicketStateEnum(10)
|
|
ticket.created_at = datetime(2026, 5, 20, 10, 0)
|
|
ticket.updated_at = datetime(2026, 5, 21, 10, 0)
|
|
return ticket
|
|
|
|
def _build_user(self, user_uuid):
|
|
user = MagicMock()
|
|
user.uuid = user_uuid
|
|
user.username = "alice"
|
|
user.first_name = "Alice"
|
|
user.last_name = "Doe"
|
|
user.email = "alice@example.com"
|
|
user.phone_number = "09120000000"
|
|
return user
|
|
|
|
@patch("apps.reports.services.data_sources.ReportDataSources.fetch_user_metrics")
|
|
@patch("apps.reports.services.data_sources.ReportDataSources.fetch_users")
|
|
def test_generate_user_report_is_public_and_returns_excel(
|
|
self,
|
|
mock_fetch_users,
|
|
mock_fetch_user_metrics,
|
|
):
|
|
user = self._build_user(uuid4())
|
|
mock_fetch_users.return_value = [user]
|
|
mock_fetch_user_metrics.return_value = {str(user.uuid): {"ticket_count": 3}}
|
|
|
|
response = self.client.post(
|
|
"/api/reports/generate/",
|
|
{
|
|
"report_type": "user",
|
|
"filters": {"status": "active"},
|
|
},
|
|
format="json",
|
|
)
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
self.assertEqual(
|
|
response["Content-Type"],
|
|
"application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
|
|
)
|
|
self.assertEqual(
|
|
response["Content-Disposition"],
|
|
'attachment; filename="user-report.xlsx"',
|
|
)
|
|
|
|
workbook = load_workbook(BytesIO(response.content))
|
|
worksheet = workbook.active
|
|
self.assertEqual(worksheet["A1"].value, "User Report")
|
|
self.assertEqual(
|
|
[cell.value for cell in worksheet[2]],
|
|
[
|
|
"User UUID",
|
|
"Username",
|
|
"First Name",
|
|
"Last Name",
|
|
"Email",
|
|
"Phone Number",
|
|
"Ticket Count",
|
|
],
|
|
)
|
|
self.assertEqual(
|
|
[cell.value for cell in worksheet[3]],
|
|
[
|
|
str(user.uuid),
|
|
"alice",
|
|
"Alice",
|
|
"Doe",
|
|
"alice@example.com",
|
|
"09120000000",
|
|
3,
|
|
],
|
|
)
|
|
mock_fetch_users.assert_called_once_with({"status": "active"})
|
|
|
|
@patch("apps.reports.services.data_sources.ReportDataSources.fetch_ticket_metrics")
|
|
@patch("apps.reports.services.data_sources.ReportDataSources.fetch_tickets")
|
|
def test_generate_tickets_report_uses_static_columns(
|
|
self,
|
|
mock_fetch_tickets,
|
|
mock_fetch_ticket_metrics,
|
|
):
|
|
ticket = self._build_ticket()
|
|
mock_fetch_tickets.return_value = [ticket]
|
|
mock_fetch_ticket_metrics.return_value = {}
|
|
|
|
response = self.client.post(
|
|
"/api/reports/generate/",
|
|
{"report_type": "tickets"},
|
|
format="json",
|
|
)
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
self.assertEqual(
|
|
response["Content-Disposition"],
|
|
'attachment; filename="tickets-report.xlsx"',
|
|
)
|
|
|
|
workbook = load_workbook(BytesIO(response.content))
|
|
worksheet = workbook.active
|
|
self.assertEqual(worksheet["A1"].value, "Tickets Report")
|
|
self.assertEqual(
|
|
[cell.value for cell in worksheet[2]],
|
|
[
|
|
"Ticket UUID",
|
|
"User UUID",
|
|
"Title",
|
|
"Description",
|
|
"State",
|
|
"Created At",
|
|
"Updated At",
|
|
],
|
|
)
|
|
self.assertEqual(
|
|
[cell.value for cell in worksheet[3]],
|
|
[
|
|
str(ticket.uuid),
|
|
str(ticket.user),
|
|
"Login issue",
|
|
"Cannot login",
|
|
"CLOSED",
|
|
"2026-05-20 10:00:00",
|
|
"2026-05-21 10:00:00",
|
|
],
|
|
)
|
|
|
|
def test_unknown_report_type_returns_validation_error(self):
|
|
response = self.client.post(
|
|
"/api/reports/generate/",
|
|
{"report_type": "unknown"},
|
|
format="json",
|
|
)
|
|
|
|
self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
|