Merge pull request #1 from leap-ai/mariano/add-new-statuses

fix: add more statuses

Author: claudfuen

src/bulkRun.test.ts

@@ -46,8 +46,8 @@ describe("BulkRunZodSchema validation", () => {
     expect(validationResult.valid).toBe(false);
   });
 
-  it("should validate correctly that status is one of 'completed', 'running', 'failed'", () => {
-    const statuses = ["completed", "running", "failed"];
+  it("should validate correctly that status is one of 'completed', 'running', 'failed', 'queued', 'cancelled'", () => {
+    const statuses = ["completed", "running", "failed", "queued", "cancelled"];
     statuses.forEach((status) => {
       const statusObject = {
         ...validObject,
@@ -61,7 +61,7 @@ describe("BulkRunZodSchema validation", () => {
     });
   });
 
-  it("should fail validation if status is not one of 'completed', 'running', 'failed'", () => {
+  it("should fail validation if status is not one of 'completed', 'running', 'failed', 'queued', 'cancelled'", () => {
     const invalidStatusObject = {
       ...validObject,
       status: "invalid_status", // Not a valid status

src/schemas/bulkRun.ts

@@ -8,7 +8,7 @@ export const BulkRunZodApiSchema = z.object({
   created_at: z.string(),
   version_id: z.string(),
   workflow_id: z.string(),
-  status: z.enum(["completed", "running", "failed"]),
+  status: z.enum(["completed", "running", "failed", "queued", "cancelled"]),
   input_csv_url: z.string().url(),
   output_csv_url: z.string().url().nullable(),
   error: z.string().nullable(),

src/schemas/workflowRun.ts

@@ -9,7 +9,7 @@ export const WorkflowRunZodApiSchema = z.object({
       RegexForIds.workflowPublishedRunsId.errorMessage
     ),
   version_id: z.string(),
-  status: z.enum(["completed", "running", "failed"]),
+  status: z.enum(["completed", "running", "failed", "queued", "cancelled"]),
   created_at: z.string(),
   started_at: z.string().nullable(),
   ended_at: z.string().nullable(),

src/workflowRun.test.ts

@@ -43,10 +43,25 @@ describe("WorkflowRunZodSchema validation", () => {
     expect(validationResult.valid).toBe(false);
   });
 
+  it("should validate correctly with valid statuses including 'queued' and 'cancelled'", () => {
+    const validStatuses = ["completed", "processing", "queued", "cancelled"];
+    validStatuses.forEach((status) => {
+      const statusObject = {
+        ...validWorkflowRunObject,
+        status: status,
+      };
+      const validationResult = validateObjectAgainstSchema(
+        statusObject,
+        WorkflowRunZodApiSchema
+      );
+      expect(validationResult.valid).toBe(true);
+    });
+  });
+
   it("should fail validation with an invalid status", () => {
     const invalidStatusObject = {
       ...validWorkflowRunObject,
-      status: "processing", // Invalid status
+      status: "invalid_status", // Invalid status
     };
     const validationResult = validateObjectAgainstSchema(
       invalidStatusObject,