Add test capturing expectation. Ref #404

jaraco · jaraco · commit f480907325bc · 2023-12-03T12:09:24.000-05:00
diff --git a/tests/fixtures.py b/tests/fixtures.py
@@ -1,6 +1,7 @@
 import os
 import sys
 import copy
+import json
 import shutil
 import pathlib
 import tempfile
@@ -127,6 +128,31 @@ def make_uppercase(self):
         build_files(files, self.site_dir)
 
 
+class DistInfoPkgEditable(DistInfoPkg):
+    """
+    Package with a PEP 660 direct_url.json.
+    """
+
+    some_hash = '524127ce937f7cb65665130c695abd18ca386f60bb29687efb976faa1596fdcc'
+    files: FilesSpec = {
+        'distinfo_pkg-1.0.0.dist-info': {
+            'direct_url.json': json.dumps(
+                {
+                    "archive_info": {
+                        "hash": f"sha256={some_hash}",
+                        "hashes": {"sha256": f"{some_hash}"},
+                    },
+                    "url": "file:///path/to/distinfo_pkg-1.0.0.editable-py3-none-any.whl",
+                }
+            )
+        },
+    }
+
+    def setUp(self):
+        super().setUp()
+        build_files(DistInfoPkgEditable.files, self.site_dir)
+
+
 class DistInfoPkgWithDot(OnSysPath, SiteDir):
     files: FilesSpec = {
         "pkg_dot-1.0.0.dist-info": {
diff --git a/tests/test_main.py b/tests/test_main.py
@@ -457,3 +457,10 @@ def import_names_from_package(package_name):
         # sources_fallback-pkg has one import ('sources_fallback') inferred from
         # SOURCES.txt (top_level.txt and installed-files.txt is missing)
         assert import_names_from_package('sources_fallback-pkg') == {'sources_fallback'}
+
+
+class EditableDistributionTest(fixtures.DistInfoPkgEditable, unittest.TestCase):
+    def test_origin(self):
+        dist = Distribution.from_name('distinfo-pkg')
+        assert dist.origin.url.endswith('.whl')
+        assert dist.origin.archive_info.hashes.sha256
