fix(screenshot-undo): do not fail on revert screenshot with different…

… state name
gemini-testing · Oct 2, 2023 · 5c51487 · 5c51487
1 parent fd771fc
commit 5c51487
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/lib/report-builder/gui.js b/lib/report-builder/gui.js
@@ -115,7 +115,8 @@ module.exports = class GuiReportBuilder extends StaticReportBuilder {
         }
 
         const previousExpectedPath = _.get(previousImage, 'expectedImg.path', null);
-        const shouldRemoveReference = _.isNull(previousImage.refImg.size);
+        const previousImageRefImgSize = _.get(previousImage, 'refImg.size', null);
+        const shouldRemoveReference = _.isNull(previousImageRefImgSize);
         const shouldRevertReference = !shouldRemoveReference;
 
         let updatedImage, removedResult;

diff --git a/test/unit/lib/report-builder/gui.js b/test/unit/lib/report-builder/gui.js
@@ -365,7 +365,7 @@ describe('GuiReportBuilder', () => {
         let reportBuilder;
 
         function stubResultData_(result, {resultId, stateName}) {
-            const res = _.defaultsDeep(result, {
+            const res = _.defaults(result, {
                 imageId: 'imageId',
                 status: UPDATED,
                 timestamp: 100500,
@@ -475,10 +475,7 @@ describe('GuiReportBuilder', () => {
             const resultId = 'result-id';
             const stateName = 's-name';
             const formattedResult = mkFormattedResultStub_({id: resultId, stateName});
-            const previousImage = {
-                expectedImg: {path: 'previous-expected-path'},
-                refImg: {path: 'previous-reference-path', size: null}
-            };
+            const previousImage = null;
             const updatedImage = {
                 id: 'updated-img-id',
                 parentId: 'parent-id'