[FIX] test: assertRaises() without environment

2019-07-11 08:48:18 +00:00
parent f6556b19a5
commit b8f4ad585e
1 changed files with 4 additions and 1 deletions
@@ -245,7 +245,10 @@ class BaseCase(TreeCase, MetaCase('DummyCase', (object,), {})):
    def _assertRaises(self, exception):
        """ Context manager that clears the environment upon failure. """
        with super(BaseCase, self).assertRaises(exception) as cm:
-            with self.env.clear_upon_failure():
+            if hasattr(self, 'env'):
+                with self.env.clear_upon_failure():
+                    yield cm
+            else:
                yield cm

    def assertRaises(self, exception, func=None, *args, **kwargs):