55from django .conf import settings
66from django .contrib .auth .models import Group , Permission
77from django .core import management
8+ from django .utils .translation import get_language as django_get_language
89
910from api .factories .country import CountryFactory
1011from 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