rabbit_jobs 0.0.1 → 0.0.3

data/.gitignore

@@ -3,6 +3,7 @@
 .bundle
 .config
 .yardoc
+.DS_Store
 Gemfile.lock
 InstalledFiles
 _yardoc

data/.rspec

@@ -0,0 +1,2 @@
+--colour
+--format documentation

data/Gemfile

@@ -1,4 +1,13 @@
-source 'https://rubygems.org'
+source :rubygems
 
-# Specify your gem's dependencies in rabbit_jobs.gemspec
 gemspec
+
+gem 'rabbit_jobs', :path => './'
+
+group :development do
+  gem 'rspec', '~> 2.8'
+  gem 'rr'
+  gem 'autotest'
+  # gem 'fs-event'
+  gem 'simplecov', require: false
+end

data/examples/configuration.rb

@@ -0,0 +1,22 @@
+# -*- encoding : utf-8 -*-
+require 'bundler/setup'
+require 'rabbit_jobs'
+require 'json'
+
+RabbitJobs.configure do |c|
+  c.host        "127.0.0.1"
+
+  c.exchange 'test_exchange', durable: true, auto_delete: false
+
+  c.queue 'rabbit_jobs_test1', durable: true, auto_delete: false, ack: true, arguments: {'x-ha-policy' => 'all'}
+  c.queue 'rabbit_jobs_test2', durable: true, auto_delete: false, ack: true, arguments: {'x-ha-policy' => 'all'}
+  c.queue 'rabbit_jobs_test3', durable: true, auto_delete: false, ack: true, arguments: {'x-ha-policy' => 'all'}
+end
+
+puts JSON.pretty_generate(RabbitJobs.config.to_hash)
+
+puts JSON.pretty_generate(RabbitJobs.config.queues)
+
+# 10.times {
+  RabbitJobs.enqueue_to('rabbit_jobs_test1', Integer, 'to_s')
+# }

data/lib/rabbit_jobs/amqp_helpers.rb

@@ -0,0 +1,45 @@
+# -*- encoding : utf-8 -*-
+
+module RabbitJobs
+  module AmqpHelpers
+
+    # Calls given block with initialized amqp
+
+    def amqp_with_exchange(&block)
+      raise ArgumentError unless block
+
+      AMQP.start(host: RabbitJobs.config.host) do |connection|
+        channel  = AMQP::Channel.new(connection)
+
+        channel.on_error do |ch, channel_close|
+          puts "Channel-level error: #{channel_close.reply_text}, shutting down..."
+          connection.close { EM.stop }
+        end
+
+        exchange = channel.direct(RabbitJobs.config[:exchange], RabbitJobs.config[:exchange_params])
+
+        # go work
+        block.call(connection, exchange)
+      end
+    end
+
+    def amqp_with_queue(routing_key, &block)
+
+      raise ArgumentError unless routing_key && block
+
+      amqp_with_exchange do |connection, exchange|
+        queue = exchange.channel.queue(RabbitJobs.config.queue_name(routing_key), RabbitJobs.config[:queues][routing_key])
+        queue.bind(exchange, :routing_key => routing_key)
+
+        # go work
+        block.call(connection, queue)
+      end
+    end
+
+    def make_queue(exchange, routing_key)
+      queue = exchange.channel.queue(RabbitJobs.config.queue_name(routing_key), RabbitJobs.config[:queues][routing_key])
+      queue.bind(exchange, :routing_key => routing_key)
+      queue
+    end
+  end
+end

data/lib/rabbit_jobs/configuration.rb

