1 files changed, 11 insertions, 1 deletions
diff --git a/spec/lib/gitlab/ci/jwt_v2_spec.rb b/spec/lib/gitlab/ci/jwt_v2_spec.rb
index 33aaa145a39..5eeab658a8e 100644
--- a/spec/lib/gitlab/ci/jwt_v2_spec.rb
+++ b/spec/lib/gitlab/ci/jwt_v2_spec.rb
@@ -7,6 +7,8 @@ RSpec.describe Gitlab::Ci::JwtV2 do
   let(:project) { build_stubbed(:project, namespace: namespace) }
   let(:user) { build_stubbed(:user) }
   let(:pipeline) { build_stubbed(:ci_pipeline, ref: 'auto-deploy-2020-03-19') }
+  let(:aud) { described_class::DEFAULT_AUD }
+
   let(:build) do
     build_stubbed(
       :ci_build,
@@ -16,7 +18,7 @@ RSpec.describe Gitlab::Ci::JwtV2 do
     )
   end
 
-  subject(:ci_job_jwt_v2) { described_class.new(build, ttl: 30) }
+  subject(:ci_job_jwt_v2) { described_class.new(build, ttl: 30, aud: aud) }
 
   it { is_expected.to be_a Gitlab::Ci::Jwt }
 
@@ -30,5 +32,13 @@ RSpec.describe Gitlab::Ci::JwtV2 do
         expect(payload[:sub]).to eq("project_path:#{project.full_path}:ref_type:branch:ref:#{pipeline.source_ref}")
       end
     end
+
+    context 'when given an aud' do
+      let(:aud) { 'AWS' }
+
+      it 'uses that aud in the payload' do
+        expect(payload[:aud]).to eq('AWS')
+      end
+    end
   end
 end