allq_client 1.0.0 → 1.1.1

data/spec/api/actions_api_spec.rb

@@ -13,6 +13,8 @@ Swagger Codegen version: 2.3.1
 require 'spec_helper'
 require 'json'
 
+DEFAULT_TUBE = "default"
+ALT_TUBE = ""
 # Unit tests for Allq::ActionsApi
 # Automatically generated by swagger-codegen (github.com/swagger-api/swagger-codegen)
 # Please update as you see appropriate
@@ -20,6 +22,10 @@ describe 'ActionsApi' do
   before do
     # run before each test
     @instance = Allq::ActionsApi.new
+    stats_result = AllQTools::get_stats[0]
+    stats_result.stats.each do |stat|
+      @instance.tube_delete(stat.tube)
+    end
   end
 
   after do
@@ -40,7 +46,12 @@ describe 'ActionsApi' do
   # @return [BasicResponse]
   describe 'bury_put test' do
     it "should work" do
-      # assertion here. ref: https://www.relishapp.com/rspec/rspec-expectations/docs/built-in-matchers
+      AllQTools::add_job
+      expect(AllQTools::get_ready(DEFAULT_TUBE)).to eq(1)
+      job = @instance.job_get(DEFAULT_TUBE)
+      @instance.bury_put(job.id)
+      expect(AllQTools::get_buried(DEFAULT_TUBE)).to eq(1)
+      expect(AllQTools::get_ready(DEFAULT_TUBE)).to eq(0)
     end
   end
 
@@ -53,7 +64,11 @@ describe 'ActionsApi' do
   # @return [BasicResponse]
   describe 'job_delete test' do
     it "should work" do
-      # assertion here. ref: https://www.relishapp.com/rspec/rspec-expectations/docs/built-in-matchers
+      AllQTools::add_job
+      expect(AllQTools::get_ready(DEFAULT_TUBE)).to eq(1)
+      job = @instance.peek_get(DEFAULT_TUBE)
+      @instance.job_delete(job.id, tube: DEFAULT_TUBE)
+      expect(AllQTools::get_ready(DEFAULT_TUBE)).to eq(0)
     end
   end
 
@@ -65,7 +80,31 @@ describe 'ActionsApi' do
   # @return [JobResponse]
   describe 'job_get test' do
     it "should work" do
-      # assertion here. ref: https://www.relishapp.com/rspec/rspec-expectations/docs/built-in-matchers
+      AllQTools::add_job
+      job = @instance.job_get(DEFAULT_TUBE)
+      expect(AllQTools::get_reserved(DEFAULT_TUBE)).to eq(1)
+      expect(job.id).to be_truthy
+    end
+  end
+
+  # unit tests for multiple job_get
+  # Job
+  # Get job from queue
+  # @param tube Name of tube
+  # @param [Hash] opts the optional parameters
+  # @return [JobResponse]
+  describe 'mulitple_job_get test' do
+    it "should work" do
+      1.upto(6) do
+        AllQTools::add_job
+      end
+      multiple_jobs = @instance.multiple_job_get(DEFAULT_TUBE, 4)
+      expect(multiple_jobs.jobs.length).to eq(4)
+      expect(AllQTools::get_ready(DEFAULT_TUBE)).to eq(2)
+
+      multiple_jobs = @instance.multiple_job_get(DEFAULT_TUBE, 4)
+      expect(multiple_jobs.jobs.length).to eq(2)
+      expect(AllQTools::get_ready(DEFAULT_TUBE)).to eq(0)
     end
   end
 
@@ -77,7 +116,10 @@ describe 'ActionsApi' do
   # @return [JobRef]
   describe 'job_post test' do
     it "should work" do
-      # assertion here. ref: https://www.relishapp.com/rspec/rspec-expectations/docs/built-in-matchers
+      AllQTools::add_job
+      expect(AllQTools::get_ready(DEFAULT_TUBE)).to eq(1)
+      job = @instance.job_get(DEFAULT_TUBE)
+      expect(job.id).to be_truthy
     end
   end
 
@@ -88,8 +130,91 @@ describe 'ActionsApi' do
   # @param [Hash] opts the optional parameters
   # @return [JobRef]
   describe 'parent_job_post test' do