@@ -0,0 +1,122 @@
+# -*- encoding : utf-8 -*-
+require 'yaml'
+
+module RabbitJobs
+
+  extend self
+
+  def configure(&block)
+    @@configuration ||= Configuration.new
+    block.call(@@configuration)
+  end
+
+  def config
+    @@configuration ||= load_config
+  end
+
+  def load_config
+    self.configure do |c|
+      c.host 'localhost'
+      c.exchange 'rabbit_jobs', auto_delete: false, durable: true
+      c.queue 'default', auto_delete: false, ack: true, durable: true
+    end
+    @@configuration
+  end
+
+  class Configuration
+    include Helpers
+
+    DEFAULT_QUEUE_PARAMS = {
+      auto_delete: false,
+      durable: true,
+      ack: true
+    }
+
+    DEFAULT_EXCHANGE_PARAMS = {
+      auto_delete: false,
+      durable: true
+    }
+
+    DEFAULT_MESSAGE_PARAMS = {
+      persistent: true,
+      nowait: false,
+      immediate: false
+    }
+
+    def to_hash
+      @data.dup
+    end
+
+    def initialize
+      @data = {
+        host: 'localhost',
+        exchange: 'rabbit_jobs',
+        exchange_params: DEFAULT_EXCHANGE_PARAMS,
+        queues: {}
+      }
+    end
+
+    def [](name)
+      @data[name]
+    end
+
+    def host(value = nil)
+      if value
+        raise ArgumentError unless value.is_a?(String) && value != ""
+        @data[:host] = value.to_s
+      else
+        @data[:host]
+      end
+    end
+
+    def exchange(value = nil, params = {})
+      if value
+        raise ArgumentError unless value.is_a?(String) && value != ""
+        @data[:exchange] = value
+        @data[:exchange_params] = DEFAULT_EXCHANGE_PARAMS.merge(params)
+      else
+        @data[:exchange]
+      end
+    end
+
+    def queue(name, params = {})
+      raise ArgumentError.new("name is #{name.inspect}") unless name && name.is_a?(String) && name != ""
+      raise ArgumentError.new("params is #{params.inspect}") unless params && params.is_a?(Hash)
+
+      if @data[:queues][name]
+        @data[:queues][name].merge!(params)
+      else
+        @data[:queues][name] = DEFAULT_QUEUE_PARAMS.merge(params)
+      end
+    end
+
+    def routing_keys
+      @data[:queues].keys
+    end
+
+    def queue_name(routing_key)
+      [@data[:exchange], routing_key].join('#')
+    end
+
+    def load_file(filename)
+      load_yaml(File.read(filename))
+    end
+
+    def load_yaml(text)
+      convert_yaml_config(YAML.load(text))
+    end
+
+    def convert_yaml_config(yaml)
+      if yaml['rabbit_jobs']
+        convert_yaml_config(yaml['rabbit_jobs'])
+      else
+        @data = {host: nil, exchange: nil, queues: {}}
+        host yaml['host']
+        exchange yaml['exchange'], symbolize_keys!(yaml['exchange_params'])
+        yaml['queues'].each do |name, params|
+          queue name, symbolize_keys!(params) || {}
+        end
+      end
+    end
+  end
+end

data/lib/rabbit_jobs/helpers.rb

@@ -0,0 +1,53 @@
+# -*- encoding : utf-8 -*-
+
+module RabbitJobs
+  module Helpers
+    def symbolize_keys!(hash)
+      hash.inject({}) do |options, (key, value)|
+        options[(key.to_sym rescue key) || key] = value
+        options
+      end
+    end
+
+
+    # Tries to find a constant with the name specified in the argument string:
+    #
+    # constantize("Module") # => Module
+    # constantize("Test::Unit") # => Test::Unit
+    #
+    # The name is assumed to be the one of a top-level constant, no matter
+    # whether it starts with "::" or not. No lexical context is taken into
+    # account:
+    #
+    # C = 'outside'
+    # module M
+    #   C = 'inside'
+    #   C # => 'inside'
+    #   constantize("C") # => 'outside', same as ::C
+    # end
+    #
+    # NameError is raised when the constant is unknown.
+    def constantize(camel_cased_word)
+      camel_cased_word = camel_cased_word.to_s
+
+      if camel_cased_word.include?('-')
+        camel_cased_word = classify(camel_cased_word)
+      end
+
+      names = camel_cased_word.split('::')
+      names.shift if names.empty? || names.first.empty?
+
+      constant = Object
+      names.each do |name|
+        args = Module.method(:const_get).arity != 1 ? [false] : []
+
+        if constant.const_defined?(name, *args)
+          constant = constant.const_get(name)
+        else
+          constant = constant.const_missing(name)
+        end
+      end
+      constant
+    end
+  end
+end

