iron_mq 3.1.0 → 4.0.0

data/test/test_beanstalkd.rb

@@ -1,7 +1,7 @@
 require 'test/unit'
 require 'beanstalk-client'
 require 'yaml'
-require 'test_base'
+require File.expand_path('test_base.rb', File.dirname(__FILE__))
 
 class BeanstalkTests < TestBase
   def setup
@@ -103,6 +103,10 @@ class BeanstalkTests < TestBase
     job = reserve(1)
     assert_equal msg, job.body, "#{job.body} does not equal #{msg}"
     job.delete
+
+    # delete queue on test complete
+    resp = @client.queue('default').delete_queue
+    assert_equal 200, resp.code, "API must response with HTTP 200 status, but returned HTTP #{resp.code}"
   end
 
   def test_timeout
@@ -126,6 +130,10 @@ class BeanstalkTests < TestBase
     assert_not_nil job, "job is nil"
     assert_equal msg, job.body, "body not the same as message."
     job.delete
+
+    # delete queue on test complete
+    resp = @client.queue('default').delete_queue
+    assert_equal 200, resp.code, "API must response with HTTP 200 status, but returned HTTP #{resp.code}"
   end
 
   def test_delay
@@ -145,6 +153,10 @@ class BeanstalkTests < TestBase
     assert_not_nil job, "job is nil"
     assert_equal msg, job.body, "body not the same as message."
     job.delete
+
+    # delete queues on test complete
+    resp = @client.queue('default').delete_queue
+    assert_equal 200, resp.code, "API must response with HTTP 200 status, but returned HTTP #{resp.code}"
   end
 
   def tube_message(tube)
@@ -156,11 +168,11 @@ class BeanstalkTests < TestBase
     omit_if @skip # bypass this test if rackspace
     begin
       job = @beanstalk.reserve(timeout)
-      puts 'got job: ' + job.inspect
-      return job
+      LOG.info 'got job: ' + job.inspect
+      job
     rescue Beanstalk::TimedOut => ex
       puts "Timed out: #{ex.message}"
-      return nil
+      nil
     end
   end
 
@@ -215,6 +227,16 @@ class BeanstalkTests < TestBase
     job = reserve(1)
     assert_equal msg1, job.body, "body #{job.body.to_s} does not equal #{msg1}"
     job.delete
+
+    # delete queues on test complete
+    resp = @client.queue('youtube').delete_queue
+    assert_equal 200, resp.code, "API must response with HTTP 200 status, but returned HTTP #{resp.code}"
+
+    resp = @client.queue('beanstalk_test').delete_queue
+    assert_equal 200, resp.code, "API must response with HTTP 200 status, but returned HTTP #{resp.code}"
+
+    resp = @client.queue('default').delete_queue
+    assert_equal 200, resp.code, "API must response with HTTP 200 status, but returned HTTP #{resp.code}"
   end
 
 

data/test/test_bulk.rb

@@ -1,4 +1,4 @@
-require 'test_base'
+require File.expand_path('test_base.rb', File.dirname(__FILE__))
 
 class TestBulk < TestBase
 
@@ -8,6 +8,7 @@ class TestBulk < TestBase
   end
 
   def test_bulk
+    LOG.info "test_bulk"
 
     q_name = 'ironmq-gem-bulk'
     queue = @client.queue(q_name)
@@ -18,15 +19,18 @@ class TestBulk < TestBase
       puts "Posting #{i}"
       res = queue.post("hello world #{i}!")
     end
-    puts "#{times} posts took #{Time.now.to_f - t.to_f}"
+    LOG.info "#{times} posts took #{Time.now.to_f - t.to_f}"
 
     t = Time.now
     res = nil
     while (res = queue.get())
       del = res.delete
     end
-    puts "#{times} gets and deletes took #{Time.now.to_f - t.to_f}"
+    LOG.info "#{times} gets and deletes took #{Time.now.to_f - t.to_f}"
 
+    # delete queue on test complete
+    resp = queue.delete_queue
+    assert_equal 200, resp.code, "API must response with HTTP 200 status, but returned HTTP #{resp.code}"
   end
 
 end