From fa0695d910e4cc2ca3211b0a0b4da3c04794cc4c Mon Sep 17 00:00:00 2001
From: derekbit <derekbit@users.noreply.github.com>
Date: Wed, 27 Sep 2023 04:28:05 +0000
Subject: [PATCH] Update testing docs

Signed-off-by: derekbit <derekbit@users.noreply.github.com>
---
 integration/test_upgrade.html | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/integration/test_upgrade.html b/integration/test_upgrade.html
index d43df825ae..0a79886a55 100644
--- a/integration/test_upgrade.html
+++ b/integration/test_upgrade.html
@@ -595,7 +595,12 @@ <h1 class="title">Module <code>tests.test_upgrade</code></h1>
         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, &#39;engineImage&#39;):
+                print(&#34;Checking engineImage...&#34;)
+                assert engine.engineImage == new_ei.image
+            else:
+                print(&#34;Checking image...&#34;)
+                assert engine.image == new_ei.image
             assert engine.currentImage == new_ei.image
 
     # Check All volumes data
@@ -1270,7 +1275,12 @@ <h2 id="prerequisite">Prerequisite</h2>
         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, &#39;engineImage&#39;):
+                print(&#34;Checking engineImage...&#34;)
+                assert engine.engineImage == new_ei.image
+            else:
+                print(&#34;Checking image...&#34;)
+                assert engine.image == new_ei.image
             assert engine.currentImage == new_ei.image
 
     # Check All volumes data