Update upgrade test case to support the validation of both 'spec.engi…

…neImage' and 'spec.image' fields. Ref: longhorn/longhorn#6777 Signed-off-by: Roger Yao <[email protected]>
longhorn · Sep 27, 2023 · 2f02f1d · 2f02f1d
1 parent bab40a4
commit 2f02f1d
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/manager/integration/tests/test_upgrade.py b/manager/integration/tests/test_upgrade.py
@@ -567,7 +567,12 @@ def test_upgrade(longhorn_upgrade_type,
         if v.name != restore_vol_name:
             volume = client.by_id_volume(v.name)
             engine = get_volume_engine(volume)
-            assert engine.image == new_ei.image
+            if hasattr(engine, 'engineImage'):
+                print("Checking engineImage...")
+                assert engine.engineImage == new_ei.image
+            else:
+                print("Checking image...")
+                assert engine.image == new_ei.image
             assert engine.currentImage == new_ei.image
 
     # Check All volumes data