1 parent b7ed3e8 commit f666486Copy full SHA for f666486
1 file changed
tests/test_optimizer.py
@@ -2350,9 +2350,14 @@ def test_query_aliased_annotate_callable_with_same_arguments(
2350
IssueFactory.create(milestone=milestone, name="foo1")
2351
IssueFactory.create(milestone=milestone, name="bar1")
2352
2353
- with assert_num_queries(1 if DjangoOptimizerExtension.enabled.get() else 3):
+ optimized = DjangoOptimizerExtension.enabled.get()
2354
+ with assert_num_queries(1 if optimized else 3) as ctx:
2355
res = gql_client.query(query)
2356
2357
+ if optimized:
2358
+ sql = ctx.captured_queries[0]["sql"]
2359
+ assert sql.count("_strawberry_alias_") == 2
2360
+
2361
assert res.data == {
2362
"milestoneList": [
2363
{
0 commit comments