diff --git a/tests/unit/enterprise/proxy/test_managed_files_hook.py b/tests/unit/enterprise/proxy/test_managed_files_hook.py index 5384e34f59b..cea81850cbf 100644 --- a/tests/unit/enterprise/proxy/test_managed_files_hook.py +++ b/tests/unit/enterprise/proxy/test_managed_files_hook.py @@ -331,14 +331,10 @@ async def test_provider_file_id_resolver_returns_owned_mappings_with_owner_scope "file-provider-2": "unified-file-id", } assert isinstance(unified_file_ids, MappingProxyType) - find_many.assert_awaited_once_with( - where={ - "OR": [{"created_by": "test-user"}, {"team_id": "test-team"}], - "flat_model_file_ids": { - "hasSome": ["file-provider-1", "file-provider-2", "file-unmanaged-2"], - }, - } - ) + find_many.assert_awaited_once() + lookup: Final = find_many.await_args.kwargs["where"] + assert lookup["OR"] == [{"created_by": "test-user"}, {"team_id": "test-team"}] + assert {"file-provider-1", "file-provider-2", "file-unmanaged-2"} <= set(lookup["flat_model_file_ids"]["hasSome"]) @pytest.mark.asyncio