spec/support/example_actor_class.rb in celluloid-0.13.0 vs spec/support/example_actor_class.rb in celluloid-0.14.0.pre

@@ -1,11 +1,13 @@
 module ExampleActorClass
-  def self.create(included_module)
+  def self.create(included_module, task_klass)
     Class.new do
       include included_module
+      task_class task_klass
       attr_reader :name
       finalizer :my_finalizer
+      execute_block_on_receiver :run_on_receiver
 
       def initialize(name)
         @name = name
         @delegate = [:bar]
       end
@@ -25,9 +27,13 @@
       def actor?
         Celluloid.actor?
       end
 
       def run(*args)
+        yield(*args)
+      end
+
+      def run_on_receiver(*args)
         yield(*args)
       end
 
       def crash
         raise ExampleCrash, "the spec purposely crashed me :("