Skip to content

Commit f666486

Browse files
test: assert sql in test_query_aliased_annotate_callable_with_same_arguments
1 parent b7ed3e8 commit f666486

1 file changed

Lines changed: 6 additions & 1 deletion

File tree

tests/test_optimizer.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2350,9 +2350,14 @@ def test_query_aliased_annotate_callable_with_same_arguments(
23502350
IssueFactory.create(milestone=milestone, name="foo1")
23512351
IssueFactory.create(milestone=milestone, name="bar1")
23522352

2353-
with assert_num_queries(1 if DjangoOptimizerExtension.enabled.get() else 3):
2353+
optimized = DjangoOptimizerExtension.enabled.get()
2354+
with assert_num_queries(1 if optimized else 3) as ctx:
23542355
res = gql_client.query(query)
23552356

2357+
if optimized:
2358+
sql = ctx.captured_queries[0]["sql"]
2359+
assert sql.count("_strawberry_alias_") == 2
2360+
23562361
assert res.data == {
23572362
"milestoneList": [
23582363
{

0 commit comments

Comments
 (0)