-    it "should work" do
-      # assertion here. ref: https://www.relishapp.com/rspec/rspec-expectations/docs/built-in-matchers
+    it "should work" do      
+      new_parent_job = Allq::NewParentJob.new(tube: 'default',
+                         body: '<PARENT_BODY>',
+                         ttl: 1200,
+                         delay: 0,
+                         priority: 5,
+                         timeout: 1000,
+                         run_on_timeout: false,
+                         limit: 2)
+      new_parent_job_response = @instance.parent_job_post(new_parent_job)
+      expect(AllQTools::get_parents(DEFAULT_TUBE)).to eq(1)
+      job = AllQTools::new_job(DEFAULT_TUBE, nil, new_parent_job_response.job_id)
+      @instance.job_post(job)
+      job = AllQTools::new_job(DEFAULT_TUBE, nil, new_parent_job_response.job_id)
+      @instance.job_post(job)
+      expect(AllQTools::get_ready(DEFAULT_TUBE)).to eq(2)
+      AllQTools::get_finish(DEFAULT_TUBE)
+      AllQTools::get_finish(DEFAULT_TUBE)
+      expect(AllQTools::get_ready(DEFAULT_TUBE)).to eq(1)
+      expect(AllQTools::get_parents(DEFAULT_TUBE)).to eq(0)
+      AllQTools::get_finish(DEFAULT_TUBE)
+      expect(AllQTools::get_ready(DEFAULT_TUBE)).to eq(0)
+    end
+  end
+
+  # unit tests for parent_job_post
+  # Parent  Job
+  # Create a parent job
+  # @param new_parent_job New Parent Job Data
+  # @param [Hash] opts the optional parameters
+  # @return [JobRef]
+  describe 'parent_job_post test with different tubes' do
+    it "should work" do      
+      new_parent_job = Allq::NewParentJob.new(tube: 'default',
+                         body: '<PARENT_BODY>',
+                         ttl: 1200,
+                         delay: 0,
+                         priority: 5,
+                         timeout: 1000,
+                         run_on_timeout: false,
+                         limit: 2)
+      new_parent_job_response = @instance.parent_job_post(new_parent_job)
+      expect(AllQTools::get_parents(DEFAULT_TUBE)).to eq(1)
+
+      job = AllQTools::new_job('foo', nil, new_parent_job_response.job_id)
+      @instance.job_post(job)
+      job = AllQTools::new_job('foo', nil, new_parent_job_response.job_id)
+      @instance.job_post(job)
+      expect(AllQTools::get_ready('foo')).to eq(2)
+      AllQTools::get_finish('foo')
+      AllQTools::get_finish('foo')
+      expect(AllQTools::get_ready(DEFAULT_TUBE)).to eq(1)
+      expect(AllQTools::get_parents(DEFAULT_TUBE)).to eq(0)
+      AllQTools::get_finish(DEFAULT_TUBE)
+      expect(AllQTools::get_ready(DEFAULT_TUBE)).to eq(0)
+    end
+  end
+
+  # unit tests for parent_job_post
+  # Parent  Job
+  # Create a parent job
+  # @param new_parent_job New Parent Job Data
+  # @param [Hash] opts the optional parameters
+  # @return [JobRef]
+  describe 'parent_job_post should timeout if child not completed in time' do
+    fit "should work" do      
+      new_parent_job = Allq::NewParentJob.new(tube: 'default',
+                         body: '<PARENT_BODY>',
+                         ttl: 1200,
+                         delay: 0,
+                         priority: 5,
+                         timeout: 2,
+                         run_on_timeout: true,
+                         limit: 2)
+      new_parent_job_response = @instance.parent_job_post(new_parent_job)
+      expect(AllQTools::get_parents(DEFAULT_TUBE)).to eq(1)
+
+      job = AllQTools::new_job('foo', nil, new_parent_job_response.job_id)
+      @instance.job_post(job)
+      expect(AllQTools::get_ready('foo')).to eq(1)
+      expect(AllQTools::get_parents(DEFAULT_TUBE)).to eq(1)
+
+      sleep(7)
+      expect(AllQTools::get_parents(DEFAULT_TUBE)).to eq(0)
+      expect(AllQTools::get_ready(DEFAULT_TUBE)).to eq(1)
     end
   end
 
