adjust tests

Author: lynnagara

src/sentry/integrations/example/integration.py

@@ -220,11 +220,6 @@ class ExampleIntegrationProvider(IntegrationProvider):
             IntegrationFeatures.STACKTRACE_LINK,
         ]
     )
-    cell_restricted = False
-
-    @property
-    def is_cell_restricted(self) -> bool:
-        return self.cell_restricted
 
     def get_pipeline_views(self) -> Sequence[PipelineView[IntegrationPipeline]]:
         return [ExampleSetupView()]

tests/sentry/integrations/test_pipeline.py

@@ -75,7 +75,7 @@ def _modify_provider(self):
             yield
 
     def _setup_cell_restriction(self):
-        self.provider.cell_restricted = True
+        setattr(self.provider, "is_cell_restricted", True)
         na_orgs = [
             self.create_organization(name="na_org"),
             self.create_organization(name="na_org_2"),
@@ -177,11 +177,9 @@ def test_is_violating_cell_restriction_failure(self, *args) -> None:
             response = self.pipeline.finish_pipeline()
             assert isinstance(response, HttpResponse)
             error_message = "This integration has already been installed on another Sentry organization which resides in a different cell. Installation could not be completed."
-            assert error_message in response.content.decode()
-
             if SiloMode.get_current_mode() == SiloMode.MONOLITH:
                 assert error_message not in response.content.decode()
-            if SiloMode.get_current_mode() == SiloMode.CONTROL:
+            else:
                 assert error_message in response.content.decode()
 
     def test_aliased_integration_key(self, *args) -> None:
@@ -706,7 +704,7 @@ def _modify_provider(self):
             yield
 
     def _setup_cell_restriction(self):
-        self.provider.cell_restricted = True
+        setattr(self.provider, "is_cell_restricted", True)
         na_orgs = [
             self.create_organization(name="na_org"),
             self.create_organization(name="na_org_2"),