diff options
Diffstat (limited to 'spec/lib/gitlab/auth/key_status_checker_spec.rb')
-rw-r--r-- | spec/lib/gitlab/auth/key_status_checker_spec.rb | 68 |
1 files changed, 68 insertions, 0 deletions
diff --git a/spec/lib/gitlab/auth/key_status_checker_spec.rb b/spec/lib/gitlab/auth/key_status_checker_spec.rb new file mode 100644 index 00000000000..b1a540eae81 --- /dev/null +++ b/spec/lib/gitlab/auth/key_status_checker_spec.rb @@ -0,0 +1,68 @@ +# frozen_string_literal: true + +require 'spec_helper' + +describe Gitlab::Auth::KeyStatusChecker do + let_it_be(:never_expires_key) { build(:personal_key, expires_at: nil) } + let_it_be(:expired_key) { build(:personal_key, expires_at: 3.days.ago) } + let_it_be(:expiring_soon_key) { build(:personal_key, expires_at: 3.days.from_now) } + let_it_be(:expires_in_future_key) { build(:personal_key, expires_at: 14.days.from_now) } + + let(:key_status_checker) { described_class.new(key) } + + describe '#show_console_message?' do + subject { key_status_checker.show_console_message? } + + context 'for an expired key' do + let(:key) { expired_key } + + it { is_expected.to eq(true) } + end + + context 'for a key expiring in the next 7 days' do + let(:key) { expiring_soon_key } + + it { is_expected.to eq(true) } + end + + context 'for a key expiring after the next 7 days' do + let(:key) { expires_in_future_key } + + it { is_expected.to eq(false) } + end + + context 'for a key that never expires' do + let(:key) { never_expires_key } + + it { is_expected.to eq(false) } + end + end + + describe '#console_message' do + subject { key_status_checker.console_message } + + context 'for an expired key' do + let(:key) { expired_key } + + it { is_expected.to eq('INFO: Your SSH key has expired. Please generate a new key.') } + end + + context 'for a key expiring in the next 7 days' do + let(:key) { expiring_soon_key } + + it { is_expected.to eq('INFO: Your SSH key is expiring soon. Please generate a new key.') } + end + + context 'for a key expiring after the next 7 days' do + let(:key) { expires_in_future_key } + + it { is_expected.to be_nil } + end + + context 'for a key that never expires' do + let(:key) { never_expires_key } + + it { is_expected.to be_nil } + end + end +end |