@@ -102,7 +227,36 @@ describe 'ActionsApi' do
   # @return [JobResponse]
   describe 'peek_get test' do
     it "should work" do
-      # assertion here. ref: https://www.relishapp.com/rspec/rspec-expectations/docs/built-in-matchers
+      AllQTools::add_job
+      job = @instance.peek_get(DEFAULT_TUBE)
+      expect(job.id).to be_truthy
+    end
+  end
+
+  describe 'peek_get buried test' do
+    it "should work" do
+      AllQTools::add_job
+      job = @instance.peek_get(DEFAULT_TUBE, buried: true)
+      expect(job.id).to be_falsey
+      job = @instance.job_get(DEFAULT_TUBE)
+      @instance.bury_put(job.id)
+      job = @instance.peek_get(DEFAULT_TUBE, buried: true)
+      expect(job.id).to be_truthy
+
+    end
+  end
+
+  # unit tests for peek_get
+  # Peek
+  # Peek at next job
+  # @param tube Tube name
+  # @param [Hash] opts the optional parameters
+  # @option opts [String] :buried Look in buried
+  # @return [JobResponse]
+  describe 'peek_get test(no jobs)' do
+    it "should work" do
+      job = @instance.peek_get(DEFAULT_TUBE)
+      expect(job.id).to be_falsey
     end
   end
 
@@ -114,7 +268,31 @@ describe 'ActionsApi' do
   # @return [BasicResponse]
   describe 'release_put test' do
     it "should work" do
-      # assertion here. ref: https://www.relishapp.com/rspec/rspec-expectations/docs/built-in-matchers
+      AllQTools::add_job
+      expect(AllQTools::get_ready(DEFAULT_TUBE)).to eq(1)
+      job = @instance.job_get(DEFAULT_TUBE)
+      expect(AllQTools::get_reserved(DEFAULT_TUBE)).to eq(1)
+      expect(job.id).to be_truthy
+      @instance.release_put(job.id)
+      expect(AllQTools::get_ready(DEFAULT_TUBE)).to eq(1)
+    end
+  end
+
+    describe 'release should increment release' do
+    it "should work" do
+      AllQTools::add_job
+      expect(AllQTools::get_ready(DEFAULT_TUBE)).to eq(1)
+      job = @instance.job_get(DEFAULT_TUBE)
+      expect(job.releases).to eq(0)
+      @instance.release_put(job.id)
+      sleep(1)
+      job = @instance.job_get(DEFAULT_TUBE)
+      expect(job.releases).to eq(1)
+      @instance.release_put(job.id)
+      sleep(1)
+      job = @instance.job_get(DEFAULT_TUBE)
+      expect(job.releases).to eq(2)
+
     end
   end
 
@@ -126,7 +304,34 @@ describe 'ActionsApi' do
   # @return [BasicResponse]
   describe 'set_children_started_put test' do
     it "should work" do
-      # assertion here. ref: https://www.relishapp.com/rspec/rspec-expectations/docs/built-in-matchers
+      new_parent_job = Allq::NewParentJob.new(tube: 'default',
+                         body: '<PARENT_BODY>',
+                         ttl: 1200,
+                         delay: 0,
+                         priority: 5,
+                         timeout: 1000,
+                         run_on_timeout: false,
+                         limit: nil)
+      new_parent_job_response = @instance.parent_job_post(new_parent_job)
+      expect(AllQTools::get_parents(DEFAULT_TUBE)).to eq(1)
+      job = AllQTools::new_job(DEFAULT_TUBE, nil, new_parent_job_response.job_id)
+      @instance.job_post(job)
+      job = AllQTools::new_job(DEFAULT_TUBE, nil, new_parent_job_response.job_id)
+      @instance.job_post(job)
+      expect(AllQTools::get_ready(DEFAULT_TUBE)).to eq(2)
+      AllQTools::get_finish(DEFAULT_TUBE)
+      AllQTools::get_finish(DEFAULT_TUBE)
+      # After jobs have already finished
+      expect(AllQTools::get_ready(DEFAULT_TUBE)).to eq(0)
+      # Parent still waiting
+      expect(AllQTools::get_parents(DEFAULT_TUBE)).to eq(1)
+      # Set limit on parent
+      @instance.set_children_started_put(new_parent_job_response.job_id)
+      # Now parent gets moved to ready
+      expect(AllQTools::get_ready(DEFAULT_TUBE)).to eq(1)
+      expect(AllQTools::get_parents(DEFAULT_TUBE)).to eq(0)
+      AllQTools::get_finish(DEFAULT_TUBE)
+      expect(AllQTools::get_ready(DEFAULT_TUBE)).to eq(0)
     end
   end
 
