@@ -103,49 +103,24 @@ def create_app(schema):
103103def test_async_execution_uses_async_extension (sentry_init ):
104104 sentry_init (integrations = [StrawberryIntegration (async_execution = True )])
105105
106- with mock .patch (
107- "sentry_sdk.integrations.strawberry._get_installed_modules" ,
108- return_value = {"flask" : "2.3.3" },
109- ):
110- # actual installed modules should not matter, the explicit option takes
111- # precedence
112- schema = strawberry .Schema (Query )
113- assert SentryAsyncExtension in schema .extensions
106+ schema = strawberry .Schema (Query )
107+ assert SentryAsyncExtension in schema .extensions
108+ assert SentrySyncExtension not in schema .extensions
114109
115110
116111def test_sync_execution_uses_sync_extension (sentry_init ):
117112 sentry_init (integrations = [StrawberryIntegration (async_execution = False )])
118113
119- with mock .patch (
120- "sentry_sdk.integrations.strawberry._get_installed_modules" ,
121- return_value = {"fastapi" : "0.103.1" , "starlette" : "0.27.0" },
122- ):
123- # actual installed modules should not matter, the explicit option takes
124- # precedence
125- schema = strawberry .Schema (Query )
126- assert SentrySyncExtension in schema .extensions
127-
128-
129- def test_infer_execution_type_from_installed_packages_async (sentry_init ):
130- sentry_init (integrations = [StrawberryIntegration ()])
131-
132- with mock .patch (
133- "sentry_sdk.integrations.strawberry._get_installed_modules" ,
134- return_value = {"fastapi" : "0.103.1" , "starlette" : "0.27.0" },
135- ):
136- schema = strawberry .Schema (Query )
137- assert SentryAsyncExtension in schema .extensions
114+ schema = strawberry .Schema (Query )
115+ assert SentrySyncExtension in schema .extensions
116+ assert SentryAsyncExtension not in schema .extensions
138117
139118
140- def test_infer_execution_type_from_installed_packages_sync (sentry_init ):
119+ def test_use_sync_extension_if_not_specified (sentry_init ):
141120 sentry_init (integrations = [StrawberryIntegration ()])
142-
143- with mock .patch (
144- "sentry_sdk.integrations.strawberry._get_installed_modules" ,
145- return_value = {"flask" : "2.3.3" },
146- ):
147- schema = strawberry .Schema (Query )
148- assert SentrySyncExtension in schema .extensions
121+ schema = strawberry .Schema (Query )
122+ assert SentrySyncExtension in schema .extensions
123+ assert SentryAsyncExtension not in schema .extensions
149124
150125
151126@pytest .mark .skipif (
0 commit comments