lost in squash somehow

TurboGears · Feb 15, 2021 · 2f5ab6d · 2f5ab6d
1 parent e037f62
commit 2f5ab6d
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/test_csrf.py b/tests/test_csrf.py
@@ -3,7 +3,10 @@
 from tg.configurator.components.session import SessionConfigurationComponent
 import webtest
 from tgext.utils.csrf import csrf_protect, csrf_token, CSRFConfigurationComponent
-from unittest.mock import patch, Mock
+try:
+    from unittest.mock import patch, Mock
+except ImportError:
+    from mock import patch, Mock
 
 
 class RootController(TGController):