2019-04-30 08:27:42 +08:00
|
|
|
# frozen_string_literal: true
|
|
|
|
|
2015-10-11 17:41:23 +08:00
|
|
|
require 'rails_helper'
|
2013-02-06 03:16:51 +08:00
|
|
|
|
2019-10-02 12:01:53 +08:00
|
|
|
describe ::Jobs::Base do
|
|
|
|
class GoodJob < ::Jobs::Base
|
2014-02-21 13:05:19 +08:00
|
|
|
attr_accessor :count
|
|
|
|
def execute(args)
|
|
|
|
self.count ||= 0
|
|
|
|
self.count += 1
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2019-10-02 12:01:53 +08:00
|
|
|
class BadJob < ::Jobs::Base
|
2014-02-21 12:31:15 +08:00
|
|
|
attr_accessor :fail_count
|
|
|
|
|
|
|
|
def execute(args)
|
|
|
|
@fail_count ||= 0
|
|
|
|
@fail_count += 1
|
|
|
|
raise StandardError
|
|
|
|
end
|
|
|
|
end
|
2015-02-10 04:47:46 +08:00
|
|
|
|
2014-02-21 13:05:19 +08:00
|
|
|
it 'handles correct jobs' do
|
|
|
|
job = GoodJob.new
|
|
|
|
job.perform({})
|
2014-12-31 22:55:03 +08:00
|
|
|
expect(job.count).to eq(1)
|
2014-02-21 13:05:19 +08:00
|
|
|
end
|
2014-02-21 12:31:15 +08:00
|
|
|
|
|
|
|
it 'handles errors in multisite' do
|
2017-07-28 09:20:09 +08:00
|
|
|
RailsMultisite::ConnectionManagement.expects(:all_dbs).returns(['default', 'default', 'default'])
|
2014-07-18 04:22:46 +08:00
|
|
|
# one exception per database
|
2015-02-10 04:47:46 +08:00
|
|
|
Discourse.expects(:handle_job_exception).times(3)
|
2014-02-21 12:31:15 +08:00
|
|
|
|
2015-02-10 04:47:46 +08:00
|
|
|
bad = BadJob.new
|
2017-07-28 09:20:09 +08:00
|
|
|
expect { bad.perform({}) }.to raise_error(Jobs::HandledExceptionWrapper)
|
2014-12-31 22:55:03 +08:00
|
|
|
expect(bad.fail_count).to eq(3)
|
2014-02-21 12:31:15 +08:00
|
|
|
end
|
2013-02-06 03:16:51 +08:00
|
|
|
|
|
|
|
it 'delegates the process call to execute' do
|
2019-10-02 12:01:53 +08:00
|
|
|
::Jobs::Base.any_instance.expects(:execute).with('hello' => 'world')
|
|
|
|
::Jobs::Base.new.perform('hello' => 'world', 'sync_exec' => true)
|
2013-02-06 03:16:51 +08:00
|
|
|
end
|
|
|
|
|
|
|
|
it 'converts to an indifferent access hash' do
|
2019-10-02 12:01:53 +08:00
|
|
|
::Jobs::Base.any_instance.expects(:execute).with(instance_of(HashWithIndifferentAccess))
|
|
|
|
::Jobs::Base.new.perform('hello' => 'world', 'sync_exec' => true)
|
2013-02-26 00:42:20 +08:00
|
|
|
end
|
2013-02-06 03:16:51 +08:00
|
|
|
|
|
|
|
end
|