data/lib/rabbit_jobs/job.rb

@@ -0,0 +1,43 @@
+# -*- encoding : utf-8 -*-
+require 'json'
+
+module RabbitJobs
+  class Job
+    include RabbitJobs::Helpers
+    include Logger
+
+    attr_accessor :params, :klass, :child
+
+    def initialize(payload)
+      begin
+        self.params = JSON.parse(payload)
+        klass_name = params.delete_at(0)
+        self.klass = constantize(klass_name)
+      rescue
+        log "JOB INIT ERROR at #{Time.now.to_s}:"
+        log $!.inspect
+        log $!.backtrace
+        log "message: #{payload.inspect}"
+        # Mailer.send(klass_name, params, $!)
+      end
+    end
+
+    def perform
+      if @child = fork
+        srand # Reseeding
+        log! "Forked #{@child} at #{Time.now} to process #{klass}.perform(#{ params.map(&:inspect).join(', ') })"
+        Process.wait(@child)
+        yield if block_given?
+      else
+        begin
+          # log 'before perform'
+          klass.perform(*params)
+          # log 'after perform'
+        rescue
+          puts $!.inspect
+        end
+        exit!
+      end
+    end
+  end
+end

data/lib/rabbit_jobs/logger.rb

@@ -0,0 +1,18 @@
+module RabbitJobs
+  module Logger
+    extend self
+
+    def log(string)
+      puts string
+    end
+
+    def log!(string)
+      @@verbose ||= false
+      log(string) if RabbitJobs::Logger.verbose
+    end
+
+    class << self
+      attr_accessor :verbose
+    end
+  end
+end

data/lib/rabbit_jobs/publisher.rb

@@ -0,0 +1,58 @@
+# -*- encoding : utf-8 -*-
+
+require 'json'
+require 'amqp'
+require 'eventmachine'
+
+module RabbitJobs
+  module Publisher
+    extend self
+    extend AmqpHelpers
+
+    def enqueue(klass, *params)
+      key = RabbitJobs.config.routing_keys.first
+      enqueue_to(key, klass, *params)
+    end
+
+    def enqueue_to(routing_key, klass, *params)
+      raise ArgumentError unless klass && routing_key
+
+      payload = ([klass.to_s] + params).to_json
+
+      amqp_with_exchange do |connection, exchange|
+
+        queue = make_queue(exchange, routing_key)
+
+        exchange.publish(payload, Configuration::DEFAULT_MESSAGE_PARAMS.merge({routing_key: routing_key})) {
+          connection.close { EM.stop }
+        }
+      end
+    end
+
+    # def spam
+    #   payload = ([klass.to_s] + params).to_json
+
+    #   amqp_with_exchange do |connection, exchange|
+    #     10000.times { |i| RabbitJobs.enqueue(RabbitJobs::TestJob, i) }
+    #     exchange.publish(payload, RabbitJobs.config.publish_params.merge({routing_key: routing_key})) {
+    #       connection.close { EM.stop }
+    #     }
+    #   end
+    # end
+
+    def purge_queue(routing_key)
+      raise ArgumentError unless routing_key
+
+      amqp_with_queue(routing_key) do |connection, queue|
+        queue.status do |number_of_messages, number_of_consumers|
+          queue.purge {
+            connection.close {
+              EM.stop
+              return number_of_messages
+            }
+          }
+        end
+      end
+    end
+  end
+end

data/lib/rabbit_jobs/tasks.rb

