test/support/test_handler.rb in m2r-1.0.0 vs test/support/test_handler.rb in m2r-2.0.0

@@ -1,10 +1,10 @@
 require 'm2r/handler'
 
 class TestHandler < M2R::Handler
   attr_reader :called_methods
-  def initialize(connection)
+  def initialize(connection, parser)
     super
     @called_methods = []
   end
 
   def on_wait()
@@ -45,7 +45,11 @@
     @called_methods << :reply
   end
 
   def after_all(request, response)
     @called_methods << :all
+  end
+
+  def on_error(request, response, error)
+    @called_methods << :error
   end
 end