Преглед изворни кода

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

Erick Tryzelaar пре 10 година
родитељ
комит
4b9d569
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