Draft Release support in PyPI

tests/common/db/packaging.py

@@ -76,6 +76,9 @@ class Meta:
 
     uploader = factory.SubFactory(UserFactory)
     description = factory.SubFactory(DescriptionFactory)
+    published = factory.Faker(
+        "date_time_between_dates", datetime_start=datetime.datetime(2008, 1, 1)
+    )
 
 
 class FileFactory(WarehouseFactory):

tests/unit/forklift/test_legacy.py

@@ -1052,14 +1052,7 @@ def test_fails_with_invalid_names(self, pyramid_config, db_request, name):
 
     @pytest.mark.parametrize(
         "conflicting_name",
-        [
-            "toast1ng",
-            "toastlng",
-            "t0asting",
-            "toast-ing",
-            "toast.ing",
-            "toast_ing",
-        ],
+        ["toast1ng", "toastlng", "t0asting", "toast-ing", "toast.ing", "toast_ing"],
     )
     def test_fails_with_ultranormalized_names(
         self, pyramid_config, db_request, conflicting_name
@@ -3013,11 +3006,7 @@ def test_upload_updates_existing_project_name(
 
     # here
     @pytest.mark.parametrize(
-        "version, expected_version",
-        [
-            ("1.0", "1.0"),
-            ("v1.0", "1.0"),
-        ],
+        "version, expected_version", [("1.0", "1.0"), ("v1.0", "1.0")],
     )
     def test_upload_succeeds_creates_release(
         self, pyramid_config, db_request, metrics, version, expected_version
@@ -3129,6 +3118,77 @@ def test_upload_succeeds_creates_release(
             ),
         ]
 
+    @pytest.mark.parametrize("is_draft", [(True,), (False,)])
+    def test_upload_succeeds_creates_draft_release(
+        self, pyramid_config, db_request, metrics, is_draft,
+    ):
+        pyramid_config.testing_securitypolicy(userid=1)
+
+        user = UserFactory.create()
+        EmailFactory.create(user=user)
+        project = ProjectFactory.create()
+        RoleFactory.create(user=user, project=project)
+
+        db_request.db.add(Classifier(classifier="Environment :: Other Environment"))
+        db_request.db.add(Classifier(classifier="Programming Language :: Python"))
+
+        filename = "{}-{}.tar.gz".format(project.name, "1.0")
+
+        db_request.user = user
+        db_request.user_agent = "warehouse-tests/6.6.6"
+        draft_header = {"Is-Draft": "True" if is_draft else "False"}
+        db_request.headers.update(draft_header)
+        db_request.POST = MultiDict(
+            {
+                "metadata_version": "1.2",
+                "name": project.name,
+                "version": "1.0",
+                "summary": "This is my summary!",
+                "filetype": "sdist",
+                "md5_digest": _TAR_GZ_PKG_MD5,
+                "content": pretend.stub(
+                    filename=filename,
+                    file=io.BytesIO(_TAR_GZ_PKG_TESTDATA),
+                    type="application/tar",
+                ),
+            }
+        )
+        db_request.POST.extend(
+            [
+                ("classifiers", "Environment :: Other Environment"),
+                ("classifiers", "Programming Language :: Python"),
+                ("requires_dist", "foo"),
+                ("requires_dist", "bar (>1.0)"),
+                ("project_urls", "Test, https://example.com/"),
+                ("requires_external", "Cheese (>1.0)"),
+                ("provides", "testing"),
+            ]
+        )
+
+        storage_service = pretend.stub(store=lambda path, filepath, meta: None)
+        db_request.find_service = lambda svc, name=None, context=None: {
+            IFileStorage: storage_service,
+            IMetricsService: metrics,
+        }.get(svc)
+
+        resp = legacy.file_upload(db_request)
+
+        assert resp.status_code == 200
+
+        # Ensure that a Release object has been created.
+        release = (
+            db_request.db.query(Release)
+            .filter((Release.project == project) & (Release.version == "1.0"))
+            .one()
+        )
+        if is_draft:
+            assert release.published is None
+        else:
+            assert release.published is not None
+        assert release.is_draft is is_draft
+        assert release.version == "1.0"
+        assert release.canonical_version == "1"
+
     def test_upload_succeeds_creates_classifier(
         self, pyramid_config, db_request, metrics, monkeypatch
     ):

tests/unit/legacy/api/test_draft.py

@@ -0,0 +1,167 @@
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from warehouse.legacy.api import draft
+
+from ....common.db.accounts import UserFactory
+from ....common.db.packaging import (
+    FileFactory,
+    JournalEntryFactory,
+    ProjectFactory,
+    ReleaseFactory,
+)
+
+
+class TestDraftIndex:
+    def test_with_results_no_serial(self, db_request):
+        project = ProjectFactory.create()
+        release = ReleaseFactory.create(project=project, version="1.0", published=None)
+        draft_project_dict = {project.name: release}
+        assert draft.draft_index(draft_project_dict, db_request) == {
+            "draft_project_dict": draft_project_dict
+        }
+        assert db_request.response.headers["X-PyPI-Last-Serial"] == "0"
+
+    def test_with_results_with_serial(self, db_request):
+        project = ProjectFactory.create()
+        release = ReleaseFactory.create(project=project, version="1.0", published=None)
+        draft_project_dict = {project.name: release}
+        user = UserFactory.create()
+        je = JournalEntryFactory.create(submitted_by=user)
+
+        assert draft.draft_index(draft_project_dict, db_request) == {
+            "draft_project_dict": draft_project_dict
+        }
+        assert db_request.response.headers["X-PyPI-Last-Serial"] == str(je.id)
+
+
+class TestDraftDetail:
+    def test_no_files_no_serial(self, db_request):
+        project = ProjectFactory.create()
+        release = ReleaseFactory.create(project=project, version="1.0", published=None)
+        db_request.matchdict["name"] = project.normalized_name
+        user = UserFactory.create()
+        JournalEntryFactory.create(submitted_by=user)
+
+        assert draft.draft_detail(release, db_request) == {
+            "project": project,
+            "files": [],
+        }
+        assert db_request.response.headers["X-PyPI-Last-Serial"] == "0"
+
+    def test_no_files_with_serial(self, db_request):
+        project = ProjectFactory.create()
+        release = ReleaseFactory.create(project=project, version="1.0", published=None)
+        db_request.matchdict["name"] = project.normalized_name
+        user = UserFactory.create()
+        je = JournalEntryFactory.create(name=project.name, submitted_by=user)
+
+        assert draft.draft_detail(release, db_request) == {
+            "project": project,
+            "files": [],
+        }
+        assert db_request.response.headers["X-PyPI-Last-Serial"] == str(je.id)
+
+    def test_with_files_no_serial(self, db_request):
+        project = ProjectFactory.create()
+        release = ReleaseFactory.create(project=project, version="1.0", published=None)
+        files = [
+            FileFactory.create(
+                release=release,
+                filename="{}-{}.tar.gz".format(project.name, release.version),
+            )
+        ]
+        db_request.matchdict["name"] = project.normalized_name
+        user = UserFactory.create()
+        JournalEntryFactory.create(submitted_by=user)
+
+        assert draft.draft_detail(release, db_request) == {
+            "project": project,
+            "files": files,
+        }
+        assert db_request.response.headers["X-PyPI-Last-Serial"] == "0"
+
+    def test_with_files_with_serial(self, db_request):
+        project = ProjectFactory.create()
+        release = ReleaseFactory.create(project=project, version="1.0", published=None)
+        files = [
+            FileFactory.create(
+                release=release,
+                filename="{}-{}.tar.gz".format(project.name, release.version),
+            )
+        ]
+        db_request.matchdict["name"] = project.normalized_name
+        user = UserFactory.create()
+        je = JournalEntryFactory.create(name=project.name, submitted_by=user)
+
+        assert draft.draft_detail(release, db_request) == {
+            "project": project,
+            "files": files,
+        }
+        assert db_request.response.headers["X-PyPI-Last-Serial"] == str(je.id)
+
+    def test_with_files_with_version_multi_digit(self, db_request):
+        project = ProjectFactory.create()
+        release_versions = [
+            "0.3.0rc1",
+            "0.3.0",
+            "0.3.0-post0",
+            "0.14.0",
+            "4.2.0",
+            "24.2.0",
+        ]
+        releases = [
+            ReleaseFactory.create(project=project, version=version, published=None)
+            for version in release_versions
+        ]
+
+        tar_files = [
+            FileFactory.create(
+                release=r,
+                filename="{}-{}.tar.gz".format(project.name, r.version),
+                packagetype="sdist",
+            )
+            for r in releases
+        ]
+        wheel_files = [
+            FileFactory.create(
+                release=r,
+                filename="{}-{}.whl".format(project.name, r.version),
+                packagetype="bdist_wheel",
+            )
+            for r in releases
+        ]
+        egg_files = [
+            FileFactory.create(
+                release=r,
+                filename="{}-{}.egg".format(project.name, r.version),
+                packagetype="bdist_egg",
+            )
+            for r in releases
+        ]
+
+        files = [
+            list(files_release)
+            for files_release in zip(egg_files, tar_files, wheel_files)
+        ]
+
+        db_request.matchdict["name"] = project.normalized_name
+        user = UserFactory.create()
+        je = JournalEntryFactory.create(name=project.name, submitted_by=user)
+
+        for release, files_release in zip(releases, files):
+            assert draft.draft_detail(release, db_request) == {
+                "project": project,
+                "files": files_release,
+            }
+
+        assert db_request.response.headers["X-PyPI-Last-Serial"] == str(je.id)