1 files changed, 19 insertions, 0 deletions
diff --git a/spec/tasks/gitlab/shell_rake_spec.rb b/spec/tasks/gitlab/shell_rake_spec.rb
index 52a9738fb51..195859eac70 100644
--- a/spec/tasks/gitlab/shell_rake_spec.rb
+++ b/spec/tasks/gitlab/shell_rake_spec.rb
@@ -22,4 +22,23 @@ RSpec.describe 'gitlab:shell rake tasks', :silence_stdout do
       run_rake_task('gitlab:shell:install')
     end
   end
+
+  describe 'setup task' do
+    it 'writes authorized keys into the file' do
+      allow(Gitlab::CurrentSettings).to receive(:authorized_keys_enabled?).and_return(true)
+      stub_env('force', 'yes')
+
+      auth_key = create(:key)
+      auth_and_signing_key = create(:key, usage_type: :auth_and_signing)
+      create(:key, usage_type: :signing)
+
+      expect_next_instance_of(Gitlab::AuthorizedKeys) do |instance|
+        expect(instance).to receive(:batch_add_keys).once do |keys|
+          expect(keys).to match_array([auth_key, auth_and_signing_key])
+        end
+      end
+
+      run_rake_task('gitlab:shell:setup')
+    end
+  end
 end