From 749f0bd90fae302f23c9d0951c5447b279782fd6 Mon Sep 17 00:00:00 2001 From: blanariu Date: Thu, 24 Jun 2021 12:45:22 +0300 Subject: [PATCH] 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 --- tests/test_octodns_source_envvar.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/tests/test_octodns_source_envvar.py b/tests/test_octodns_source_envvar.py index 0714883..38b63dd 100644 --- a/tests/test_octodns_source_envvar.py +++ b/tests/test_octodns_source_envvar.py @@ -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