updated tests

Author: callmecampos

tests/common/db/packaging.py

@@ -75,6 +75,7 @@ class Meta:
 
     uploader = factory.SubFactory(UserFactory)
     description = factory.SubFactory(DescriptionFactory)
+    files = []
 
 
 class FileFactory(WarehouseFactory):

tests/unit/search/test_tasks.py

@@ -31,7 +31,7 @@
     unindex_project,
 )
 
-from ...common.db.packaging import ProjectFactory, ReleaseFactory
+from ...common.db.packaging import FileFactory, ProjectFactory, ReleaseFactory
 
 
 def test_project_docs(db_session):
@@ -45,6 +45,15 @@ def test_project_docs(db_session):
         for p in projects
     }
 
+    for p in projects:
+        for r in releases[p]:
+            f = FileFactory.create(
+                release=r,
+                filename="{}-{}.tar.gz".format(p.name, r.version),
+                python_version="source",
+            )
+            r.files.append(f)
+
     assert list(_project_docs(db_session)) == [
         {
             "_id": p.normalized_name,
@@ -75,6 +84,15 @@ def test_single_project_doc(db_session):
         for p in projects
     }
 
+    for p in projects:
+        for r in releases[p]:
+            f = FileFactory.create(
+                release=r,
+                filename="{}-{}.tar.gz".format(p.name, r.version),
+                python_version="source",
+            )
+            r.files.append(f)
+
     assert list(_project_docs(db_session, project_name=projects[1].name)) == [
         {
             "_id": p.normalized_name,

warehouse/search/tasks.py

@@ -37,11 +37,15 @@
 from warehouse.utils.db import windowed_query
 
 
-def _project_docs(db, project_name=None):
+def _project_docs(db, project_name=None, show_empty_releases=False):
+
+    queries = [Release.yanked.is_(False)]
+    if not show_empty_releases:
+        queries.append(Release.files)
 
     releases_list = (
         db.query(Release.id)
-        .filter(Release.yanked.is_(False), Release.files)
+        .filter(*queries)
         .order_by(
             Release.project_id,
             Release.is_prerelease.nullslast(),