require "sidekiq/testing" require 'rails_helper' require_dependency 'jobs/base' describe Jobs do describe 'enqueue' do describe 'when queue_jobs is true' do before do SiteSetting.expects(:queue_jobs?).at_least_once.returns(true) end it 'enqueues a job in sidekiq' do Sidekiq::Client.expects(:enqueue).with(Jobs::ProcessPost, post_id: 1, current_site_id: 'default') Jobs.enqueue(:process_post, post_id: 1) end it "does not pass current_site_id when 'all_sites' is present" do Sidekiq::Client.expects(:enqueue).with(Jobs::ProcessPost, post_id: 1) Jobs.enqueue(:process_post, post_id: 1, all_sites: true) end it "doesn't execute the job" do Sidekiq::Client.stubs(:enqueue) Jobs::ProcessPost.any_instance.expects(:perform).never Jobs.enqueue(:process_post, post_id: 1) end it "should enqueue with the correct database id when the current_site_id option is given" do Sidekiq::Client.expects(:enqueue).with do |arg1, arg2| arg2[:current_site_id] == 'test_db' && arg2[:sync_exec].nil? end Jobs.enqueue(:process_post, post_id: 1, current_site_id: 'test_db') end end describe 'when queue_jobs is false' do before do SiteSetting.expects(:queue_jobs?).at_least_once.returns(false) end it "doesn't enqueue in sidekiq" do Sidekiq::Client.expects(:enqueue).with(Jobs::ProcessPost, {}).never Jobs.enqueue(:process_post, post_id: 1) end it "executes the job right away" do Jobs::ProcessPost.any_instance.expects(:perform).with(post_id: 1, sync_exec: true, current_site_id: "default") Jobs.enqueue(:process_post, post_id: 1) end context 'and current_site_id option is given and does not match the current connection' do before do Sidekiq::Client.stubs(:enqueue) Jobs::ProcessPost.any_instance.stubs(:execute).returns(true) end it 'should not execute the job' do Jobs::ProcessPost.any_instance.expects(:execute).never Jobs.enqueue(:process_post, post_id: 1, current_site_id: 'test_db') rescue nil end it 'should raise an exception' do expect { Jobs.enqueue(:process_post, post_id: 1, current_site_id: 'test_db') }.to raise_error(ArgumentError) end it 'should not connect to the given database' do RailsMultisite::ConnectionManagement.expects(:establish_connection).never Jobs.enqueue(:process_post, post_id: 1, current_site_id: 'test_db') rescue nil end end end end describe 'cancel_scheduled_job' do it 'deletes the matching job' do SiteSetting.queue_jobs = true Sidekiq::Testing.disable! do scheduled_jobs = Sidekiq::ScheduledSet.new scheduled_jobs.clear expect(scheduled_jobs.size).to eq(0) Jobs.enqueue_in(1.year, :run_heartbeat, topic_id: 123) Jobs.enqueue_in(2.years, :run_heartbeat, topic_id: 456) Jobs.enqueue_in(3.years, :run_heartbeat, topic_id: 123, current_site_id: 'foo') Jobs.enqueue_in(4.years, :run_heartbeat, topic_id: 123, current_site_id: 'bar') expect(scheduled_jobs.size).to eq(4) Jobs.cancel_scheduled_job(:run_heartbeat, topic_id: 123) expect(scheduled_jobs.size).to eq(3) Jobs.cancel_scheduled_job(:run_heartbeat, topic_id: 123, all_sites: true) expect(scheduled_jobs.size).to eq(1) end end end describe 'enqueue_at' do it 'calls enqueue_in for you' do Timecop.freeze(Time.zone.now) do Jobs.expects(:enqueue_in).with(3 * 60 * 60, :eat_lunch, {}).returns(true) Jobs.enqueue_at(3.hours.from_now, :eat_lunch, {}) end end it 'handles datetimes that are in the past' do Timecop.freeze(Time.zone.now) do Jobs.expects(:enqueue_in).with(0, :eat_lunch, {}).returns(true) Jobs.enqueue_at(3.hours.ago, :eat_lunch, {}) end end end end