infopark-politics 0.2.10 → 0.2.11

data/lib/politics/static_queue_worker.rb

@@ -125,9 +125,9 @@ module Politics
       seize_leadership
     end
 
-    def seize_leadership(duration = nil)
-      @memcache_client.set(token, uri, duration || iteration_length)
-      @nominated_at = Time.now
+    def seize_leadership(duration = iteration_length)
+      @memcache_client.set(token, uri, duration)
+      @nominated_at = Time.now + duration - iteration_length
     end
 
     def perform_leader_duties

data/spec/static_queue_worker_spec.rb

@@ -181,4 +181,47 @@ describe Worker do
       end
     end
   end
+
+  describe "when seizing leadership" do
+    before do
+      @worker.stub!(:uri).and_return('myself')
+      @worker.stub!(:iteration_length).and_return 123
+      @worker.stub!(:token).and_return('dcc-group')
+    end
+
+    it "should set itself to leader" do
+      @@memcache_client.should_receive(:set).with(anything(), 'myself', anything())
+      @worker.seize_leadership
+    end
+
+    it "should seize the leadership for the amount of seconds given" do
+      @@memcache_client.should_receive(:set).with(anything(), anything(), 666)
+      @worker.seize_leadership 666
+    end
+
+    it "should seize the leadership for iteration_length if no duration is given" do
+      @@memcache_client.should_receive(:set).with(anything(), anything(), 123)
+      @worker.seize_leadership
+    end
+
+    it "should seize the leadership for the worker's group" do
+      @@memcache_client.should_receive(:set).with('dcc-group', anything(), anything())
+      @worker.seize_leadership
+    end
+
+    it "should have the next iteration exactly when the seized leadership ends" do
+      now = Time.now
+      Time.stub!(:now).and_return now
+      end_of_leadership = now + 666
+
+      @worker.seize_leadership 666
+      @worker.until_next_iteration.should == 666
+
+      @worker.seize_leadership
+      @worker.until_next_iteration.should == 123
+
+      @worker.seize_leadership 6
+      @worker.until_next_iteration.should == 6
+    end
+  end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: infopark-politics
 version: !ruby/object:Gem::Version 
-  version: 0.2.10
+  version: 0.2.11
 platform: ruby
 authors: 
 - Mike Perham