1 files changed, 39 insertions, 2 deletions
diff --git a/spec/lib/gitlab/ssh_public_key_spec.rb b/spec/lib/gitlab/ssh_public_key_spec.rb
index e1a588a4b7d..38486b313cb 100644
--- a/spec/lib/gitlab/ssh_public_key_spec.rb
+++ b/spec/lib/gitlab/ssh_public_key_spec.rb
@@ -21,6 +21,14 @@ RSpec.describe Gitlab::SSHPublicKey, lib: true do
     end
   end
 
+  describe '.supported_types' do
+    it 'returns array with the names of supported technologies' do
+      expect(described_class.supported_types).to eq(
+        [:rsa, :dsa, :ecdsa, :ed25519]
+      )
+    end
+  end
+
   describe '.supported_sizes(name)' do
     where(:name, :sizes) do
       [
@@ -31,14 +39,43 @@ RSpec.describe Gitlab::SSHPublicKey, lib: true do
       ]
     end
 
-    subject { described_class.supported_sizes(name) }
-
     with_them do
       it { expect(described_class.supported_sizes(name)).to eq(sizes) }
       it { expect(described_class.supported_sizes(name.to_s)).to eq(sizes) }
     end
   end
 
+  describe '.supported_algorithms' do
+    it 'returns all supported algorithms' do
+      expect(described_class.supported_algorithms).to eq(
+        %w(
+        ssh-rsa
+        ssh-dss
+        ecdsa-sha2-nistp256 ecdsa-sha2-nistp384 ecdsa-sha2-nistp521
+        ssh-ed25519
+        )
+      )
+    end
+  end
+
+  describe '.supported_algorithms_for_name' do
+    where(:name, :algorithms) do
+      [
+        [:rsa, %w(ssh-rsa)],
+        [:dsa, %w(ssh-dss)],
+        [:ecdsa, %w(ecdsa-sha2-nistp256 ecdsa-sha2-nistp384 ecdsa-sha2-nistp521)],
+        [:ed25519, %w(ssh-ed25519)]
+      ]
+    end
+
+    with_them do
+      it "returns all supported algorithms for #{params[:name]}" do
+        expect(described_class.supported_algorithms_for_name(name)).to eq(algorithms)
+        expect(described_class.supported_algorithms_for_name(name.to_s)).to eq(algorithms)
+      end
+    end
+  end
+
   describe '.sanitize(key_content)' do
     let(:content) { build(:key).key }