Merge pull request #4145 from dehilsterlexis/WU_ADD_PROTECT

feat: Add Protect & Unprotect functions to Visualization WU Class
hpcc-systems · Dec 18, 2023 · 16ee388 · 16ee388
2 parents e61ff70 + fb2b4e9
commit 16ee388
Show file tree

Hide file tree

Showing 4 changed files with 31 additions and 3 deletions.
diff --git a/packages/comms/src/ecl/workunit.ts b/packages/comms/src/ecl/workunit.ts
@@ -381,6 +381,14 @@ export class Workunit extends StateObject<UWorkunitState, IWorkunitState> implem
     abort() {
         return this.WUAction("Abort");
     }
+
+    protect() {
+        return this.WUAction("Protect");
+    }
+
+    unprotect() {
+        return this.WUAction("Unprotect");
+    }
 
     delete() {
         return this.WUAction("Delete");

diff --git a/packages/comms/src/services/wsWorkunits.ts b/packages/comms/src/services/wsWorkunits.ts
@@ -1646,7 +1646,7 @@ export namespace WUQueryDetails {
 }
 
 export namespace WUAction {
-    export type Type = "SetToFailed" | "Pause" | "PauseNow" | "Resume" | "Abort" | "Delete" | "Restore" | "Deschedule" | "Reschedule";
+    export type Type = "SetToFailed" | "Pause" | "PauseNow" | "Resume" | "Abort" | "Delete" | "Restore" | "Deschedule" | "Reschedule" | "Protect" | "Unprotect";
     export interface Request {
         Wuids: string[];
         WUActionType: Type;

diff --git a/tests/test-comms/.vscode/launch.json b/tests/test-comms/.vscode/launch.json
@@ -16,6 +16,9 @@
             "type": "msedge",
             "url": "file:///${workspaceRoot}/test.html",
             "webRoot": "${workspaceFolder}",
+            "runtimeArgs": [
+                "--disable-web-security"
+            ],
             "sourceMapPathOverrides": {
                 "webpack:///./*": "${workspaceRoot}/*",
                 "webpack:///*": "/*"

diff --git a/tests/test-comms/src/ecl/workunit.spec.ts b/tests/test-comms/src/ecl/workunit.spec.ts
@@ -57,6 +57,7 @@ allPeople;
         });
         it("result schema", function () {
             return wu1.fetchResults().then((results) => {
+                expect(wu1.isComplete(), "isComplete").is.true;
                 expect(results.length).equals(1);
                 expect(results[0].Name).to.equal("Result 1");
                 expect(results[0].Sequence).to.equal(0);
@@ -68,6 +69,7 @@ allPeople;
         });
         it("results", function () {
             return wu1.fetchResults().then((results) => {
+                expect(wu1.isComplete(), "isComplete").is.true;
                 expect(results.length).equals(1);
                 return wu1.CResults[0].fetchRows().then(response => {
                     expect(response.length).to.equal(3);
@@ -97,9 +99,24 @@ allPeople;
             await newWu.delete();
             expect(newWu.isDeleted(), "isDeleted").is.true;
         });
-        it("delete", function () {
+        it("protect", function () {
+            return wu1.protect().then(() => {
+                expect(wu1.Protected).to.be.true;
+            });
+        });
+        it("delete (protected - should fail)", function () {
+            return wu1.delete().then(function (response) {
+                expect(wu1.isDeleted(), "isDeleted").is.false;
+                return response;
+            });
+        });
+        it("unprotect", function () {
+            return wu1.unprotect().then(() => {
+                expect(wu1.Protected).to.be.false;
+            });
+        });
+        it("delete (unprotected)", function () {
             return wu1.delete().then(function (response) {
-                expect(wu1.isComplete(), "isComplete").is.true;
                 expect(wu1.isDeleted(), "isDeleted").is.true;
                 return response;
             });