1 files changed, 9 insertions, 1 deletions
diff --git a/vid-webpack-master/cypress/support/steps/drawingBoard/drawingBoardTree.steps.ts b/vid-webpack-master/cypress/support/steps/drawingBoard/drawingBoardTree.steps.ts
index a940ae15e..3bd60d785 100644
--- a/vid-webpack-master/cypress/support/steps/drawingBoard/drawingBoardTree.steps.ts
+++ b/vid-webpack-master/cypress/support/steps/drawingBoard/drawingBoardTree.steps.ts
@@ -25,6 +25,10 @@ function drawingBoardTreeClickOnContextMenuOptionByName(optionName : string) : C
       return cy.getElementByDataTestsId('context-menu-edit').click({force : true});
     case 'Delete':
       return cy.getElementByDataTestsId('context-menu-delete').trigger('mouseover').click();
+    case 'Upgrade':
+      return cy.getElementByDataTestsId('context-menu-upgrade').trigger('mouseover').click();
+    case 'Undo Upgrade':
+      return cy.getElementByDataTestsId('context-menu-undoUpgrade').trigger('mouseover').click();
     default:
       return cy.getElementByDataTestsId('context-menu-duplicate').click({force : true});
   }
@@ -57,6 +61,11 @@ function IsDeleteTagShownOnNode(index: number)
   cy.getElementByDataTestsId('delete-status-type').eq(index).should("contain.text", "Delete").should("contain.css", "opacity", "1");
 }
 
+function IsUpgradeTagShownOnNode(index: number)
+{
+  cy.getElementByDataTestsId('upgrade-status-type').eq(index).should("contain.text", "Upgrade").should("contain.css", "opacity", "1");
+}
+
 function IsDeleteTagNotShownOnNode(index: number)
 {
   cy.getElementByDataTestsId('delete-status-type').eq(index).should("contain.text", "Delete").should("contain.css", "opacity", "0");
@@ -68,6 +77,5 @@ Cypress.Commands.add('nodeWithLineThrough', nodeWithLineThrough);
 Cypress.Commands.add('nodeWithoutLineThrough', nodeWithoutLineThrough);
 Cypress.Commands.add('IsDeleteTagShownOnNode', IsDeleteTagShownOnNode);
 Cypress.Commands.add('IsDeleteTagNotShownOnNode', IsDeleteTagNotShownOnNode);
-
 Cypress.Commands.add('isNodeDeleted', isNodeDeleted);
 Cypress.Commands.add('isNodeNotDeleted', isNodeNotDeleted);