Fix k8s tests (#69)

(cherry picked from commit 027fa0d) (cherry picked from commit e0c7db7)
astronomer · Dec 31, 2019 · e2e20ca · e2e20ca
1 parent 06f19c9
commit e2e20ca
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/airflow/contrib/kubernetes/istio.py b/airflow/contrib/kubernetes/istio.py
@@ -18,7 +18,11 @@
 from airflow import AirflowException
 from airflow.utils.log.logging_mixin import LoggingMixin
 from kubernetes.stream import stream
-from packaging.version import parse as semantic_version
+try:
+    from packaging.version import parse as semantic_version
+except ImportError:
+    # Python 2
+    from distutils.version import LooseVersion as semantic_version
 
 
 class SidecarNames:

diff --git a/tests/kubernetes/test_istio.py b/tests/kubernetes/test_istio.py
@@ -19,7 +19,7 @@
 from airflow.contrib.kubernetes.istio import Istio
 
 import unittest
-from unittest.mock import MagicMock, patch
+from tests.compat import MagicMock, patch
 
 
 def mock_stream(func, *args, **kwargs):