@@ -0,0 +1,57 @@
+# require 'resque/tasks'
+# will give you the resque tasks
+
+namespace :rj do
+  task :setup
+
+  desc "Start a Rabbit Jobs worker"
+  task :work => [ :preload, :setup ] do
+    require 'rabbit_jobs'
+
+    queues = (ENV['QUEUES'] || ENV['QUEUE']).to_s.split(',')
+
+    begin
+      worker = RabbitJobs::Worker.new(*queues)
+      RabbitJobs::Logger.verbose = true if ENV['VERBOSE']
+      # worker.very_verbose = ENV['VVERBOSE']
+    end
+
+    if ENV['BACKGROUND']
+      Process.daemon(true)
+    end
+
+    if ENV['PIDFILE']
+      File.open(ENV['PIDFILE'], 'w') { |f| f << worker.pid }
+    end
+
+    # worker.log "Starting worker #{worker.pid}"
+    # worker.verbose = true
+    worker.work 1
+    # worker.work(ENV['INTERVAL'] || 5) # interval, will block
+  end
+
+  desc "Start multiple Resque workers. Should only be used in dev mode."
+  task :workers do
+    threads = []
+
+    ENV['COUNT'].to_i.times do
+      threads << Thread.new do
+        system "rake resque:work"
+      end
+    end
+
+    threads.each { |thread| thread.join }
+  end
+
+  # Preload app files if this is Rails
+  task :preload => :setup do
+    if defined?(Rails) && Rails.respond_to?(:application)
+      # Rails 3
+      Rails.application.eager_load!
+    elsif defined?(Rails::Initializer)
+      # Rails 2.3
+      $rails_rake_task = false
+      Rails::Initializer.run :load_application_classes
+    end
+  end
+end

data/lib/rabbit_jobs/version.rb

@@ -1,3 +1,3 @@
 module RabbitJobs
-  VERSION = "0.0.1"
+  VERSION = "0.0.3"
 end

data/lib/rabbit_jobs/worker.rb

@@ -0,0 +1,104 @@
+# -*- encoding : utf-8 -*-
+
+module RabbitJobs
+  class Worker
+    include AmqpHelpers
+    include Logger
+
+    # Workers should be initialized with an array of string queue
+    # names. The order is important: a Worker will check the first
+    # queue given for a job. If none is found, it will check the
+    # second queue name given. If a job is found, it will be
+    # processed. Upon completion, the Worker will again check the
+    # first queue given, and so forth. In this way the queue list
+    # passed to a Worker on startup defines the priorities of queues.
+    #
+    # If passed a single "*", this Worker will operate on all queues
+    # in alphabetical order. Queues can be dynamically added or
+    # removed without needing to restart workers using this method.
+    def initialize(*queues)
+      @queues = queues.map { |queue| queue.to_s.strip }.flatten.uniq
+      if @queues == ['*'] || @queues.empty?
+        @queues = RabbitJobs.config.routing_keys
+      end
+    end
+
+    def queues
+      @queues || ['default']
+    end
+
+    # Subscribes to channel and working on jobs
+    def work(time = 10)
+      startup
+
+      processed_count = 0
+      amqp_with_exchange do |connection, exchange|
+        exchange.channel.prefetch(1)
+
+        check_shutdown = Proc.new {
+          if @shutdown
+            log "Processed jobs: #{processed_count}"
+            log "Stopping worker..."
+            connection.close { EM.stop { exit! } }
+          end
+        }
+
+        queues.each do |routing_key|
+          queue = make_queue(exchange, routing_key)
+
+          log "Worker ##{Process.pid} <= #{exchange.name}##{routing_key}"
+
+          queue.subscribe(ack: true) do |metadata, payload|
+            @job = Job.new(payload)
+            @job.perform
+            metadata.ack
+            processed_count += 1
+            check_shutdown.call
+          end
+        end
+
+        EM.add_timer(time) do
+          self.shutdown
+        end
+
+        EM.add_periodic_timer(1) do
+          check_shutdown.call
+        end
+      end
+    end
+
+    def shutdown
+      @shutdown = true
+    end
+
+    def startup
+      # prune_dead_workers
+
+      # Fix buffering so we can `rake rj:work > resque.log` and
+      # get output from the child in there.
+      $stdout.sync = true
+
+      @shutdown = false
+
+      Signal.trap('TERM') { shutdown }
+      Signal.trap('INT')  { shutdown! }
+    end
+
+    def shutdown!
+      shutdown
+      kill_child
+    end
+
+    def kill_child
+      if @job && @job.child
+        # log! "Killing child at #{@child}"
+        if Kernel.system("ps -o pid,state -p #{@job.child}")
+          Process.kill("KILL", @job.child) rescue nil
+        else
+          # log! "Child #{@child} not found, restarting."
+          # shutdown
+        end
+      end
+    end
+  end
+end

data/lib/rabbit_jobs.rb

