stalker 0.3.0 → 0.4.0

data/README.md

@@ -76,6 +76,17 @@ Tidbits
 * The stalk binary is just for convenience, you can also run a worker with a straight Ruby command:
     $ ruby -r jobs -e Stalker.work
 
+Running the tests
+-----------------
+
+If you wish to hack on Stalker, install these extra gems:
+
+    $ gem install contest mocha turn
+
+Run the tests:
+
+    $ turn
+
 Meta
 ----
 

data/VERSION

@@ -1 +1 @@
-0.3.0
+0.4.0

data/lib/stalker.rb

@@ -5,9 +5,14 @@ require 'uri'
 module Stalker
 	extend self
 
-	def enqueue(job, args={})
+	def enqueue(job, args={}, opts={})
+		pri   = opts[:pri]   || 65536
+		delay = opts[:delay] || 0
+		ttr   = opts[:ttr]   || 120
 		beanstalk.use job
-		beanstalk.put [ job, args ].to_json
+		beanstalk.put [ job, args ].to_json, pri, delay, ttr
+	rescue Beanstalk::NotConnected => e
+		failed_connection(e)
 	end
 
 	def job(j, &block)
@@ -18,7 +23,7 @@ module Stalker
 	class NoJobsDefined < RuntimeError; end
 	class NoSuchJob < RuntimeError; end
 
-	def work(jobs=nil)
+	def prep(jobs=nil)
 		raise NoJobsDefined unless defined?(@@handlers)
 
 		jobs ||= all_jobs
@@ -34,14 +39,15 @@ module Stalker
 		beanstalk.list_tubes_watched.each do |server, tubes|
 			tubes.each { |tube| beanstalk.ignore(tube) unless jobs.include?(tube) }
 		end
-
-		loop do
-			work_one_job
-		end
 	rescue Beanstalk::NotConnected => e
 		failed_connection(e)
 	end
 
+	def work(jobs=nil)
+		prep(jobs)
+		loop { work_one_job }
+	end
+
 	def work_one_job
 		job = beanstalk.reserve
 		name, args = JSON.parse job.body
@@ -121,4 +127,8 @@ module Stalker
 	def all_jobs
 		@@handlers.keys
 	end
+
+	def clear!
+		@@handlers = nil
+	end
 end

data/test/stalker_test.rb

@@ -0,0 +1,24 @@
+require File.dirname(__FILE__) + '/../lib/stalker'
+require 'contest'
+require 'mocha'
+
+module Stalker
+	def log(msg)
+	end
+end
+
+class StalkerTest < Test::Unit::TestCase
+	setup do
+		Stalker.clear!
+		$result = -1
+	end
+
+	test "enqueue and work a job" do
+		val = rand(999999)
+		Stalker.job('my.job') { |args| $result = args['val'] }
+		Stalker.enqueue('my.job', :val => val)
+		Stalker.prep
+		Stalker.work_one_job
+		assert_equal val, $result
+	end
+end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: stalker
 version: !ruby/object:Gem::Version 
-  hash: 19
+  hash: 15
   prerelease: false
   segments: 
   - 0
-  - 3
+  - 4
   - 0
-  version: 0.3.0
+  version: 0.4.0
 platform: ruby
 authors: 
 - Adam Wiggins
@@ -15,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-06-29 00:00:00 -07:00
+date: 2010-07-29 00:00:00 -07:00
 default_executable: stalk
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -60,6 +60,7 @@ files:
 - VERSION
 - bin/stalk
 - lib/stalker.rb
+- test/stalker_test.rb
 - examples/enqueue.rb
 - examples/jobs.rb
 has_rdoc: true
@@ -97,5 +98,6 @@ signing_key:
 specification_version: 3
 summary: A job queueing and background workers system using Beanstalkd.
 test_files: 
+- test/stalker_test.rb
 - examples/enqueue.rb
 - examples/jobs.rb