1 files changed, 34 insertions, 0 deletions
diff --git a/spec/lib/gitlab/ci/jwt_v2/claim_mapper/repository_spec.rb b/spec/lib/gitlab/ci/jwt_v2/claim_mapper/repository_spec.rb
new file mode 100644
index 00000000000..0dd0d2fcf0d
--- /dev/null
+++ b/spec/lib/gitlab/ci/jwt_v2/claim_mapper/repository_spec.rb
@@ -0,0 +1,34 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe Gitlab::Ci::JwtV2::ClaimMapper::Repository, feature_category: :continuous_integration do
+  let_it_be(:sha) { '35fa264414ee3ed7d0b8a6f5da40751c8600a772' }
+  let_it_be(:pipeline) { build_stubbed(:ci_pipeline, ref: 'test-branch-for-claim-mapper', sha: sha) }
+
+  let(:url) { 'gitlab.com/gitlab-org/gitlab//.gitlab-ci.yml' }
+  let(:project_config) { instance_double(Gitlab::Ci::ProjectConfig, url: url) }
+
+  subject(:mapper) { described_class.new(project_config, pipeline) }
+
+  describe '#to_h' do
+    it 'returns expected claims' do
+      expect(mapper.to_h).to eq({
+        ci_config_ref_uri: 'gitlab.com/gitlab-org/gitlab//.gitlab-ci.yml@refs/heads/test-branch-for-claim-mapper',
+        ci_config_sha: sha
+      })
+    end
+
+    context 'when ref is a tag' do
+      let_it_be(:tag) { 'test-tag-for-claim-mapper' }
+      let_it_be(:pipeline) { build_stubbed(:ci_pipeline, tag: tag, ref: tag, sha: sha) }
+
+      it 'returns expected claims' do
+        expect(mapper.to_h).to eq({
+          ci_config_ref_uri: 'gitlab.com/gitlab-org/gitlab//.gitlab-ci.yml@refs/tags/test-tag-for-claim-mapper',
+          ci_config_sha: sha
+        })
+      end
+    end
+  end
+end