Skip to content

Commit 46c9801

Browse files
committed
fix(eap): Update test cases for simplified eap generate pdf
1 parent 2da4822 commit 46c9801

1 file changed

Lines changed: 8 additions & 9 deletions

File tree

eap/test_views.py

Lines changed: 8 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
from django.conf import settings
66
from django.contrib.auth.models import Group, Permission
77
from django.core import management
8+
from django.utils.translation import get_language as django_get_language
89

910
from api.factories.country import CountryFactory
1011
from api.factories.disaster_type import DisasterTypeFactory
@@ -1236,7 +1237,7 @@ def setUp(self):
12361237
self.url = "/api/v2/pdf-export/"
12371238

12381239
@mock.patch("api.serializers.generate_url.delay")
1239-
def test_create_simplified_eap_export(self, mock_generate_url):
1240+
def test_simplified_eap_export(self, mock_generate_url):
12401241
self.simplified_eap = SimplifiedEAPFactory.create(
12411242
eap_registration=self.eap_registration,
12421243
created_by=self.user,
@@ -1254,20 +1255,18 @@ def test_create_simplified_eap_export(self, mock_generate_url):
12541255
with self.capture_on_commit_callbacks(execute=True):
12551256
response = self.client.post(self.url, data, format="json")
12561257
self.assert_201(response)
1257-
export = Export.objects.first()
1258-
self.assertIsNotNone(export)
1258+
self.assertIsNotNone(response.data["id"], response.data)
12591259

1260-
expected_url = (
1261-
f"{settings.GO_WEB_INTERNAL_URL}/" f"{Export.ExportType.SIMPLIFIED_EAP}/" f"{self.simplified_eap.id}/export/"
1262-
)
1263-
self.assertEqual(export.url, expected_url)
1260+
expected_url = f"{settings.GO_WEB_INTERNAL_URL}/{Export.ExportType.SIMPLIFIED_EAP}/{self.simplified_eap.id}/export/"
1261+
self.assertEqual(response.data["url"], expected_url)
12641262
self.assertEqual(response.data["status"], Export.ExportStatus.PENDING)
12651263

12661264
self.assertEqual(mock_generate_url.called, True)
12671265
title = f"{self.national_society.name}-{self.disaster_type.name}"
12681266
mock_generate_url.assert_called_once_with(
1269-
export.url,
1270-
export.id,
1267+
expected_url,
1268+
response.data["id"],
12711269
self.user.id,
12721270
title,
1271+
django_get_language(),
12731272
)

0 commit comments

Comments
 (0)