1 files changed, 60 insertions, 2 deletions

diff --git a/spec/models/concerns/enums/sbom_spec.rb b/spec/models/concerns/enums/sbom_spec.rb
index e2f56cc637d..3bbdf619a8c 100644
--- a/spec/models/concerns/enums/sbom_spec.rb
+++ b/spec/models/concerns/enums/sbom_spec.rb
@@ -3,9 +3,9 @@
 require "spec_helper"
 
 RSpec.describe Enums::Sbom, feature_category: :dependency_management do
-  describe '.purl_types' do
-    using RSpec::Parameterized::TableSyntax
+  using RSpec::Parameterized::TableSyntax
 
+  describe '.purl_types' do
     subject(:actual_purl_type) { described_class.purl_types[package_manager] }
 
     where(:given_package_manager, :expected_purl_type) do
@@ -35,5 +35,63 @@ RSpec.describe Enums::Sbom, feature_category: :dependency_management do
         expect(actual_purl_type).to eql(expected_purl_type)
       end
     end
+
+    it 'contains all of the dependency scanning and container scanning purl types' do
+      expect(described_class::DEPENDENCY_SCANNING_PURL_TYPES + described_class::CONTAINER_SCANNING_PURL_TYPES)
+        .to eql(described_class::PURL_TYPES.keys)
+    end
+  end
+
+  describe '.dependency_scanning_purl_type?' do
+    where(:purl_type, :expected) do
+      :composer  | false
+      'composer' | true
+      'conan'    | true
+      'gem'      | true
+      'golang'   | true
+      'maven'    | true
+      'npm'      | true
+      'nuget'    | true
+      'pypi'     | true
+      'unknown'  | false
+      'apk'      | false
+      'rpm'      | false
+      'deb'      | false
+      'wolfi'    | false
+    end
+
+    with_them do
+      it 'returns true if the purl_type is for dependency_scanning' do
+        actual = described_class.dependency_scanning_purl_type?(purl_type)
+        expect(actual).to eql(expected)
+      end
+    end
+  end
+
+  describe '.container_scanning_purl_type?' do
+    where(:purl_type, :expected) do
+      'composer'    | false
+      'conan'       | false
+      'gem'         | false
+      'golang'      | false
+      'maven'       | false
+      'npm'         | false
+      'nuget'       | false
+      'pypi'        | false
+      'unknown'     | false
+      :apk          | false
+      'apk'         | true
+      'rpm'         | true
+      'deb'         | true
+      'cbl-mariner' | true
+      'wolfi'       | true
+    end
+
+    with_them do
+      it 'returns true if the purl_type is for container_scanning' do
+        actual = described_class.container_scanning_purl_type?(purl_type)
+        expect(actual).to eql(expected)
+      end
+    end
   end
 end