Add to VFM context menu action "Add pause upon completion".

[vid.git] / vid-webpack-master / src / app / drawingBoard / service-planning / objectsToTree / models / vfModule / vfModule.model.info.spec.ts

diff --git a/vid-webpack-master/src/app/drawingBoard/service-planning/objectsToTree/models/vfModule/vfModule.model.info.spec.ts b/vid-webpack-master/src/app/drawingBoard/service-planning/objectsToTree/models/vfModule/vfModule.model.info.spec.ts
index c225687..42ee238 100644 (file)
--- a/vid-webpack-master/src/app/drawingBoard/service-planning/objectsToTree/models/vfModule/vfModule.model.info.spec.ts
+++ b/vid-webpack-master/src/app/drawingBoard/service-planning/objectsToTree/models/vfModule/vfModule.model.info.spec.ts
@@ -20,6 +20,7 @@ import {VfModuleUpgradePopupService} from "../../../../../shared/components/gene
 import {instance, mock, when} from "ts-mockito";
 import each from "jest-each";
 import {VfModule} from "../../../../../shared/models/vfModule";
+import {VfModuleTreeNode} from "../../../../../shared/models/vfModuleTreeNode";
 
 class MockAppStore<T> {
   getState() {
@@ -127,6 +128,26 @@ describe('VFModule Model Info', () => {
     expect(nextLevel).toBeNull();
   });
 
+  each([
+    ['afterCompletion', 'afterCompletion'],
+    ['undefined', undefined]
+  ]).
+  test('createNode should return pauseInstantiation status %s', (description, pauseInstantiationStatus) => {
+    const modelName: string = "vfModuleModelName";
+    const vfModuleInstance =  {
+      "vfModuleInstanceName": {
+        "pauseInstantiation": pauseInstantiationStatus,
+      }
+    };
+    const currentModel = {};
+    const parentModel = {};
+    const serviceModelId = "serviceModelId";
+
+    let actual: VfModuleTreeNode = vfModuleModel.createNode(<any>vfModuleInstance, <any>currentModel, <any>parentModel, modelName, 0, serviceModelId);
+    let expected: string = pauseInstantiationStatus;
+    expect(actual.pauseInstantiation).toEqual(expected);
+  });
+
   test('getModel should return Module model', () => {
     let model = vfModuleModel.getModel({
       "uuid": "25284168-24bb-4698-8cb4-3f509146eca5",