1 files changed, 44 insertions, 0 deletions
diff --git a/spec/support/shared_examples/models/ci/metadata_id_tokens_shared_examples.rb b/spec/support/shared_examples/models/ci/metadata_id_tokens_shared_examples.rb
new file mode 100644
index 00000000000..0c71ebe7a4d
--- /dev/null
+++ b/spec/support/shared_examples/models/ci/metadata_id_tokens_shared_examples.rb
@@ -0,0 +1,44 @@
+# frozen_string_literal: true
+
+RSpec.shared_examples_for 'has ID tokens' do |ci_type|
+  subject(:ci) { FactoryBot.build(ci_type) }
+
+  describe 'delegations' do
+    it { is_expected.to delegate_method(:id_tokens).to(:metadata).allow_nil }
+  end
+
+  describe '#id_tokens?' do
+    subject { ci.id_tokens? }
+
+    context 'without metadata' do
+      let(:ci) { FactoryBot.build(ci_type) }
+
+      it { is_expected.to be_falsy }
+    end
+
+    context 'with metadata' do
+      let(:ci) { FactoryBot.build(ci_type, metadata: FactoryBot.build(:ci_build_metadata, id_tokens: id_tokens)) }
+
+      context 'when ID tokens exist' do
+        let(:id_tokens) { { TEST_JOB_JWT: { id_token: { aud: 'developers ' } } } }
+
+        it { is_expected.to be_truthy }
+      end
+
+      context 'when ID tokens do not exist' do
+        let(:id_tokens) { {} }
+
+        it { is_expected.to be_falsy }
+      end
+    end
+  end
+
+  describe '#id_tokens=' do
+    it 'assigns the ID tokens to the CI job' do
+      id_tokens = [{ 'JOB_ID_TOKEN' => { 'id_token' => { 'aud' => 'https://gitlab.test ' } } }]
+      ci.id_tokens = id_tokens
+
+      expect(ci.id_tokens).to match_array(id_tokens)
+    end
+  end
+end