1 files changed, 34 insertions, 0 deletions
diff --git a/spec/lib/gitlab/grape_logging/loggers/token_logger_spec.rb b/spec/lib/gitlab/grape_logging/loggers/token_logger_spec.rb
new file mode 100644
index 00000000000..d2022a28a90
--- /dev/null
+++ b/spec/lib/gitlab/grape_logging/loggers/token_logger_spec.rb
@@ -0,0 +1,34 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe Gitlab::GrapeLogging::Loggers::TokenLogger do
+  subject { described_class.new }
+
+  describe ".parameters" do
+    let(:token_id) { 1 }
+    let(:token_type) { "PersonalAccessToken" }
+
+    describe 'when no token information is available' do
+      let(:mock_request) { instance_double(ActionDispatch::Request, 'env', env: {}) }
+
+      it 'returns an empty hash' do
+        expect(subject.parameters(mock_request, nil)).to eq({})
+      end
+    end
+
+    describe 'when token information is available' do
+      let(:mock_request) do
+        instance_double(ActionDispatch::Request, 'env',
+          env: {
+            'gitlab.api.token' => { 'token_id': token_id, 'token_type': token_type }
+          }
+        )
+      end
+
+      it 'adds the token information to log parameters' do
+        expect(subject.parameters(mock_request, nil)).to eq( { 'token_id': 1, 'token_type': "PersonalAccessToken" })
+      end
+    end
+  end
+end