spec/workers/ssh_keys/expired_notification_worker_spec.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71

# frozen_string_literal: true

require 'spec_helper'

RSpec.describe SshKeys::ExpiredNotificationWorker, type: :worker, feature_category: :compliance_management do
  subject(:worker) { described_class.new }

  it 'uses a cronjob queue' do
    expect(worker.sidekiq_options_hash).to include(
      'queue_namespace' => :cronjob
    )
  end

  describe '#perform' do
    let_it_be(:user) { create(:user) }

    context 'with a large batch' do
      let_it_be_with_reload(:keys) { create_list(:key, 20, :expired_today, user: user) }

      before do
        stub_const("SshKeys::ExpiredNotificationWorker::BATCH_SIZE", 5)
      end

      it 'updates all keys regardless of batch size' do
        worker.perform

        expect(keys.pluck(:expiry_notification_delivered_at)).not_to include(nil)
      end
    end

    context 'with expiring key today' do
      let_it_be_with_reload(:expired_today) { create(:key, :expired_today, user: user) }

      it 'invoke the notification service' do
        expect_next_instance_of(Keys::ExpiryNotificationService) do |expiry_service|
          expect(expiry_service).to receive(:execute)
        end

        worker.perform
      end

      it 'updates notified column' do
        expect { worker.perform }.to change { expired_today.reload.expiry_notification_delivered_at }
      end

      include_examples 'an idempotent worker' do
        subject do
          perform_multiple(worker: worker)
        end
      end
    end

    context 'when key has expired in the past' do
      let_it_be(:expired_past) { create(:key, :expired, user: user) }

      it 'does not update notified column' do
        expect { worker.perform }.not_to change { expired_past.reload.expiry_notification_delivered_at }
      end

      context 'when key has already been notified of expiration' do
        before do
          expired_past.update_attribute(:expiry_notification_delivered_at, 1.day.ago)
        end

        it 'does not update notified column' do
          expect { worker.perform }.not_to change { expired_past.reload.expiry_notification_delivered_at }
        end
      end
    end
  end
end