mock in_test_environ

Author: aliu39

tests/sentry/seer/endpoints/test_seer_rpc.py

@@ -112,16 +112,21 @@ def test_generic_exceptions_return_500(self) -> None:
         path = self._get_path("get_organization_slug")
         data: dict[str, Any] = {"args": {"org_id": 1}, "meta": {}}
 
-        with patch(
-            "sentry.seer.endpoints.seer_rpc.SeerRpcServiceEndpoint._dispatch_to_local_method"
-        ) as mock_dispatch:
-            mock_dispatch.side_effect = RuntimeError("Unexpected internal error")
-
-            response = self.client.post(
-                path, data=data, HTTP_AUTHORIZATION=self.auth_header(path, data)
-            )
-
-        assert response.status_code == 500
+        for is_test_environment in [True, False]:
+            with patch(
+                "sentry.seer.endpoints.seer_rpc.in_test_environment",
+                return_value=is_test_environment,
+            ):
+                with patch(
+                    "sentry.seer.endpoints.seer_rpc.SeerRpcServiceEndpoint._dispatch_to_local_method"
+                ) as mock_dispatch:
+                    mock_dispatch.side_effect = RuntimeError("Unexpected internal error")
+
+                    response = self.client.post(
+                        path, data=data, HTTP_AUTHORIZATION=self.auth_header(path, data)
+                    )
+
+                assert response.status_code == 500
 
 
 class TestSeerRpcMethods(APITestCase):