Fix bug in envvar source tests when run in Python 2.7

Due to an import error the envvar source tests would not run
in Python 2.7
This commit is contained in:
blanariu
2021-06-24 12:45:22 +03:00
parent 3cc0fac817
commit 749f0bd90f
+7 -1
View File
@@ -1,6 +1,12 @@
from six import text_type
from unittest import TestCase
from unittest.mock import patch
try:
# Python 3
from unittest.mock import patch
except ImportError:
# Python 2
from mock import patch
from octodns.source.envvar import EnvVarSource
from octodns.source.envvar import EnvironmentVariableNotFoundException