aboutsummaryrefslogtreecommitdiffstats
path: root/activejob/test/support/integration/adapters/que.rb
blob: ba7657a42a0e6a72aa50e6f6ecf2214b2654aba3 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
module QueJobsManager
  def setup
    require 'sequel'
    ActiveJob::Base.queue_adapter = :que
    que_url = ENV['QUE_DATABASE_URL'] || 'postgres:///active_jobs_que_int_test'
    uri = URI.parse(que_url)
    user = uri.user||ENV['USER']
    pass = uri.password
    db   = uri.path[1..-1]
    %x{#{"PGPASSWORD=\"#{pass}\"" if pass} psql -c 'drop database if exists "#{db}"' -U #{user} -t template1}
    %x{#{"PGPASSWORD=\"#{pass}\"" if pass} psql -c 'create database "#{db}"' -U #{user} -t template1}
    Que.connection = Sequel.connect(que_url)
    Que.migrate!
    Que.mode = :off
    Que.worker_count = 1
  rescue Sequel::DatabaseConnectionError
    puts "Cannot run integration tests for que. To be able to run integration tests for que you need to install and start postgresql.\n"
    exit
  end

  def clear_jobs
    Que.clear!
  end

  def start_workers
    @thread = Thread.new do
      loop do
        Que::Job.work("integration_tests")
        sleep 0.5
      end
    end
  end

  def stop_workers
    @thread.kill
  end
end