lib/deas/test_runner.rb in deas-0.28.0 vs lib/deas/test_runner.rb in deas-0.29.0

@@ -1,17 +1,24 @@
 require 'rack/multipart'
 require 'deas/router'
 require 'deas/runner'
+require 'deas/view_handler'
 
 module Deas
 
+  InvalidServiceHandlerError = Class.new(StandardError)
+
   class TestRunner < Runner
 
     attr_reader :return_value
 
     def initialize(handler_class, args = nil)
-      args = (args || {}).dup
+      if !handler_class.include?(Deas::ViewHandler)
+        raise InvalidServiceHandlerError, "#{handler_class.inspect} is not a"\
+                                          " Deas::ServiceHandler"
+      end
 
+      args = (args || {}).dup
       super(handler_class, {
         :request  => args.delete(:request),
         :response => args.delete(:response),
         :session  => args.delete(:session),
         :params   => NormalizedParams.new(args.delete(:params) || {}).value,