bumbleworks 0.0.19 → 0.0.20

data/lib/bumbleworks.rb

@@ -24,7 +24,7 @@ module Bumbleworks
       def_delegators :configuration, setting, "#{setting.to_s}="
     end
 
-    def_delegators Bumbleworks::Ruote, :dashboard, :start_worker!
+    def_delegators Bumbleworks::Ruote, :dashboard, :start_worker!, :cancel_all_processes!
     def_delegator Bumbleworks::ProcessDefinition, :define, :define_process
 
     # @public

data/lib/bumbleworks/ruote.rb

@@ -2,6 +2,8 @@ require "ruote"
 
 module Bumbleworks
   class Ruote
+    class CancelTimeout < StandardError; end
+
     class << self
       def dashboard(options = {})
         @dashboard ||= begin
@@ -41,6 +43,21 @@ module Bumbleworks
         dashboard.launch(dashboard.variables[name], *args)
       end
 
+      def cancel_all_processes!(options = {})
+        options[:timeout] ||= 5
+        dashboard.processes.each do |ps|
+          dashboard.cancel(ps.wfid)
+        end
+
+        cancel_start_time = Time.now
+        while dashboard.processes.count > 0
+          if (Time.now - cancel_start_time) > options[:timeout]
+            raise CancelTimeout, "Cancel taking too long - #{dashboard.processes.count} processes remain"
+          end
+          sleep 0.1
+        end
+      end
+
       def register_participants(&block)
         dashboard.register(&block) if block
         set_catchall_if_needed

data/lib/bumbleworks/version.rb

@@ -1,3 +1,3 @@
 module Bumbleworks
-  VERSION = "0.0.19"
+  VERSION = "0.0.20"
 end

data/spec/lib/bumbleworks/ruote_spec.rb

@@ -3,6 +3,44 @@ describe Bumbleworks::Ruote do
     Bumbleworks.reset!
   end
 
+  describe ".cancel_all_processes!" do
+    before :each do
+      Bumbleworks.storage = {}
+      Bumbleworks::Ruote.register_participants
+      Bumbleworks.start_worker!
+    end
+
+    it 'cancels all processes' do
+      5.times do |i|
+        Bumbleworks.define_process "do_nothing_#{i}" do
+          participant :ref => "lazy_guy_#{i}", :task => 'absolutely_nothing'
+        end
+        Bumbleworks.launch!("do_nothing_#{i}")
+        Bumbleworks.dashboard.wait_for("lazy_guy_#{i}".to_sym)
+      end
+      Bumbleworks.dashboard.processes.count.should == 5
+      described_class.cancel_all_processes!
+      Bumbleworks.dashboard.processes.count.should == 0
+    end
+
+    it 'times out if processes are not cancelled in time' do
+      Bumbleworks.define_process "time_hog" do
+        sequence :on_cancel => 'ignore_parents' do
+          pigheaded :task => 'whatever'
+        end
+        define 'ignore_parents' do
+          wait '1s'
+        end
+      end
+      Bumbleworks.launch!('time_hog')
+      Bumbleworks.dashboard.wait_for(:pigheaded)
+      Bumbleworks.dashboard.processes.count.should == 1
+      expect {
+        described_class.cancel_all_processes!(:timeout => 0.5)
+      }.to raise_error(Bumbleworks::Ruote::CancelTimeout)
+    end
+  end
+
   describe '.dashboard' do
     it 'raises an error if no storage is defined' do
       Bumbleworks.storage = nil

data/spec/lib/bumbleworks_spec.rb

@@ -111,6 +111,11 @@ describe Bumbleworks do
       Bumbleworks::Ruote.should_receive(:start_worker!).and_return(:lets_do_it)
       Bumbleworks.start_worker!.should == :lets_do_it
     end
+
+    it 'includes cancel_all_processes!' do
+      Bumbleworks::Ruote.should_receive(:cancel_all_processes!).and_return(:cancelling)
+      Bumbleworks.cancel_all_processes!.should == :cancelling
+    end
   end
 
   describe '.launch!' do

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: bumbleworks
 version: !ruby/object:Gem::Version
-  version: 0.0.19
+  version: 0.0.20
   prerelease: 
 platform: ruby
 authors:
@@ -12,7 +12,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-07-09 00:00:00.000000000 Z
+date: 2013-07-10 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ruote