lib/neo/dci/context.rb in neo-dci-0.3.0 vs lib/neo/dci/context.rb in neo-dci-0.4.0

@@ -1,7 +1,5 @@
-require 'on'
-
 module Neo
   module DCI
     class Context
       class << self
         private :new
@@ -15,14 +13,19 @@
         @callbacks
       end
 
       def self.call(*args, &block)
         context = new(*args)
-        context.callback = On.new(*callbacks, &block)
+        context.callback = result_class.new(*callbacks, &block)
         context.call
         raise NoCallbackCalled, callbacks unless context.callback.callback
       rescue NotImplementedError
         raise
+      end
+
+      def self.result_class(klass = :reader)
+        @result_class = klass unless klass == :reader
+        @result_class || ContextResult
       end
 
       def call
         raise NotImplementedError
       end