def test_package_query_count(django_assert_max_num_queries, package_count, version_count, community_site): with CaptureQueriesContext(connection) as context: for package_id in range(package_count): package = PackageFactory.create( owner=UploaderIdentityFactory.create( name=f"uploader_{package_id}" ), name=f"package_{package_id}", ) for version_id in range(version_count): PackageVersionFactory.create( package=package, name=f"package_{package_id}", version_number=f"{version_id}.0.0", ) PackageListing.objects.create(package=package, community=community_site.community) creation_queries = len(context) packages = get_package_listing_queryset(community_site) with django_assert_max_num_queries(package_count + creation_queries + 1): serializer = PackageListingSerializer(packages, many=True, context={"community_site": community_site}) _ = serializer.data