gearman-ruby 2.0.0

data/test/job_test.rb

@@ -0,0 +1,30 @@
+require File.dirname(__FILE__) + '/test_helper'
+
+class JobTest < Test::Unit::TestCase
+
+  def setup
+    @handle = "H:foo:1"
+    @mock_client = mock()
+    @job = Gearman::Job.new(@mock_client, @handle)
+  end
+
+  def test_supports_report_status
+    @mock_client.expects(:send).with(:work_status, [@handle, 1, 5].join("\0"))
+    @job.report_status(1, 5)
+  end
+
+  def test_supports_send_partial
+    @mock_client.expects(:send).with(:work_data, [@handle, "bar"].join("\0"))
+    @job.send_partial("bar")
+  end
+
+  def test_supports_send_data
+    @mock_client.expects(:send).with(:work_data, [@handle, "bar"].join("\0"))
+    @job.send_data("bar")
+  end
+
+  def test_supports_report_warning
+    @mock_client.expects(:send).with(:work_warning, [@handle, "danger"].join("\0"))
+    @job.report_warning("danger")
+  end
+end

data/test/protocol_test.rb

@@ -0,0 +1,132 @@
+require File.dirname(__FILE__) + '/test_helper'
+
+class ProtocolTest < Test::Unit::TestCase
+
+  def test_encode_request
+    payload = ["foo", "123", "bar"].join("\0")
+    expected = "\0REQ" + [Gearman::Protocol::COMMANDS_NUMERIC[:submit_job], payload.size].pack("NN") + payload
+    assert_equal expected, Gearman::Protocol.encode_request(:submit_job, payload)
+  end
+
+  def test_decode_response
+    response = "\0RES" + [Gearman::Protocol::COMMANDS_NUMERIC[:work_data], 3].pack("NN") + handle + "\0foo"
+    packets = Gearman::Protocol.decode_response(response)
+    assert_equal 1, packets.size
+    assert_equal [:work_data, "H:wrk.acme:1", "foo"], packets.first
+  end
+
+  def test_decodes_multiple_response_packets
+    response = "\0RES" + [Gearman::Protocol::COMMANDS_NUMERIC[:work_data], 3].pack("NN") + handle + "\0foo"
+    response << "\0RES" + [Gearman::Protocol::COMMANDS_NUMERIC[:work_data], 3].pack("NN") + handle + "\0bar"
+    response << "\0RES" + [Gearman::Protocol::COMMANDS_NUMERIC[:work_data], 3].pack("NN") + handle + "\0baz"
+
+    assert_equal 3, Gearman::Protocol.decode_response(response).size
+  end
+
+  def test_response_packet
+    packet = [:work_data, handle, "foo"]
+    assert_equal packet, Gearman::Protocol.response_packet(*packet)
+  end
+
+  def test_decodes_work_complete
+    data     = "esta complet"
+    response = "\0RES" + [Gearman::Protocol::COMMANDS_NUMERIC[:work_complete], data.size].pack("NN") + [handle, data].join("\0")
+    assert_equal [:work_complete, handle, data], Gearman::Protocol.decode_response(response).first
+  end
+
+  def test_decodes_work_exception
+    data     = "{native perl exception object}"
+    response = "\0RES" + [Gearman::Protocol::COMMANDS_NUMERIC[:work_exception], data.size].pack("NN") + [handle, data].join("\0")
+    assert_equal [:work_exception, handle, data], Gearman::Protocol.decode_response(response).first
+  end
+
+  def test_decodes_work_warning
+    data     = "I warn you, dude"
+    response = "\0RES" + [Gearman::Protocol::COMMANDS_NUMERIC[:work_warning], data.size].pack("NN") + [handle, data].join("\0")
+    assert_equal [:work_warning, handle, data], Gearman::Protocol.decode_response(response).first
+  end
+
+  def test_decodes_work_data
+    data     = "foo"
+    response = "\0RES" + [Gearman::Protocol::COMMANDS_NUMERIC[:work_data], data.size].pack("NN") + [handle, data].join("\0")
+    assert_equal [:work_data, handle, data], Gearman::Protocol.decode_response(response).first
+  end
+
+  def test_decodes_error
+    data     = "error"
+    response = "\0RES" + [Gearman::Protocol::COMMANDS_NUMERIC[:error], data.size].pack("NN") + [handle, data].join("\0")
+    assert_equal [:error, handle, data], Gearman::Protocol.decode_response(response).first
+  end
+
+  def test_decodes_job_assign
+    function  = "foo_function"
+    arguments = "arguments"
+
+    payload  = [handle, function, arguments].join("\0")
+    response = "\0RES" + [Gearman::Protocol::COMMANDS_NUMERIC[:job_assign], payload.size].pack("NN") + payload
+    assert_equal [:job_assign, handle, function, arguments], Gearman::Protocol.decode_response(response).first
+  end
+
+  def test_decodes_job_assign_uniq
+    function  = "foo_function"
+    arguments = "arguments"
+    unique_id = "123-657"
+
+    payload  = [handle, function, unique_id, arguments].join("\0")
+    response = "\0RES" + [Gearman::Protocol::COMMANDS_NUMERIC[:job_assign_uniq], payload.size].pack("NN") + payload
+    assert_equal [:job_assign_uniq, handle, function, arguments, unique_id], Gearman::Protocol.decode_response(response).first
+  end
+
+  def test_decodes_work_fail
+    response = "\0RES" + [Gearman::Protocol::COMMANDS_NUMERIC[:work_fail], 0].pack("NN") + handle
+    assert_equal [:work_fail, handle], Gearman::Protocol.decode_response(response).first
+  end
+
+  def test_decodes_job_created
+    response = "\0RES" + [Gearman::Protocol::COMMANDS_NUMERIC[:job_created], 0].pack("NN") + handle
+    assert_equal [:job_created, handle], Gearman::Protocol.decode_response(response).first
+  end
+
+  def test_decodes_no_job
+    response = "\0RES" + [Gearman::Protocol::COMMANDS_NUMERIC[:no_job], 0].pack("NN") + handle
+    assert_equal [:no_job, handle], Gearman::Protocol.decode_response(response).first
+  end
+
+  def test_decodes_noop
+    response = "\0RES" + [Gearman::Protocol::COMMANDS_NUMERIC[:noop], 0].pack("NN") + handle
+    assert_equal [:noop, handle], Gearman::Protocol.decode_response(response).first
+  end
+
+  def test_decodes_work_status
+    numerator   = "1"
+    denominator = "5"
+
+    payload  = [handle, numerator, denominator].join("\0")
+    response = "\0RES" + [Gearman::Protocol::COMMANDS_NUMERIC[:work_status], payload.size].pack("NN") + payload
+
+    assert_equal [:work_status, handle, numerator, denominator], Gearman::Protocol.decode_response(response).first
+  end
+
+  def test_decodes_status_res
+    known       = "1"
+    running     = "0"
+    numerator   = "1"
+    denominator = "4"
+
+    payload = [handle, known, running, numerator, denominator].join("\0")
+    response = "\0RES" + [Gearman::Protocol::COMMANDS_NUMERIC[:status_res], payload.size].pack("NN") + payload
+
+    assert_equal [:status_res, handle, known, running, numerator, denominator], Gearman::Protocol.decode_response(response).first
+  end
+
+  def test_raises_on_invalid_command
+    response = "\0RES" + [6969, 0].pack("NN")
+    assert_raises(Gearman::ProtocolError) { Gearman::Protocol.decode_response(response) }
+    assert_raises(Gearman::ProtocolError) { Gearman::Protocol.response_packet(*[6969, handle, ''])}
+  end
+
+private
+  def handle
+    "H:wrk.acme:1"
+  end
+end