@@ -1,5 +1,36 @@
-require "rabbit_jobs/version"
+# -*- encoding : utf-8 -*-
+
+require 'rabbit_jobs/version'
+
+require 'rabbit_jobs/helpers'
+require 'rabbit_jobs/amqp_helpers'
+require 'rabbit_jobs/configuration'
+require 'rabbit_jobs/logger'
+
+require 'rabbit_jobs/job'
+require 'rabbit_jobs/publisher'
+require 'rabbit_jobs/worker'
 
 module RabbitJobs
-  # Your code goes here...
-end
+  extend self
+
+  def enqueue(klass, *params)
+    RabbitJobs::Publisher.enqueue(klass, *params)
+  end
+
+  def enqueue_to(routing_key, klass, *params)
+    RabbitJobs::Publisher.enqueue_to(routing_key, klass, *params)
+  end
+
+  class TestJob < RabbitJobs::Job
+    def self.perform(*params)
+      # puts "processing in job: " + params.inspect
+      # sleep 0.1
+
+      # if rand(3) == 0
+      #   puts "ERROR TEXT"
+      #   raise "ERROR TEXT"
+      # end
+    end
+  end
+end

data/lib/tasks/rabbit_jobs.rake

@@ -0,0 +1,3 @@
+
+$LOAD_PATH.unshift File.dirname(__FILE__) + '/../../lib'
+require 'rabbit_jobs/tasks'

data/rabbit_jobs.gemspec

@@ -1,5 +1,6 @@
 # -*- encoding: utf-8 -*-
-require File.expand_path('../lib/rabbit_jobs/version', __FILE__)
+$:.push File.expand_path("../lib", __FILE__)
+require 'rabbit_jobs/version'
 
 Gem::Specification.new do |gem|
   gem.authors       = ["Pavel Lazureykis"]
@@ -7,6 +8,7 @@ Gem::Specification.new do |gem|
   gem.description   = %q{Background jobs on RabbitMQ}
   gem.summary       = %q{Background jobs on RabbitMQ}
   gem.homepage      = ""
+  gem.date          = Time.now.strftime('%Y-%m-%d')
 
   gem.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
   gem.files         = `git ls-files`.split("\n")
@@ -15,7 +17,5 @@ Gem::Specification.new do |gem|
   gem.require_paths = ["lib"]
   gem.version       = RabbitJobs::VERSION
 
-  gem.add_development_dependency "rspec", "~> 2.8"
-
   gem.add_dependency "amqp", "~> 0.9"
 end

data/spec/fixtures/config.yml

@@ -0,0 +1,17 @@
+rabbit_jobs:
+  host: example.com
+  exchange: 'my_exchange'
+  exchange_params:
+    durable: true
+    auto_delete: false
+  queues:
+    durable_queue:
+      durable: true
+      auto_delete: false
+      ack: true
+      arguments:
+        x-ha-policy: all
+    fast_queue:
+      durable: false
+      auto_delete: true
+      ack: false

data/spec/integration/publisher_spec.rb

@@ -0,0 +1,17 @@
+# -*- encoding : utf-8 -*-
+require 'spec_helper'
+require 'json'
+
+describe RabbitJobs::Publisher do
+  it 'should publish message to queue' do
+
+    RabbitJobs.configure do |c|
+      # c.host "somehost.lan"
+      c.exchange 'test', auto_delete: true
+      c.queue 'rspec_queue', auto_delete: true
+    end
+
+    RabbitJobs.enqueue(Integer, 'some', 'other', 'params')
+    RabbitJobs::Publisher.purge_queue('rspec_queue').should == 1
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,20 @@
+# -*- encoding : utf-8 -*-
+require 'simplecov'
+SimpleCov.start {
+  add_filter "spec" # ignore gems
+}
+
+require 'rr'
+
+require 'rabbit_jobs'
+
+RSpec.configure do |config|
+  config.mock_with :rr
+  # or if that doesn't work due to a version incompatibility
+  # config.mock_with RR::Adapters::Rspec
+
+  config.before(:each) do
+    # clear config options
+    RabbitJobs.class_variable_set '@@configuration', nil
+  end
+end

