@forgehive/task 0.2.0 → 0.2.1

package/src/test/add-listener.test.ts

@@ -1,28 +1,31 @@
-import { Task, type TaskRecord } from '../index'
-import { Schema, type InferSchema } from '@forgehive/schema'
-
-describe('Listener tests', () => {
-  it('Should record one item', async () => {
-    const tape: TaskRecord<{ value: number }, { value: number }>[] = []
-    const task = new Task(function (_argv: { value: number }) {
-      return _argv
+import { Task, type ExecutionRecord } from '../index'
+
+describe('Add listener', () => {
+  it('Should add a listener to the task', async () => {
+    const tape: ExecutionRecord[] = []
+    const task = new Task(async (argv: { value: number }) => {
+      return { value: argv.value }
     })
 
     task.addListener((record) => {
       tape.push(record)
     })
 
-    await task.run({ value: 5 })
-
-    expect(tape.length).toBe(1)
-    expect(tape[0].input).toEqual({ value: 5 })
-    expect(tape[0].output).toEqual({ value: 5 })
+    await task.run({ value: 1 })
+    expect(tape).toEqual([{
+      input: { value: 1 },
+      output: { value: 1 },
+      boundaries: {},
+      taskName: undefined,
+      metadata: {},
+      type: 'success'
+    }])
   })
 
-  it('Should record execution error', async () => {
-    const tape: TaskRecord<{ value: number }, never>[] = []
-    const task = new Task(function (_argv: { value: number }) {
-      throw new Error('This should happen')
+  it('Should add a listener to the task and catch error', async () => {
+    const tape: ExecutionRecord[] = []
+    const task = new Task(async (_argv: { value: number }) => {
+      throw new Error('Test error')
     })
 
     task.addListener((record) => {
@@ -30,83 +33,106 @@ describe('Listener tests', () => {
     })
 
     try {
-      await task.run({ value: 5 })
-    } catch (e) {
-      // Error is expected
+      await task.run({ value: 1 })
+    } catch (error) {
+      // Expected error
     }
 
-    expect(tape.length).toBe(1)
-    expect(tape[0].input).toEqual({ value: 5 })
-    expect(tape[0].error).toBe('This should happen')
+    expect(tape).toEqual([{
+      input: { value: 1 },
+      error: 'Test error',
+      boundaries: {},
+      taskName: undefined,
+      metadata: {},
+      type: 'error'
+    }])
   })
 
-  it('Should record validation error', async () => {
-    const tape: TaskRecord<{ value: number | null }, { value: number }>[] = []
-    const schema = new Schema({
-      value: Schema.number().min(5)
-    })
-
-    const task = new Task(function (_argv: InferSchema<typeof schema>) {
-      return _argv
-    }, {
-      schema
+  it('Should add a listener to the task and catch error with null input', async () => {
+    const tape: ExecutionRecord[] = []
+    const task = new Task(async (argv: { value: number | null }) => {
+      if (argv.value === null) {
+        throw new Error('Value is null')
+      }
+      return { value: argv.value }
     })
 
-    task.addListener<{ value: number | null }, { value: number }>((record) => {
+    task.addListener((record) => {
       tape.push(record)
     })
 
     try {
-      await task.run({ value: 3 })
-    } catch (e) {
-      // Error is expected
+      await task.run({ value: null })
+    } catch (error) {
+      // Expected error
     }
 
-    expect(tape.length).toBe(1)
-    expect(tape[0].input).toEqual({ value: 3 })
-    expect(tape[0].error).toContain('Invalid input on:')
-    expect(tape[0].error).toContain('value:')
-    expect(tape[0].error).toContain('Number must be greater than or equal to 5')
+    expect(tape).toEqual([{
+      input: { value: null },
+      error: 'Value is null',
+      boundaries: {},
+      taskName: undefined,
+      metadata: {},
+      type: 'error'
+    }])
   })
 
-  it('Should record multiple records', async () => {
-    const tape: TaskRecord<{ value: number }, { value: number }>[] = []
-    const task = new Task(function (_argv: { value: number }) {
-      return _argv
+  it('Should add a listener to the task and call twice', async () => {
+    const tape: ExecutionRecord[] = []
+    const task = new Task(async (argv: { value: number }) => {
+      return { value: argv.value }
     })
 
     task.addListener((record) => {
       tape.push(record)
     })
 
-    await task.run({ value: 5 })
-    await task.run({ value: 6 })
-
-    expect(tape.length).toBe(2)
-    expect(tape[0].input).toEqual({ value: 5 })
-    expect(tape[0].output).toEqual({ value: 5 })
-
-    expect(tape[1].input).toEqual({ value: 6 })
-    expect(tape[1].output).toEqual({ value: 6 })
+    await task.run({ value: 1 })
+    await task.run({ value: 2 })
+
+    expect(tape).toEqual([
+      {
+        input: { value: 1 },
+        output: { value: 1 },
+        boundaries: {},
+        taskName: undefined,
+        metadata: {},
+        type: 'success'
+      },
+      {
+        input: { value: 2 },
+        output: { value: 2 },
+        boundaries: {},
+        taskName: undefined,
+        metadata: {},
+        type: 'success'
+      }
+    ])
   })
 
-  it('Should stop recording', async () => {
-    const tape: TaskRecord<{ value: number }, { value: number }>[] = []
-    const task = new Task(function (_argv: { value: number }) {
-      return _argv
+  it('Should add a listener and remove it', async () => {
+    const tape: ExecutionRecord[] = []
+    const task = new Task(async (argv: { value: number }) => {
+      return { value: argv.value }
     })
 
     task.addListener((record) => {
       tape.push(record)
     })
 
-    await task.run({ value: 5 })
+    await task.run({ value: 1 })
 
     task.removeListener()
-    await task.run({ value: 6 })
 
-    expect(tape.length).toBe(1)
-    expect(tape[0].input).toEqual({ value: 5 })
-    expect(tape[0].output).toEqual({ value: 5 })
+    await task.run({ value: 2 })
+
+    expect(tape).toEqual([{
+      input: { value: 1 },
+      output: { value: 1 },
+      boundaries: {},
+      taskName: undefined,
+      metadata: {},
+      type: 'success'
+    }])
   })
 })

package/src/test/task-with-boundaries.test.ts

@@ -1,4 +1,4 @@
-import { createTask, Schema, TaskRecord, BoundaryTapeData } from '../index'
+import { createTask, Schema, ExecutionRecord, BoundaryTapeData } from '../index'
 
 // Need to add proxy cache mode to the boundaries
 describe('Boundaries tasks tests', () => {
@@ -148,7 +148,7 @@ describe('Boundaries tasks tests', () => {
     }
 
     // Use the correct type definition for records
-    const records: TaskRecord<{value: number}, Promise<number>>[] = []
+    const records: ExecutionRecord[] = []
 
     // Create a schema for the task that accepts a number
     const schema = new Schema({
@@ -192,7 +192,7 @@ describe('Boundaries tasks tests', () => {
     expect(records.length).toBe(3)
 
     // Sort records by input value for consistent testing
-    const sortedRecords = [...records].sort((a, b) => a.input.value - b.input.value)
+    const sortedRecords = [...records].sort((a, b) => (a.input as {value: number}).value - (b.input as {value: number}).value)
 
     // Check record for first task (value: 2)
     expect(sortedRecords[0].input).toEqual({ value: 2 })