Add unit tests

Signed-off-by: tdruez <[email protected]>

Author: tdruez

component_catalog/models.py

@@ -2045,7 +2045,8 @@ def inferred_repo_url(self):
 
     def infer_download_url(self):
         """Infer the download URL deduced from the Package URL (purl)."""
-        return download.infer_download_url(self.package_url)
+        if self.package_url:
+            return download.infer_download_url(self.package_url)
 
     def get_url(self, name, params=None, include_identifier=False):
         if not params:

component_catalog/tests/test_models.py

@@ -2576,6 +2576,15 @@ def test_package_model_inferred_repo_url_property(self):
         expected = "https://github.com/package-url/packageurl-python/tree/v0.10.4"
         self.assertEqual(expected, package1.inferred_repo_url)
 
+    def test_package_model_infer_download_url(self):
+        package1 = make_package(self.dataspace, filename="package")
+        self.assertIsNone(package1.infer_download_url())
+
+        package1.set_package_url("pkg:nuget/[email protected]")
+        package1.save()
+        expected_download_url = "https://www.nuget.org/api/v2/package/Azure.Core/1.45.0"
+        self.assertEqual(expected_download_url, package1.infer_download_url())
+
     @mock.patch("dejacode_toolkit.purldb.PurlDB.find_packages")
     def test_package_model_get_purldb_entries(self, mock_find_packages):
         purl1 = "pkg:pypi/[email protected]"
@@ -2758,6 +2767,12 @@ def test_package_queryset_has_package_url(self):
         qs = Package.objects.has_package_url()
         self.assertQuerySetEqual(qs, [package1])
 
+    def test_package_queryset_has_download_url(self):
+        package1 = make_package(self.dataspace, download_url="https://download.url")
+        make_package(self.dataspace)
+        qs = Package.objects.has_download_url()
+        self.assertQuerySetEqual(qs, [package1])
+
     def test_package_queryset_annotate_sortable_identifier(self):
         package1 = make_package(self.dataspace, package_url="pkg:pypi/[email protected]")
         package2 = make_package(self.dataspace)

product_portfolio/tests/test_api.py

@@ -407,6 +407,7 @@ def test_api_product_endpoint_load_sboms_action(self):
 
         data = {
             "input_file": ContentFile("{}", name="sbom.json"),
+            "infer_download_urls": True,
             "update_existing_packages": False,
             "scan_all_packages": False,
         }
@@ -436,6 +437,7 @@ def test_api_product_endpoint_import_manifests_action(self):
 
         data = {
             "input_file": ContentFile("Content", name="requirements.txt"),
+            "infer_download_urls": True,
             "update_existing_packages": False,
             "scan_all_packages": False,
         }

product_portfolio/tests/test_models.py

@@ -544,6 +544,45 @@ def test_product_model_is_locked_property(self):
         product = Product.unsecured_objects.get(pk=product.pk)
         self.assertTrue(product.is_locked)
 
+    def test_product_model_improve_packages_from_purl(self):
+        product = make_product(self.dataspace)
+        package1 = make_package(self.dataspace, package_url="pkg:nuget/[email protected]")
+        make_product_package(product, package=package1)
+
+        product.refresh_from_db()
+        updated_packages = product.improve_packages_from_purl()
+        self.assertEqual([package1], updated_packages)
+
+        package1.refresh_from_db()
+        expected_download_url = "https://www.nuget.org/api/v2/package/Azure.Core/1.45.0"
+        self.assertEqual(expected_download_url, package1.download_url)
+
+    @mock.patch("dje.tasks.scancodeio_submit_scan.delay")
+    def test_product_model_scan_all_packages_task(self, mock_scancodeio_submit_scan):
+        product = make_product(self.dataspace)
+        package1 = make_package(self.dataspace, package_url="pkg:nuget/[email protected]")
+        make_product_package(product, package=package1)
+
+        mock_scancodeio_submit_scan.return_value = None
+        product.scan_all_packages_task(user=self.super_user)
+        mock_scancodeio_submit_scan.assert_called_with(
+            uris=[],
+            user_uuid=self.super_user.uuid,
+            dataspace_uuid=self.super_user.dataspace.uuid,
+        )
+        package1.refresh_from_db()
+        self.assertEqual("", package1.download_url)
+
+        expected_download_url = "https://www.nuget.org/api/v2/package/Azure.Core/1.45.0"
+        product.scan_all_packages_task(user=self.super_user, infer_download_urls=True)
+        mock_scancodeio_submit_scan.assert_called_with(
+            uris=[expected_download_url],
+            user_uuid=self.super_user.uuid,
+            dataspace_uuid=self.super_user.dataspace.uuid,
+        )
+        package1.refresh_from_db()
+        self.assertEqual(expected_download_url, package1.download_url)
+
     @mock.patch("component_catalog.models.Package.update_from_purldb")
     def test_product_model_improve_packages_from_purldb(self, mock_update_from_purldb):
         mock_update_from_purldb.return_value = 1