data/spec/unit/configuration_spec.rb

@@ -0,0 +1,89 @@
+# -*- encoding : utf-8 -*-
+require 'spec_helper'
+
+describe RabbitJobs::Configuration do
+  it 'builds configuration from configure block' do
+    RabbitJobs.configure do |c|
+      c.host "somehost.lan"
+
+      c.exchange 'my_exchange', durable: true, auto_delete: false
+
+      c.queue 'durable_queue', durable: true,  auto_delete: false, ack: true, arguments: {'x-ha-policy' => 'all'}
+      c.queue 'fast_queue',    durable: false, auto_delete: true,  ack: false
+    end
+
+    RabbitJobs.config.to_hash.should == {
+      host: "somehost.lan",
+      exchange: "my_exchange",
+      exchange_params: {
+        durable: true,
+        auto_delete: false
+      },
+      queues: {
+        "durable_queue" => {
+          durable: true,
+          auto_delete: false,
+          ack: true,
+          arguments: {"x-ha-policy"=>"all"}
+        },
+        "fast_queue" => {
+          durable: false,
+          auto_delete: true,
+          ack: false
+        },
+      }
+    }
+  end
+
+  it 'builds configuration from yaml' do
+    RabbitJobs.config.load_file(File.expand_path('../../fixtures/config.yml', __FILE__))
+
+    RabbitJobs.config.to_hash.should == {
+      host: "example.com",
+      exchange: "my_exchange",
+      exchange_params: {
+        durable: true,
+        auto_delete: false
+      },
+      queues: {
+        "durable_queue" => {
+          durable: true,
+          auto_delete: false,
+          ack: true,
+          arguments: {"x-ha-policy"=>"all"}
+        },
+        "fast_queue" => {
+          durable: false,
+          auto_delete: true,
+          ack: false
+        }
+      }
+    }
+  end
+
+  it 'use default config' do
+    RabbitJobs.config.to_hash.should == {
+      host: "localhost",
+      exchange: "rabbit_jobs",
+      exchange_params: {
+        auto_delete: false,
+        durable: true
+      },
+      queues: {
+        "default" => {
+          auto_delete: false,
+          ack: true,
+          durable: true
+        }
+      }
+    }
+  end
+
+  it 'returns settings on some methods' do
+    RabbitJobs.config.host.should == 'localhost'
+    RabbitJobs.config[:host].should == 'localhost'
+    RabbitJobs.config.routing_keys.should == ['default']
+    RabbitJobs.config.exchange.should == 'rabbit_jobs'
+    RabbitJobs.config.queue_name('default').should == 'rabbit_jobs#default'
+  end
+end

data/spec/unit/logger_spec.rb

@@ -0,0 +1,23 @@
+# -*- encoding : utf-8 -*-
+require 'spec_helper'
+
+describe RabbitJobs::Logger do
+  it '#log should write messages to stdout' do
+    mock($stdout).puts "hello"
+    RabbitJobs::Logger.log("hello")
+  end
+
+  it '#log! should not write messages to stdout in normal mode' do
+    RabbitJobs::Logger.verbose = false
+
+    dont_allow(RabbitJobs::Logger).log("hello")
+    RabbitJobs::Logger.log!("hello")
+  end
+
+  it '#log! should write messages to stdout in verbose mode' do
+    RabbitJobs::Logger.verbose = true
+
+    mock(RabbitJobs::Logger).log("hello")
+    RabbitJobs::Logger.log!("hello")
+  end
+end

data/spec/unit/rabbit_jobs_spec.rb

@@ -0,0 +1,11 @@
+# -*- encoding : utf-8 -*-
+
+describe RabbitJobs do
+  it 'should pass enqueue methods to publisher' do
+    mock(RabbitJobs::Publisher).enqueue(Integer, 1, 2, "string")
+    RabbitJobs.enqueue(Integer, 1, 2, "string")
+
+    mock(RabbitJobs::Publisher).enqueue_to('default_queue', Integer, 1, 2, "string")
+    RabbitJobs.enqueue_to('default_queue', Integer, 1, 2, "string")
+  end
+end

data/spec/unit/worker_spec.rb

