add support for missing engine statuses

Author: stepansergeevitch

src/service/engine/model.ts

@@ -52,7 +52,7 @@ export class EngineModel {
   }
 
   async delete() {
-    if (this.current_status_summary == EngineStatusSummary.DELETING) {
+    if (this.current_status_summary == EngineStatusSummary.DRAINING) {
       return;
     }
     const query = `DROP ENGINE "${this.name}"`;

src/service/engine/types.ts

@@ -26,11 +26,10 @@ export function processEngineStatus(
 }
 
 export enum EngineStatusSummary {
-  DELETING = "Dropping",
-  REPAIRING = "Repairing",
-  RUNNING = "Running",
   STARTING = "Starting",
-  STARTING_INITIALIZING = "Started",
+  RUNNING = "Running",
+  RESIZING = "Resizing",
+  DRAINING = "Draining",
   STOPPED = "Stopped",
   STOPPING = "Stopping",
   // Specific for V1, here are just for typing match
@@ -39,7 +38,9 @@ export enum EngineStatusSummary {
   RESTARTING = "ENGINE_STATUS_SUMMARY_RESTARTING",
   RESTARTING_INITIALIZING = "ENGINE_STATUS_SUMMARY_RESTARTING_INITIALIZING",
   UNSPECIFIED = "ENGINE_STATUS_SUMMARY_UNSPECIFIED",
-  UPGRADING = "ENGINE_STATUS_SUMMARY_UPGRADING"
+  UPGRADING = "ENGINE_STATUS_SUMMARY_UPGRADING",
+  REPAIRING = "ENGINE_STATUS_SUMMARY_REPAIRING",
+  STARTING_INITIALIZING = "ENGINE_STATUS_SUMMARY_STARTING_INITIALIZING",
 }
 
 export enum EngineType {

src/service/engine/v1/types.ts

@@ -9,7 +9,7 @@ export type Engine = {
 };
 
 export enum EngineStatusSummary {
-  DELETING = "ENGINE_STATUS_SUMMARY_DELETING",
+  DRAINING = "ENGINE_STATUS_SUMMARY_DELETING",
   REPAIRING = "ENGINE_STATUS_SUMMARY_REPAIRING",
   RUNNING = "ENGINE_STATUS_SUMMARY_RUNNING",
   STARTING = "ENGINE_STATUS_SUMMARY_STARTING",

test/unit/v2/engine.test.ts

@@ -589,6 +589,23 @@ describe("engine service", () => {
     delete process.env.ENGINE_NAME;
   });
 
+  it("Parse engine statuses", async () => {
+    // Taken from https://docs.firebolt.io/Overview/engine-fundamentals.html#viewing-and-understanding-engine-status
+    const statuses = [
+      ["STARTING", EngineStatusSummary.STARTING],
+      ["RUNNING", EngineStatusSummary.RUNNING],
+      ["RESIZING", EngineStatusSummary.RESIZING],
+      ["DRAINING", EngineStatusSummary.DRAINING],
+      ["STOPPING", EngineStatusSummary.STOPPING],
+      ["STOPPED", EngineStatusSummary.STOPPED]
+    ];
+    for (const [rawStatus, status] of statuses) {
+      expect(processEngineStatus(rawStatus)).toEqual(status);
+    }
+    expect(processEngineStatus(undefined)).toBe(undefined);
+    expect(processEngineStatus("unexisting")).not.toBeTruthy();
+  });
+
   it("Parses different engine statuses correctly", async () => {
     const statuses = ["RUNNING", "Running", "running"];
     for (const status of statuses) {