1 files changed, 90 insertions, 0 deletions
diff --git a/activejob/test/support/integration/adapters/sneakers.rb b/activejob/test/support/integration/adapters/sneakers.rb
new file mode 100644
index 0000000000..f21bb38a32
--- /dev/null
+++ b/activejob/test/support/integration/adapters/sneakers.rb
@@ -0,0 +1,90 @@
+require 'sneakers/runner'
+require 'sneakers/publisher'
+require 'timeout'
+
+module Sneakers
+  class Publisher
+    def safe_ensure_connected
+      @mutex.synchronize do
+        ensure_connection! unless connected?
+      end
+    end
+  end
+end
+
+
+module SneakersJobsManager
+  def setup
+    ActiveJob::Base.queue_adapter = :sneakers
+    Sneakers.configure  :heartbeat => 2,
+                        :amqp => 'amqp://guest:guest@localhost:5672',
+                        :vhost => '/',
+                        :exchange => 'active_jobs_sneakers_int_test',
+                        :exchange_type => :direct,
+                        :daemonize => true,
+                        :threads => 1,
+                        :workers => 1,
+                        :pid_path => Rails.root.join("tmp/sneakers.pid").to_s,
+                        :log => Rails.root.join("log/sneakers.log").to_s
+    unless can_run?
+      puts "Cannot run integration tests for sneakers. To be able to run integration tests for sneakers you need to install and start rabbitmq.\n"
+      exit
+    end
+  end
+
+  def clear_jobs
+    bunny_queue.purge
+  end
+
+  def start_workers
+    @pid = fork do
+      queues = %w(integration_tests)
+      workers = queues.map do |q|
+        worker_klass = "ActiveJobWorker"+Digest::MD5.hexdigest(q)
+        Sneakers.const_set(worker_klass, Class.new(ActiveJob::QueueAdapters::SneakersAdapter::JobWrapper) do
+          from_queue q
+        end)
+      end
+      Sneakers::Runner.new(workers).run
+    end
+    begin
+      Timeout.timeout(10) do
+        while bunny_queue.status[:consumer_count] == 0
+          sleep 0.5
+        end
+      end
+    rescue Timeout::Error
+      stop_workers
+      raise "Failed to start sneakers worker"
+    end
+  end
+
+  def stop_workers
+    Process.kill 'TERM', @pid
+    Process.kill 'TERM', File.open(Rails.root.join("tmp/sneakers.pid").to_s).read.to_i
+  rescue
+  end
+
+  def can_run?
+    begin
+      bunny_publisher
+    rescue => e
+      return false
+    end
+    true
+  end
+
+  protected
+    def bunny_publisher
+      @bunny_publisher ||= begin
+        p = ActiveJob::QueueAdapters::SneakersAdapter::JobWrapper.send(:publisher)
+        p.safe_ensure_connected
+        p
+      end
+    end
+
+    def bunny_queue
+      @queue ||= bunny_publisher.exchange.channel.queue "integration_tests", durable: true
+    end
+
+end