@@ -138,7 +343,24 @@ describe 'ActionsApi' do
   # @return [TubeRef]
   describe 'throttle_post test' do
     it "should work" do
-      # assertion here. ref: https://www.relishapp.com/rspec/rspec-expectations/docs/built-in-matchers
+      throttle = Allq::Throttle.new(tps: 5, tube: "throttled")
+      @instance.throttle_post(throttle)
+    end
+  end
+
+  # unit tests for stats
+  # Stats Action Count
+  # Checks action count
+  # @param throttle Throttle info
+  # @param [Hash] opts the optional parameters
+  # @return [TubeRef]
+  describe 'throttle_post test' do
+    it "should work" do
+      AllQTools::add_job
+      expect(AllQTools::get_ready(DEFAULT_TUBE)).to eq(1)
+      @instance.job_get(DEFAULT_TUBE)
+      stats_result = AllQTools::get_stats[0]
+      expect(stats_result.action_count > 1).to eq(true)
     end
   end
 
@@ -150,7 +372,10 @@ describe 'ActionsApi' do
   # @return [BasicResponse]
   describe 'touch_put test' do
     it "should work" do
-      # assertion here. ref: https://www.relishapp.com/rspec/rspec-expectations/docs/built-in-matchers
+      AllQTools::add_job
+      expect(AllQTools::get_ready(DEFAULT_TUBE)).to eq(1)
+      job = @instance.job_get(DEFAULT_TUBE)
+      @instance.touch_put(job.id)
     end
   end
 
@@ -162,8 +387,185 @@ describe 'ActionsApi' do
   # @return [BasicResponse]
   describe 'tube_delete test' do
     it "should work" do
-      # assertion here. ref: https://www.relishapp.com/rspec/rspec-expectations/docs/built-in-matchers
+      alt_tube_name = "alt_tube"
+      new_job = Allq::NewJob.new(tube: alt_tube_name,
+                       body: "Some body",
+                       ttl: 1200,
+                       delay: 0,
+                       priority: 5,
+                       parent_id: nil)
+      @instance.job_post(new_job)
+      new_job = Allq::NewJob.new(tube: alt_tube_name,
+                       body: "Some body",
+                       ttl: 1200,
+                       delay: 0,
+                       priority: 5,
+                       parent_id: nil)
+      @instance.job_post(new_job)
+      new_job = Allq::NewJob.new(tube: alt_tube_name,
+                       body: "Some body",
+                       ttl: 1200,
+                       delay: 0,
+                       priority: 5,
+                       parent_id: nil)
+      @instance.job_post(new_job)
+      new_job = Allq::NewJob.new(tube: "alt_tube2",
+                       body: "Some body",
+                       ttl: 1200,
+                       delay: 0,
+                       priority: 5,
+                       parent_id: nil)
+      @instance.job_post(new_job)
+      new_job = Allq::NewJob.new(tube: "alt_tube2",
+                       body: "Some body",
+                       ttl: 1200,
+                       delay: 0,
+                       priority: 5,
+                       parent_id: nil)
+      @instance.job_post(new_job)
+      expect(AllQTools::get_ready("alt_tube2")).to eq(2) # OK..it's there
+      reserved_job = @instance.job_get("alt_tube2")
+
+      # Other tube has 1 ready, 1 reserved
+      expect(AllQTools::get_ready("alt_tube2")).to eq(1)
+      expect(AllQTools::get_reserved("alt_tube2")).to eq(1)
+
+      # Get 1 into buried, 1 into reserved, and 1 in ready so we can clear it
+      expect(AllQTools::get_ready(alt_tube_name)).to eq(3) # OK..it's there
+      reserved_job = @instance.job_get(alt_tube_name)
+      expect(AllQTools::get_reserved(alt_tube_name)).to eq(1) # OK..it's there
+      reserved_job = @instance.job_get(alt_tube_name)
+      expect(AllQTools::get_reserved(alt_tube_name)).to eq(2) # OK..it's there
+      @instance.bury_put(reserved_job.id)
+      expect(AllQTools::get_buried(alt_tube_name)).to eq(1)
+
+      # Now check that first tube is empty
+      @instance.tube_delete(alt_tube_name)
+      expect(AllQTools::get_buried(alt_tube_name)).to eq(0)
+      expect(AllQTools::get_reserved(alt_tube_name)).to eq(0) # OK..it's there
+      expect(AllQTools::get_ready(alt_tube_name)).to eq(0)
+      
+      # ..and other tubes are fine
+      expect(AllQTools::get_ready("alt_tube2")).to eq(1)
+      expect(AllQTools::get_reserved("alt_tube2")).to eq(1)
     end
   end
 
