99from launchpad .artifacts .apple .zipped_xcarchive import ZippedXCArchive
1010from launchpad .artifacts .artifact import Artifact
1111from launchpad .constants import (
12- DistributionState ,
12+ InstallableAppErrorCode ,
1313 ProcessingErrorCode ,
1414 ProcessingErrorMessage ,
1515)
@@ -142,7 +142,7 @@ def test_processing_error_message_enum_values(self):
142142
143143 def test_do_distribution_unknown_artifact_type_skips (self ):
144144 mock_sentry_client = Mock (spec = SentryClient )
145- mock_sentry_client .update_artifact .return_value = None
145+ mock_sentry_client .update_distribution_error .return_value = None
146146 self .processor ._sentry_client = mock_sentry_client
147147
148148 unknown_artifact = Mock (spec = Artifact )
@@ -152,19 +152,16 @@ def test_do_distribution_unknown_artifact_type_skips(self):
152152 "test-org-id" , "test-project-id" , "test-artifact-id" , unknown_artifact , mock_info
153153 )
154154
155- mock_sentry_client .update_artifact .assert_called_once_with (
155+ mock_sentry_client .update_distribution_error .assert_called_once_with (
156156 org = "test-org-id" ,
157- project = "test-project-id" ,
158157 artifact_id = "test-artifact-id" ,
159- data = {
160- "distribution_state" : DistributionState .NOT_RAN .value ,
161- "distribution_skip_reason" : "unsupported" ,
162- },
158+ error_code = InstallableAppErrorCode .SKIPPED .value ,
159+ error_message = "unsupported" ,
163160 )
164161
165162 def test_do_distribution_invalid_code_signature_skips (self ):
166163 mock_sentry_client = Mock (spec = SentryClient )
167- mock_sentry_client .update_artifact .return_value = None
164+ mock_sentry_client .update_distribution_error .return_value = None
168165 self .processor ._sentry_client = mock_sentry_client
169166
170167 artifact = Mock (spec = ZippedXCArchive )
@@ -174,20 +171,17 @@ def test_do_distribution_invalid_code_signature_skips(self):
174171
175172 self .processor ._do_distribution ("test-org-id" , "test-project-id" , "test-artifact-id" , artifact , mock_info )
176173
177- mock_sentry_client .update_artifact .assert_called_once_with (
174+ mock_sentry_client .update_distribution_error .assert_called_once_with (
178175 org = "test-org-id" ,
179- project = "test-project-id" ,
180176 artifact_id = "test-artifact-id" ,
181- data = {
182- "distribution_state" : DistributionState .NOT_RAN .value ,
183- "distribution_skip_reason" : "invalid_signature" ,
184- },
177+ error_code = InstallableAppErrorCode .SKIPPED .value ,
178+ error_message = "invalid_signature" ,
185179 )
186180 mock_sentry_client .upload_installable_app .assert_not_called ()
187181
188182 def test_do_distribution_simulator_build_skips (self ):
189183 mock_sentry_client = Mock (spec = SentryClient )
190- mock_sentry_client .update_artifact .return_value = None
184+ mock_sentry_client .update_distribution_error .return_value = None
191185 self .processor ._sentry_client = mock_sentry_client
192186
193187 artifact = Mock (spec = ZippedXCArchive )
@@ -197,14 +191,11 @@ def test_do_distribution_simulator_build_skips(self):
197191
198192 self .processor ._do_distribution ("test-org-id" , "test-project-id" , "test-artifact-id" , artifact , mock_info )
199193
200- mock_sentry_client .update_artifact .assert_called_once_with (
194+ mock_sentry_client .update_distribution_error .assert_called_once_with (
201195 org = "test-org-id" ,
202- project = "test-project-id" ,
203196 artifact_id = "test-artifact-id" ,
204- data = {
205- "distribution_state" : DistributionState .NOT_RAN .value ,
206- "distribution_skip_reason" : "simulator" ,
207- },
197+ error_code = InstallableAppErrorCode .SKIPPED .value ,
198+ error_message = "simulator" ,
208199 )
209200 mock_sentry_client .upload_installable_app .assert_not_called ()
210201
0 commit comments