Fix imagePullPolicy missing in tests (apache#13316)

The apache#13289 fixed imagePullPolicy behaviour but broke master K8S tests. This PR attempts to fix it.
astronomer · Dec 26, 2020 · 6f246b0 · 6f246b0
1 parent a1e9195
commit 6f246b0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/kubernetes_tests/test_kubernetes_pod_operator.py b/kubernetes_tests/test_kubernetes_pod_operator.py
@@ -90,6 +90,7 @@ def setUp(self):
                 'containers': [
                     {
                         'image': 'ubuntu:16.04',
+                        'imagePullPolicy': 'IfNotPresent',
                         'args': ["echo 10"],
                         'command': ["bash", "-cx"],
                         'env': [],
@@ -857,6 +858,7 @@ def test_pod_template_file(
                         'env': [],
                         'envFrom': [],
                         'image': 'apache/airflow:stress-2020.07.10-1.0.4',
+                        'imagePullPolicy': 'IfNotPresent',
                         'name': 'base',
                         'ports': [],
                         'resources': {'limits': {'memory': '200Mi'}, 'requests': {'memory': '100Mi'}},

diff --git a/kubernetes_tests/test_kubernetes_pod_operator_backcompat.py b/kubernetes_tests/test_kubernetes_pod_operator_backcompat.py
@@ -88,6 +88,7 @@ def setUp(self):
                 'containers': [
                     {
                         'image': 'ubuntu:16.04',
+                        'imagePullPolicy': 'IfNotPresent',
                         'args': ["echo 10"],
                         'command': ["bash", "-cx"],
                         'env': [],