HH-32061 add method class to _handler

hhru · Feb 21, 2013 · 2604fdd · 2604fdd
1 parent 617d083
commit 2604fdd
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/frontik/testing/service_mock.py b/frontik/testing/service_mock.py
@@ -252,6 +252,8 @@ def process_callbacks(self):
                         callback()
 
     def call(self, method, *arg, **kwarg):
+        if hasattr(method, 'im_class'):
+            self._handler.__class__ = type('Page', (method.im_class,) + self._handler.__class__.__bases__, {})
         handler = self._handler
         handler.prepare()
         handler._finished = False