data/test/test_helper.rb

@@ -0,0 +1,31 @@
+$:.unshift(File.expand_path(File.dirname(__FILE__) + '/../lib'))
+require 'gearman'
+require 'test/unit'
+require 'mocha'
+
+def start_gearmand(port = 4730, debug = false)
+  log = debug ? "-l /tmp/gearmand.log -vvv" : ""
+  system "gearmand -d -p #{port} --pid-file=#{gearmand_pidfile(port)} #{log}"
+  gearmand_pid(port)
+end
+
+def stop_gearmand(port = 4730)
+  Process.kill "KILL", gearmand_pid(port)
+  File.unlink gearmand_pidfile(port)
+end
+
+def teardown_gearmands
+  glob = "/tmp/gearmand_*_#{$$}.pid"
+  Dir[glob].each do |pidfile|
+    Process.kill "KILL", `cat #{pidfile}`.to_i
+    File.unlink pidfile
+  end
+end
+
+def gearmand_pid(port = 4730)
+  `cat #{gearmand_pidfile(port)}`.to_i
+end
+
+def gearmand_pidfile(port)
+  "/tmp/gearmand_#{port}_#{$$}.pid"
+end

data/test/util_test.rb

@@ -0,0 +1,12 @@
+require File.dirname(__FILE__) + '/test_helper'
+
+class UtilTest < Test::Unit::TestCase
+
+  def test_ability_prefix_name_builder
+    assert_equal(Gearman::Util.ability_name_with_prefix("test","a"),"test\ta")
+  end
+
+  def test_ability_name_for_perl
+    assert_equal(Gearman::Util.ability_name_for_perl("test","a"),"test\ta")
+  end
+end

