2 files changed, 12 insertions, 0 deletions
diff --git a/tests/kolla_container_tests/test_docker_worker.py b/tests/kolla_container_tests/test_docker_worker.py
index cd351e4ef..e32da66cb 100644
--- a/tests/kolla_container_tests/test_docker_worker.py
+++ b/tests/kolla_container_tests/test_docker_worker.py
@@ -1251,6 +1251,12 @@ class TestAttrComp(base.BaseTestCase):
         self.dw = get_DockerWorker({'privileged': False})
         self.assertTrue(self.dw.compare_privileged(container_info))
 
+    def test_compare_labels_image_missing(self):
+        container_info = {'Config': dict(Labels={'kolla_version': '1.0.1'})}
+        self.dw = get_DockerWorker({'labels': {}})
+        self.dw.check_image = mock.MagicMock(return_value=dict())
+        self.assertFalse(self.dw.compare_labels(container_info))
+
     def test_compare_labels_neg(self):
         container_info = {'Config': dict(Labels={'kolla_version': '2.0.1'})}
         self.dw = get_DockerWorker({'labels': {'kolla_version': '2.0.1'}})
diff --git a/tests/kolla_container_tests/test_podman_worker.py b/tests/kolla_container_tests/test_podman_worker.py
index 048a7a797..04d6d987b 100644
--- a/tests/kolla_container_tests/test_podman_worker.py
+++ b/tests/kolla_container_tests/test_podman_worker.py
@@ -1264,6 +1264,12 @@ class TestAttrComp(base.BaseTestCase):
         self.pw = get_PodmanWorker({'privileged': False})
         self.assertTrue(self.pw.compare_privileged(container_info))
 
+    def test_compare_labels_image_missing(self):
+        container_info = {'Config': dict(Labels={'kolla_version': '1.0.1'})}
+        self.pw = get_PodmanWorker({'labels': {}})
+        self.pw.check_image = mock.MagicMock(return_value=dict())
+        self.assertFalse(self.pw.compare_labels(container_info))
+
     def test_compare_labels_neg(self):
         container_info = {'Config': dict(Labels={'kolla_version': '2.0.1'})}
         self.pw = get_PodmanWorker({'labels': {'kolla_version': '2.0.1'}})