1 files changed, 104 insertions, 0 deletions
diff --git a/spec/models/ci/catalog/components_project_spec.rb b/spec/models/ci/catalog/components_project_spec.rb
new file mode 100644
index 00000000000..d7e0ee2079c
--- /dev/null
+++ b/spec/models/ci/catalog/components_project_spec.rb
@@ -0,0 +1,104 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe Ci::Catalog::ComponentsProject, feature_category: :pipeline_composition do
+  using RSpec::Parameterized::TableSyntax
+
+  let_it_be(:files) do
+    {
+      'templates/secret-detection.yml' => "spec:\n inputs:\n  website:\n---\nimage: alpine_1",
+      'templates/dast/template.yml' => 'image: alpine_2',
+      'templates/template.yml' => 'image: alpine_3',
+      'templates/blank-yaml.yml' => '',
+      'templates/dast/sub-folder/template.yml' => 'image: alpine_4',
+      'tests/test.yml' => 'image: alpine_5',
+      'README.md' => 'Read me'
+    }
+  end
+
+  let_it_be(:project) do
+    create(
+      :project, :custom_repo,
+      description: 'Simple, complex, and other components',
+      files: files
+    )
+  end
+
+  let_it_be(:catalog_resource) { create(:ci_catalog_resource, project: project) }
+
+  let(:components_project) { described_class.new(project, project.default_branch) }
+
+  describe '#fetch_component_paths' do
+    it 'retrieves all the paths for valid components' do
+      paths = components_project.fetch_component_paths(project.default_branch)
+
+      expect(paths).to contain_exactly(
+        'templates/blank-yaml.yml', 'templates/dast/template.yml', 'templates/secret-detection.yml',
+        'templates/template.yml'
+      )
+    end
+  end
+
+  describe '#extract_component_name' do
+    context 'with invalid component path' do
+      it 'raises an error' do
+        expect(components_project.extract_component_name('not-template/this-is-wrong.yml')).to be_nil
+      end
+    end
+
+    context 'with valid component paths' do
+      where(:path, :name) do
+        'templates/secret-detection.yml'         | 'secret-detection'
+        'templates/dast/template.yml'            | 'dast'
+        'templates/template.yml'                 | 'template'
+        'templates/blank-yaml.yml'               | 'blank-yaml'
+      end
+
+      with_them do
+        it 'extracts the component name from the path' do
+          expect(components_project.extract_component_name(path)).to eq(name)
+        end
+      end
+    end
+  end
+
+  describe '#extract_inputs' do
+    context 'with valid inputs' do
+      it 'extracts the inputs from a blob' do
+        blob = "spec:\n inputs:\n  website:\n---\nimage: alpine_1"
+
+        expect(components_project.extract_inputs(blob)).to eq({ website: nil })
+      end
+    end
+
+    context 'with invalid inputs' do
+      it 'raises InvalidFormatError' do
+        blob = "spec:\n inputs:\n  website:\n---\nsome: invalid: string"
+
+        expect do
+          components_project.extract_inputs(blob)
+        end.to raise_error(::Gitlab::Config::Loader::FormatError,
+          /mapping values are not allowed in this context/)
+      end
+    end
+  end
+
+  describe '#fetch_component' do
+    where(:component_name, :content, :path) do
+      'secret-detection' | "spec:\n inputs:\n  website:\n---\nimage: alpine_1" | 'templates/secret-detection.yml'
+      'dast'             | 'image: alpine_2'                                   | 'templates/dast/template.yml'
+      'template'         | 'image: alpine_3'                                   | 'templates/template.yml'
+      'blank-yaml'       | ''                                                  | 'templates/blank-yaml.yml'
+    end
+
+    with_them do
+      it 'fetches the content for a component' do
+        data = components_project.fetch_component(component_name)
+
+        expect(data.path).to eq(path)
+        expect(data.content).to eq(content)
+      end
+    end
+  end
+end