浏览代码

HUE-2436 [desktop] Make sure to reset the mako test wrapper

Erick Tryzelaar 10 年之前
父节点
当前提交
4b9d5696a4
共有 1 个文件被更改,包括 4 次插入2 次删除
  1. 4 2
      desktop/core/src/desktop/lib/test_runners.py

+ 4 - 2
desktop/core/src/desktop/lib/test_runners.py

@@ -39,8 +39,10 @@ def _instrumented_test_render(self, *args, **data):
       return self.original_callable_(context, *args, **kwargs)
     self.original_callable_ = self.callable_
     self.callable_ = mako_callable_
-    response = runtime._render(self, self.original_callable_, args, data)
-    self.callable_ = self.original_callable_
+    try:
+      response = runtime._render(self, self.original_callable_, args, data)
+    finally:
+      self.callable_ = self.original_callable_
     return response