Rspec 测试 retry_on

Question

我有一份这样的工作

class CrawlSsbHistory < BaseJob
    retry_on(Ssb::SessionExpired) do
      Ssb::Login.call
    end

    def perform
      response = Ssb::Client.get_data
      SsbHistory.last.destroy
    end
  end

我有这样的测试

it "retries the job if session error" do
      allow(Ssb::Client).to receive(:get_data).and_raise(Ssb::SessionExpired)
      allow(Ssb::Login).to receive(:call)
      described_class.perform_now # it is CrawlSsbHistory
      expect(Ssb::Login).to have_received(:call)
    end

CrawlSsbHistory 是抓取一些数据的工作。 它调用 Ssb::Client.get_data 来获取数据。

在 Ssb::Client.get_data 中，如果会话过期，我会引发 Ssb::SessionExpired。 这样我就可以使用 retry_on 捕获作业中引发的错误。 那么如果发生了，我想尝试一下这份工作。

但我有这样的错误

(Ssb::Login (class)).call(*(any args))
           expected: 1 time with any arguments
           received: 0 times with any arguments

作业没有调用 retry_on 吗？ 还是我测试错了？ 如何制作 rspec 来测试 retry_on 是否正常工作并且调用了 Ssb::Login.call？

Answer 1

retry_on不会在每次重试时立即调用该块。 只有当尝试用尽时。 否则，它只会重新安排作业。

从文档：

如果自定义逻辑的重试尝试失败，您还可以传递一个将被调用的块，而不是让异常冒泡。 这个块是以作业实例作为第一个参数和错误实例作为第二个参数产生的。

Answer 2

我假设您正在测试 ActiveJob，我猜retry_on会将作业排入队列而不是立即执行，因此您可以尝试使用ActiveJob::TestHelper设置作业队列

你的测试用例应该是：

require 'rails_helper'
RSpec.describe CrawlSsbJob, type: :job do
 include ActiveJob::TestHelper
 before(:all) do
   ActiveJob::Base.queue_adapter = :test
 end

 it "retries the job if session error" do
   allow(Ssb::Client).to receive(:get_data).and_raise(Ssb::SessionExpired)
   expect(Ssb::Login).to receive(:call)
   # enqueue job
   perform_enqueued_jobs {
    described_class.perform_later # or perform_now
   end
 end
end