data/test/worker_test.rb

@@ -0,0 +1,45 @@
+require File.dirname(__FILE__) + '/test_helper'
+
+class WorkerTest < Test::Unit::TestCase
+
+  def test_accepts_multiple_job_servers
+    Gearman::Evented::WorkerReactor.expects(:connect).times(2)
+
+    EM.run do
+      EM.add_timer(0.1) { EM.stop_event_loop }
+      Gearman::Worker.new(["localhost:4730", "localhost:4731"]).work
+    end
+  end
+
+  def test_accepts_exactly_one_job_server
+    Gearman::Evented::WorkerReactor.expects(:connect).with("localhost", "4730", {:abilities => {}}).times(1)
+
+    EM.run do
+      EM.add_timer(0.1) { EM.stop_event_loop }
+      Gearman::Worker.new("localhost:4730").work
+    end
+  end
+
+  def test_passes_abilities_to_reactor
+    worker = Gearman::Worker.new("localhost:4730")
+    worker.add_ability("foo") {|data, job| "noop!"}
+    worker.add_ability("bar") {|data, job| "nothing to see here!" }
+    worker.remove_ability("bar")
+
+    assert_equal true, worker.has_ability?("foo")
+    assert_equal false, worker.has_ability?("bar")
+
+    Gearman::Evented::WorkerReactor.expects(:connect).with do |host, port, opts|
+      assert_equal "localhost", host
+      assert_equal 4730, port.to_i
+      assert_equal 1, opts[:abilities].size
+      assert_equal true, opts[:abilities]['foo'][:callback].is_a?(Proc)
+      true
+    end
+
+    EM.run do
+      EM.add_timer(0.1) { EM.stop_event_loop }
+      worker.work
+    end
+  end
+end

metadata

@@ -0,0 +1,133 @@
+--- !ruby/object:Gem::Specification 
+name: gearman-ruby
+version: !ruby/object:Gem::Version 
+  version: 2.0.0
+platform: ruby
+authors: 
+- Kim Altintop
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-12-11 00:00:00 +01:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: eventmachine
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 0.12.8
+    version: 
+description: Library for the Gearman distributed job system
+email: kim.altintop@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- LICENSE
+- README
+files: 
+- .gitignore
+- HOWTO
+- LICENSE
+- README
+- Rakefile
+- TODO
+- VERSION.yml
+- examples/calculus_client.rb
+- examples/calculus_worker.rb
+- examples/client.rb
+- examples/client_background.rb
+- examples/client_data.rb
+- examples/client_exception.rb
+- examples/client_prefix.rb
+- examples/evented_client.rb
+- examples/evented_worker.rb
+- examples/gearman_environment.sh
+- examples/scale_image.rb
+- examples/scale_image_worker.rb
+- examples/server.rb
+- examples/worker.rb
+- examples/worker_data.rb
+- examples/worker_exception.rb
+- examples/worker_prefix.rb
+- gearman-ruby.gemspec
+- lib/gearman.rb
+- lib/gearman/client.rb
+- lib/gearman/evented/client.rb
+- lib/gearman/evented/reactor.rb
+- lib/gearman/evented/worker.rb
+- lib/gearman/job.rb
+- lib/gearman/protocol.rb
+- lib/gearman/server.rb
+- lib/gearman/task.rb
+- lib/gearman/taskset.rb
+- lib/gearman/util.rb
+- lib/gearman/worker.rb
+- test/basic_integration_test.rb
+- test/crash_test.rb
+- test/job_test.rb
+- test/protocol_test.rb
+- test/test_helper.rb
+- test/util_test.rb
+- test/worker_test.rb
+has_rdoc: true
+homepage: http://github.com/kim/gearman-ruby
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --charset=UTF-8
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.5
+signing_key: 
+specification_version: 3
+summary: Library for the Gearman distributed job system
+test_files: 
+- test/basic_integration_test.rb
+- test/crash_test.rb
+- test/job_test.rb
+- test/protocol_test.rb
+- test/test_helper.rb
+- test/util_test.rb
+- test/worker_test.rb
+- examples/calculus_client.rb
+- examples/calculus_worker.rb
+- examples/client.rb
+- examples/client_background.rb
+- examples/client_data.rb
+- examples/client_echo.rb
+- examples/client_exception.rb
+- examples/client_prefix.rb
+- examples/evented_client.rb
+- examples/evented_worker.rb
+- examples/scale_image.rb
+- examples/scale_image_worker.rb
+- examples/server.rb
+- examples/worker.rb
+- examples/worker_data.rb
+- examples/worker_echo.rb
+- examples/worker_echo_pprof.rb
+- examples/worker_exception.rb
+- examples/worker_prefix.rb