@@ -50,7 +50,19 @@ def test_get_returns_empty_validation_payload_when_no_errors(self):
5050 self .endpoint_under_test ,
5151 )
5252
53- assert response .json () == {"errors" : []}
53+ response_json = response .json ()
54+
55+ assert response_json ["payload" ] == {"errors" : []}
56+ assert response_json ["report_data" ] == {
57+ "reporting_year" : self .report_version .report .reporting_year_id ,
58+ "report_version_id" : self .report_version .id ,
59+ }
60+ assert response_json ["operation_data" ] == {
61+ "naics_code" : self .report_version .report .operation .naics_code .naics_code ,
62+ "operation_type" : self .report_version .report .operation .type ,
63+ "is_operation_opted_out" : False ,
64+ }
65+
5466 mock_validate .assert_called_once ()
5567
5668 def test_get_returns_validation_errors_with_context (self ):
@@ -99,7 +111,9 @@ def test_get_returns_validation_errors_with_context(self):
99111 self .endpoint_under_test ,
100112 )
101113
102- assert response .json () == {
114+ response_json = response .json ()
115+
116+ assert response_json ["payload" ] == {
103117 "errors" : [
104118 {
105119 "key" : "error_required_fields" ,
@@ -133,6 +147,18 @@ def test_get_returns_validation_errors_with_context(self):
133147 },
134148 ],
135149 }
150+
151+ assert response_json ["report_data" ] == {
152+ "reporting_year" : self .report_version .report .reporting_year_id ,
153+ "report_version_id" : self .report_version .id ,
154+ }
155+
156+ assert response_json ["operation_data" ] == {
157+ "naics_code" : self .report_version .report .operation .naics_code .naics_code ,
158+ "operation_type" : self .report_version .report .operation .type ,
159+ "is_operation_opted_out" : False ,
160+ }
161+
136162 mock_validate .assert_called_once ()
137163
138164 def test_validates_report_version_id (self ):
0 commit comments