Fix test_set_wildcard_domain[postgres]

jvstme · jvstme · commit 54f389ed29d8 · 2026-01-26T22:17:09.000+01:00
diff --git a/src/dstack/_internal/server/services/gateways/__init__.py b/src/dstack/_internal/server/services/gateways/__init__.py
@@ -419,6 +419,7 @@ async def get_project_gateway_model_by_name_for_update(
             res = await session.execute(
                 select(GatewayModel)
                 .where(GatewayModel.id.in_([gateway_id]), *filters)
+                .options(selectinload(GatewayModel.gateway_compute))
                 .with_for_update(key_share=True)
             )
             yield res.scalar_one_or_none()

Original file line number	Diff line number	Diff line change
`@@ -419,6 +419,7 @@ async def get_project_gateway_model_by_name_for_update(`
`419`	`419`	`res = await session.execute(`
`420`	`420`	`select(GatewayModel)`
`421`	`421`	`.where(GatewayModel.id.in_([gateway_id]), *filters)`
	`422`	`+ .options(selectinload(GatewayModel.gateway_compute))`
`422`	`423`	`.with_for_update(key_share=True)`
`423`	`424`	`)`
`424`	`425`	`yield res.scalar_one_or_none()`