spec/spec_helper.rb in angelo-0.1.2 vs spec/spec_helper.rb in angelo-0.1.3

@@ -6,5 +6,79 @@
 require 'angelo/rspec/helpers'
 Celluloid.logger.level = ::Logger::ERROR
 include Angelo::RSpec::Helpers
 
 TEST_APP_ROOT = File.expand_path '../test_app_root', __FILE__
+
+CK = 'ANGELO_CONCURRENCY' # concurrency key
+DC = 5                    # default concurrency
+CONCURRENCY = ENV.key?(CK) ? ENV[CK].to_i : DC
+
+# https://gist.github.com/tkareine/739662
+#
+class CountDownLatch
+  attr_reader :count
+
+  def initialize(to)
+    @count = to.to_i
+    raise ArgumentError, "cannot count down from negative integer" unless @count >= 0
+    @lock = Mutex.new
+    @condition = ConditionVariable.new
+  end
+
+  def count_down
+    @lock.synchronize do
+      @count -= 1 if @count > 0
+      @condition.broadcast if @count == 0
+    end
+  end
+
+  def wait
+    @lock.synchronize do
+      @condition.wait(@lock) while @count > 0
+    end
+  end
+
+end
+
+module Cellper
+
+  @@stop = false
+  @@testers = {}
+
+  def define_action sym, &block
+    define_method sym, &block
+  end
+
+  def remove_action sym
+    remove_method sym
+  end
+
+  def unstop!
+    @@stop = false
+  end
+
+  def stop!
+    @@stop = true
+  end
+
+  def stop?
+    @@stop
+  end
+
+  def testers; @@testers; end
+
+end
+
+class Reactor
+  include Celluloid::IO
+  extend Cellper
+end
+
+$reactor = Reactor.new
+
+class ActorPool
+  include Celluloid
+  extend Cellper
+end
+
+$pool = ActorPool.pool size: CONCURRENCY