+  # unit tests for delay
+  # Delay
+  describe 'delay test' do
+    it "should work" do
+      new_job = Allq::NewJob.new(tube: DEFAULT_TUBE,
+                       body: "Some body",
+                       ttl: 1200,
+                       delay: 1,
+                       priority: 5,
+                       parent_id: nil)
+      @instance.job_post(new_job)
+      expect(AllQTools::get_ready(DEFAULT_TUBE)).to eq(0) # Too soon to show up
+      sleep(4)
+      expect(AllQTools::get_ready(DEFAULT_TUBE)).to eq(1) # OK..it's there
+    end
+  end  
+
+  # unit tests for expire
+  # Expire
+  describe 'expire test' do
+    it "should work" do
+      new_job = Allq::NewJob.new(tube: DEFAULT_TUBE,
+                       body: "Some body",
+                       ttl: 1,
+                       delay: 0,
+                       priority: 5,
+                       parent_id: nil)
+      @instance.job_post(new_job)
+      sleep(1)
+      @instance.job_get(DEFAULT_TUBE)
+      expect(AllQTools::get_reserved(DEFAULT_TUBE)).to eq(1)
+      sleep(5)
+      expect(AllQTools::get_reserved(DEFAULT_TUBE)).to eq(0) # Should expire
+      expect(AllQTools::get_ready(DEFAULT_TUBE)).to eq(1) # ..and be back in ready
+    end
+  end  
+
+end
+
+
+module AllQTools
+  class << self
+
+    def new_job(tube = 'default', body = nil, parent_id = nil)
+      body = (0...8).map { (65 + rand(26)).chr }.join unless body
+
+      new_job = Allq::NewJob.new(tube: tube,
+                       body: body,
+                       ttl: 1200,
+                       delay:  0,
+                       priority: 5,
+                       parent_id:parent_id)
+      return new_job
+    end 
+
+    def add_job(tube = DEFAULT_TUBE)
+      client = Allq::ActionsApi.new
+      job = new_job(tube)
+      job = client.job_post(new_job)
+      return job
+    end
+
+    def get_finish(tube)
+      client = Allq::ActionsApi.new
+      job = client.job_get(tube)
+      client.job_delete(job.id)
+    end
+
+    def get_stats
+      client = Allq::AdminApi.new
+      client.stats_get
+    end
+
+    def get_ready(tube)
+      stats = get_stats
+      count = 0
+      find_tueq(stats, tube) do |found_stat|
+        count += found_stat.ready.to_i
+      end
+      return count
+    end
+
+    def get_reserved(tube)
+      stats = get_stats
+      count = 0
+      find_tueq(stats, tube) do |found_stat|
+        count += found_stat.reserved.to_i
+      end
+      return count
+    end
+
+    def get_buried(tube)
+      stats = get_stats
+      count = 0
+      find_tueq(stats, tube) do |found_stat|
+        count += found_stat.buried.to_i
+      end
+      return count
+    end
+
+    def get_parents(tube)
+      stats = get_stats
+      count = 0
+      find_tueq(stats, tube) do |found_stat|
+        count += found_stat.parents.to_i
+      end
+      return count
+    end
+
+    def find_tueq(all_stats, tube)
+      all_stats.each do |server_stat|
+        server_stat.stats.each do |stat|
+          yield stat if stat.tube == tube
+        end
+      end
+    end
+  end
 end