lib/sidekiq/actor.rb in sidekiq-2.17.4 vs lib/sidekiq/actor.rb in sidekiq-2.17.5

@@ -1,7 +1,38 @@
 module Sidekiq
   module Actor
+
+    module ClassMethods
+      def trap_exit(*args)
+      end
+      def new_link(*args)
+        new(*args)
+      end
+    end
+
+    module InstanceMethods
+      def current_actor
+        self
+      end
+      def after(interval)
+      end
+      def alive?
+        !@dead
+      end
+      def terminate
+        @dead = true
+      end
+      def defer
+        yield
+      end
+    end
+
     def self.included(klass)
-      klass.send(:include, Celluloid)
+      if $TESTING
+        klass.__send__(:include, InstanceMethods)
+        klass.__send__(:extend, ClassMethods)
+      else
+        klass.__send__(:include, Celluloid)
+      end
     end
   end
 end