From ee9fc120242b83c64bf77af1b72082ac76319332 Mon Sep 17 00:00:00 2001
From: Kir Shatrov <shatrov@me.com>
Date: Wed, 26 Sep 2018 17:25:20 +0100
Subject: Make AJ::Base#enqueue return false if the job wasn't enqueued

---
 activejob/test/cases/callbacks_test.rb          | 29 +++++++++++++++++++++++++
 activejob/test/jobs/abort_before_enqueue_job.rb |  9 ++++++++
 2 files changed, 38 insertions(+)
 create mode 100644 activejob/test/jobs/abort_before_enqueue_job.rb

(limited to 'activejob/test')

diff --git a/activejob/test/cases/callbacks_test.rb b/activejob/test/cases/callbacks_test.rb
index df6ce16858..267d58a7c8 100644
--- a/activejob/test/cases/callbacks_test.rb
+++ b/activejob/test/cases/callbacks_test.rb
@@ -2,6 +2,7 @@
 
 require "helper"
 require "jobs/callback_job"
+require "jobs/abort_before_enqueue_job"
 
 require "active_support/core_ext/object/inclusion"
 
@@ -22,4 +23,32 @@ class CallbacksTest < ActiveSupport::TestCase
     assert "CallbackJob ran around_enqueue_start".in? enqueued_callback_job.history
     assert "CallbackJob ran around_enqueue_stop".in? enqueued_callback_job.history
   end
+
+  test "#enqueue returns false when before_enqueue aborts callback chain and return_false_on_aborted_enqueue = true" do
+    begin
+      prev = ActiveJob::Base.return_false_on_aborted_enqueue
+      ActiveJob::Base.return_false_on_aborted_enqueue = true
+      assert_equal false, AbortBeforeEnqueueJob.new.enqueue
+    ensure
+      ActiveJob::Base.return_false_on_aborted_enqueue = prev
+    end
+  end
+
+  test "#enqueue returns self when before_enqueue aborts callback chain and return_false_on_aborted_enqueue = false" do
+    begin
+      prev = ActiveJob::Base.return_false_on_aborted_enqueue
+      ActiveJob::Base.return_false_on_aborted_enqueue = false
+      job = AbortBeforeEnqueueJob.new
+      assert_deprecated do
+        assert_equal job, job.enqueue
+      end
+    ensure
+      ActiveJob::Base.return_false_on_aborted_enqueue = prev
+    end
+  end
+
+  test "#enqueue returns self when the job was enqueued" do
+    job = CallbackJob.new
+    assert_equal job, job.enqueue
+  end
 end
diff --git a/activejob/test/jobs/abort_before_enqueue_job.rb b/activejob/test/jobs/abort_before_enqueue_job.rb
new file mode 100644
index 0000000000..fd278eccf4
--- /dev/null
+++ b/activejob/test/jobs/abort_before_enqueue_job.rb
@@ -0,0 +1,9 @@
+# frozen_string_literal: true
+
+class AbortBeforeEnqueueJob < ActiveJob::Base
+  before_enqueue { throw(:abort) }
+
+  def perform
+    raise "This should never be called"
+  end
+end
-- 
cgit v1.2.3