@@ -0,0 +1,49 @@
+# -*- encoding : utf-8 -*-
+require 'spec_helper'
+
+describe RabbitJobs::Worker do
+  describe 'methods' do
+    before :each do
+      @worker = RabbitJobs::Worker.new
+    end
+
+    it '#initialize with default options' do
+      @worker.queues.should == ['default']
+    end
+
+    it '#startup' do
+      @worker.instance_variable_get('@shutdown').should_not == true
+
+      mock(Signal).trap('TERM')
+      mock(Signal).trap('INT')
+
+      @worker.startup
+
+      @worker.instance_variable_get('@shutdown').should_not == true
+    end
+
+    it '#shutdown should set @shutdown to true' do
+      @worker.instance_variable_get('@shutdown').should_not == true
+      @worker.shutdown
+      @worker.instance_variable_get('@shutdown').should == true
+    end
+
+    it '#shutdown! should kill child process' do
+      mock(@worker.kill_child)
+      mock(@worker.shutdown)
+
+      @worker.shutdown!
+    end
+
+    it '#kill_child' do
+      job = RabbitJobs::Job.new(['RabbitJobs'].to_json)
+      job.instance_variable_set '@child', 123123
+      @worker.instance_variable_set('@job', job)
+
+      mock(Kernel).system("ps -o pid,state -p #{123123}") { true }
+      mock(Process).kill("KILL", 123123)
+
+      @worker.kill_child
+    end
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rabbit_jobs
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.3
   prerelease: 
 platform: ruby
 authors:
@@ -9,22 +9,11 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-01-24 00:00:00.000000000 Z
+date: 2012-01-27 00:00:00.000000000 Z
 dependencies:
-- !ruby/object:Gem::Dependency
-  name: rspec
-  requirement: &70100539559980 !ruby/object:Gem::Requirement
-    none: false
-    requirements:
-    - - ~>
-      - !ruby/object:Gem::Version
-        version: '2.8'
-  type: :development
-  prerelease: false
-  version_requirements: *70100539559980
 - !ruby/object:Gem::Dependency
   name: amqp
-  requirement: &70100539559360 !ruby/object:Gem::Requirement
+  requirement: &70147046160300 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ~>
@@ -32,7 +21,7 @@ dependencies:
         version: '0.9'
   type: :runtime
   prerelease: false
-  version_requirements: *70100539559360
+  version_requirements: *70147046160300
 description: Background jobs on RabbitMQ
 email:
 - lazureykis@gmail.com
@@ -41,13 +30,31 @@ extensions: []
 extra_rdoc_files: []
 files:
 - .gitignore
+- .rspec
 - Gemfile
 - LICENSE
 - README.md
 - Rakefile
+- examples/configuration.rb
 - lib/rabbit_jobs.rb
+- lib/rabbit_jobs/amqp_helpers.rb
+- lib/rabbit_jobs/configuration.rb
+- lib/rabbit_jobs/helpers.rb
+- lib/rabbit_jobs/job.rb
+- lib/rabbit_jobs/logger.rb
+- lib/rabbit_jobs/publisher.rb
+- lib/rabbit_jobs/tasks.rb
 - lib/rabbit_jobs/version.rb
+- lib/rabbit_jobs/worker.rb
+- lib/tasks/rabbit_jobs.rake
 - rabbit_jobs.gemspec
+- spec/fixtures/config.yml
+- spec/integration/publisher_spec.rb
+- spec/spec_helper.rb
+- spec/unit/configuration_spec.rb
+- spec/unit/logger_spec.rb
+- spec/unit/rabbit_jobs_spec.rb
+- spec/unit/worker_spec.rb
 homepage: ''
 licenses: []
 post_install_message: 
@@ -72,4 +79,11 @@ rubygems_version: 1.8.15
 signing_key: 
 specification_version: 3
 summary: Background jobs on RabbitMQ
-test_files: []
+test_files:
+- spec/fixtures/config.yml
+- spec/integration/publisher_spec.rb
+- spec/spec_helper.rb
+- spec/unit/configuration_spec.rb
+- spec/unit/logger_spec.rb
+- spec/unit/rabbit_jobs_spec.rb
+- spec/unit/worker_spec.rb