debian-mirror-gitlab/spec/models/key_spec.rb

142 lines
4.1 KiB
Ruby
Raw Normal View History

2014-09-02 18:07:02 +05:30
require 'spec_helper'
2015-12-23 02:04:40 +05:30
describe Key, models: true do
2017-08-17 22:00:37 +05:30
include EmailHelpers
2014-09-02 18:07:02 +05:30
describe "Associations" do
2015-04-26 12:48:37 +05:30
it { is_expected.to belong_to(:user) }
2014-09-02 18:07:02 +05:30
end
describe "Validation" do
2015-04-26 12:48:37 +05:30
it { is_expected.to validate_presence_of(:title) }
2017-08-17 22:00:37 +05:30
it { is_expected.to validate_length_of(:title).is_at_most(255) }
2015-04-26 12:48:37 +05:30
it { is_expected.to validate_presence_of(:key) }
2017-08-17 22:00:37 +05:30
it { is_expected.to validate_length_of(:key).is_at_most(5000) }
it { is_expected.to allow_value('ssh-foo').for(:key) }
it { is_expected.to allow_value('ecdsa-foo').for(:key) }
it { is_expected.not_to allow_value('foo-bar').for(:key) }
2014-09-02 18:07:02 +05:30
end
describe "Methods" do
2016-09-13 17:45:13 +05:30
let(:user) { create(:user) }
2015-04-26 12:48:37 +05:30
it { is_expected.to respond_to :projects }
2015-09-11 14:41:01 +05:30
it { is_expected.to respond_to :publishable_key }
describe "#publishable_keys" do
2016-09-13 17:45:13 +05:30
it 'replaces SSH key comment with simple identifier of username + hostname' do
2017-08-17 22:00:37 +05:30
expect(build(:key, user: user).publishable_key).to include("#{user.name} (#{Gitlab.config.gitlab.host})")
end
end
describe "#update_last_used_at" do
let(:key) { create(:key) }
context 'when key was not updated during the last day' do
before do
allow_any_instance_of(Gitlab::ExclusiveLease).to receive(:try_obtain).
and_return('000000')
end
it 'enqueues a UseKeyWorker job' do
expect(UseKeyWorker).to receive(:perform_async).with(key.id)
key.update_last_used_at
end
end
context 'when key was updated during the last day' do
before do
allow_any_instance_of(Gitlab::ExclusiveLease).to receive(:try_obtain).
and_return(false)
end
it 'does not enqueue a UseKeyWorker job' do
expect(UseKeyWorker).not_to receive(:perform_async)
key.update_last_used_at
end
2015-09-11 14:41:01 +05:30
end
end
2014-09-02 18:07:02 +05:30
end
2016-06-22 15:30:34 +05:30
context "validation of uniqueness (based on fingerprint uniqueness)" do
2014-09-02 18:07:02 +05:30
let(:user) { create(:user) }
it "accepts the key once" do
2015-04-26 12:48:37 +05:30
expect(build(:key, user: user)).to be_valid
2014-09-02 18:07:02 +05:30
end
it "does not accept the exact same key twice" do
create(:key, user: user)
2015-04-26 12:48:37 +05:30
expect(build(:key, user: user)).not_to be_valid
2014-09-02 18:07:02 +05:30
end
it "does not accept a duplicate key with a different comment" do
create(:key, user: user)
duplicate = build(:key, user: user)
duplicate.key << ' extra comment'
2015-04-26 12:48:37 +05:30
expect(duplicate).not_to be_valid
2014-09-02 18:07:02 +05:30
end
end
context "validate it is a fingerprintable key" do
it "accepts the fingerprintable key" do
2015-04-26 12:48:37 +05:30
expect(build(:key)).to be_valid
2014-09-02 18:07:02 +05:30
end
2015-04-26 12:48:37 +05:30
it 'rejects an unfingerprintable key that contains a space' do
key = build(:key)
# Not always the middle, but close enough
2015-09-11 14:41:01 +05:30
key.key = key.key[0..100] + ' ' + key.key[101..-1]
2015-04-26 12:48:37 +05:30
expect(key).not_to be_valid
2014-09-02 18:07:02 +05:30
end
2015-04-26 12:48:37 +05:30
it 'rejects the unfingerprintable key (not a key)' do
expect(build(:key, key: 'ssh-rsa an-invalid-key==')).not_to be_valid
2014-09-02 18:07:02 +05:30
end
2015-09-11 14:41:01 +05:30
it 'rejects the multiple line key' do
key = build(:key)
2015-12-23 02:04:40 +05:30
key.key.tr!(' ', "\n")
2015-09-11 14:41:01 +05:30
expect(key).not_to be_valid
end
2014-09-02 18:07:02 +05:30
end
context 'callbacks' do
2016-09-13 17:45:13 +05:30
it 'adds new key to authorized_file' do
2017-08-17 22:00:37 +05:30
key = build(:personal_key, id: 7)
expect(GitlabShellWorker).to receive(:perform_async).with(:add_key, key.shell_id, key.key)
key.save!
2014-09-02 18:07:02 +05:30
end
2016-09-13 17:45:13 +05:30
it 'removes key from authorized_file' do
2017-08-17 22:00:37 +05:30
key = create(:personal_key)
expect(GitlabShellWorker).to receive(:perform_async).with(:remove_key, key.shell_id, key.key)
key.destroy
end
end
describe '#key=' do
let(:valid_key) do
"ssh-rsa AAAAB3NzaC1yc2EAAAABJQAAAIEAiPWx6WM4lhHNedGfBpPJNPpZ7yKu+dnn1SJejgt4596k6YjzGGphH2TUxwKzxcKDKKezwkpfnxPkSMkuEspGRt/aZZ9wa++Oi7Qkr8prgHc4soW6NUlfDzpvZK2H5E7eQaSeP3SAwGmQKUFHCddNaP0L+hM7zhFNzjFvpaMgJw0= dummy@gitlab.com"
end
it 'strips white spaces' do
expect(described_class.new(key: " #{valid_key} ").key).to eq(valid_key)
end
end
describe 'notification' do
let(:user) { create(:user) }
it 'sends a notification' do
perform_enqueued_jobs do
create(:key, user: user)
end
should_email(user)
2014-09-02 18:07:02 +05:30
end
end
end