diff options
Diffstat (limited to 'testsuite/common.py')
-rw-r--r-- | testsuite/common.py | 8 |
1 files changed, 2 insertions, 6 deletions
diff --git a/testsuite/common.py b/testsuite/common.py index fc79e73e7..3aeeaf5f2 100644 --- a/testsuite/common.py +++ b/testsuite/common.py @@ -3,12 +3,7 @@ import re import sys import unittest from mock import patch, MagicMock, _patch, DEFAULT - -try: - from functools import wraps -except ImportError: - def wraps(wrapped): - return lambda f: f +from Bcfg2.Compat import wraps datastore = "/" @@ -33,6 +28,7 @@ try: except ImportError: has_django = False + try: from mock import call except ImportError: |