Add unit test for the package_content related features

Signed-off-by: tdruez <tdruez@aboutcode.org>

Author: tdruez

component_catalog/tests/test_models.py

@@ -2711,6 +2711,42 @@ def test_package_model_update_from_purldb_multiple_entries(self, mock_get_purldb
         self.assertEqual(["Keyword1", "Keyword2"], package1.keywords)
         self.assertEqual("Python", package1.primary_language)
 
+    @mock.patch("component_catalog.models.Package.get_purldb_entries")
+    def test_package_model_update_from_purldb_multiple_entries_package_content(
+        self, mock_get_entries
+    ):
+        purldb_entry_binary = {
+            "uuid": "e133e70b-8dd3-4cf1-9711-72b1f57523a0",
+            "purl": "pkg:pypi/boto3@1.37.26?file_name=boto3-1.37.26-py3-none-any.whl",
+            "type": "pypi",
+            "name": "boto3",
+            "version": "1.37.26",
+            "filename": "boto3-1.37.26-py3-none-any.whl",
+            "download_url": "https://files.pythonhosted.org/packages/boto3-1.37.26-py3-none-any.whl",
+            "package_content": "binary",
+        }
+        purldb_entry_source = {
+            "uuid": "326aa7a8-4f28-406d-89f9-c1404916925b",
+            "purl": "pkg:pypi/boto3@1.37.26?file_name=boto3-1.37.26.tar.gz",
+            "type": "pypi",
+            "name": "boto3",
+            "version": "1.37.26",
+            "filename": "boto3-1.37.26.tar.gz",
+            "download_url": "https://files.pythonhosted.org/packages/boto3-1.37.26.tar.gz",
+            "package_content": "source_archive",
+        }
+
+        mock_get_entries.return_value = [purldb_entry_binary, purldb_entry_source]
+        package1 = make_package(self.dataspace, package_url="pkg:pypi/boto3@1.37.26")
+        updated_fields = package1.update_from_purldb(self.user)
+        expected = ["download_url", "filename", "package_content"]
+        self.assertEqual(expected, sorted(updated_fields))
+
+        package1.refresh_from_db()
+        self.assertEqual(purldb_entry_source["download_url"], package1.download_url)
+        self.assertEqual(purldb_entry_source["filename"], package1.filename)
+        self.assertEqual("source_archive", package1.get_package_content_display())
+
     @mock.patch("component_catalog.models.Package.get_purldb_entries")
     def test_package_model_update_from_purldb_duplicate_exception(self, mock_get_purldb_entries):
         package_url = "pkg:pypi/django@3.0"

component_catalog/tests/test_views.py

@@ -3413,7 +3413,7 @@ def test_component_catalog_package_add_view_initial_data(
             "description": "Abbot Java GUI Test Library",
             "license_expression": "bsd-new OR eps-1.0 OR apache-2.0 OR mit",
             "declared_license_expression": "bsd-new OR eps-1.0 OR apache-2.0 OR mit",
-            'package_content': Package.PackageContentType.BINARY,
+            "package_content": Package.PackageContentType.BINARY,
         }
         self.assertEqual(expected, response.context["form"].initial)
 

purldb/tests/test_purldb_toolkit.py

@@ -12,6 +12,7 @@
 
 from dejacode_toolkit.purldb import PurlDB
 from dejacode_toolkit.purldb import pick_purldb_entry
+from dejacode_toolkit.purldb import pick_source_package
 from dje.models import Dataspace
 from dje.tests import create_user
 
@@ -75,3 +76,29 @@ def test_purldb_toolkit_pick_purldb_entry(self):
         self.assertEqual(entry2, pick_purldb_entry([entry1, entry2], purl=purl2))
         self.assertIsNone(pick_purldb_entry([entry1, entry1], purl=purl1))
         self.assertIsNone(pick_purldb_entry([entry1, entry2], purl=purl3))
+
+    def test_purldb_toolkit_pick_source_package(self):
+        self.assertIsNone(pick_source_package(None))
+        self.assertIsNone(pick_source_package([]))
+
+        entry_binary = {
+            "purl": "pkg:pypi/boto3@1.37.26?file_name=boto3-1.37.26-py3-none-any.whl",
+            "filename": "boto3-1.37.26-py3-none-any.whl",
+            "download_url": "https://files.pythonhosted.org/boto3-1.37.26-py3-none-any.whl",
+            "package_content": "binary",
+        }
+        entry_source = {
+            "purl": "pkg:pypi/boto3@1.37.26?file_name=boto3-1.37.26.tar.gz",
+            "filename": "boto3-1.37.26.tar.gz",
+            "download_url": "https://files.pythonhosted.org/boto3-1.37.26.tar.gz",
+            "package_content": "source_archive",
+        }
+
+        self.assertEqual(entry_binary, pick_source_package([entry_binary]))
+        self.assertIsNone(pick_source_package([entry_binary, entry_binary]))
+
+        self.assertEqual(entry_source, pick_source_package([entry_source]))
+        self.assertEqual(entry_source, pick_source_package([entry_source, entry_source]))
+
+        self.assertEqual(entry_source, pick_source_package([entry_source, entry_binary]))
+        self.assertEqual(entry_source, pick_source_package([entry_binary, entry_source]))