@outputai/core 0.14.0 → 0.14.1-next.1ec4416.0

package/src/tracing/processors/s3/redis_client.spec.js

@@ -1,179 +0,0 @@
-import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
-
-const logCalls = { warn: [], error: [] };
-vi.mock( '#logger', () => ( {
-  createChildLogger: () => ( {
-    warn: ( ...args ) => logCalls.warn.push( args ),
-    error: ( ...args ) => logCalls.error.push( args )
-  } )
-} ) );
-
-const getVarsMock = vi.fn();
-vi.mock( './configs.js', () => ( { getVars: () => getVarsMock() } ) );
-
-const createClientImpl = vi.fn();
-vi.mock( 'redis', () => ( { createClient: opts => createClientImpl( opts ) } ) );
-
-async function loadModule() {
-  vi.resetModules();
-  return import( './redis_client.js' );
-}
-
-describe( 'tracing/processors/s3/redis_client', () => {
-  beforeEach( () => {
-    vi.clearAllMocks();
-    getVarsMock.mockReturnValue( {} );
-    logCalls.warn = [];
-    logCalls.error = [];
-  } );
-
-  afterEach( () => {
-    vi.useRealTimers();
-  } );
-
-  it( 'throws when config redisUrl is missing', async () => {
-    getVarsMock.mockReturnValue( {} );
-    const { getRedisClient } = await loadModule();
-    await expect( getRedisClient() ).rejects.toThrow();
-  } );
-
-  it( 'creates client with url, connects once, then reuses cached when ping is PONG', async () => {
-    getVarsMock.mockReturnValue( { redisUrl: 'redis://localhost:6379' } );
-
-    const pingMock = vi.fn().mockResolvedValue( 'PONG' );
-    const connectMock = vi.fn().mockResolvedValue();
-    const created = [];
-    createClientImpl.mockImplementation( opts => {
-      created.push( opts );
-      return { connect: connectMock, ping: pingMock };
-    } );
-
-    const { getRedisClient } = await loadModule();
-
-    const c1 = await getRedisClient();
-    const c2 = await getRedisClient();
-
-    expect( created ).toHaveLength( 1 );
-    expect( connectMock ).toHaveBeenCalledTimes( 1 );
-    expect( pingMock ).toHaveBeenCalledTimes( 1 );
-    expect( c1 ).toBe( c2 );
-    expect( created[0] ).toMatchObject( { url: 'redis://localhost:6379', socket: { keepAlive: 15000 } } );
-  } );
-
-  it( 'closes stale client and reconnects when ping fails', async () => {
-    getVarsMock.mockReturnValue( { redisUrl: 'redis://localhost:6379' } );
-
-    const quitMock = vi.fn().mockResolvedValue();
-    const connectMock = vi.fn().mockResolvedValue();
-    const pingMock = vi.fn()
-      .mockResolvedValueOnce( 'PONG' )
-      .mockRejectedValueOnce( new Error( 'Connection lost' ) )
-      .mockResolvedValueOnce( 'PONG' );
-
-    const created = [];
-    createClientImpl.mockImplementation( opts => {
-      created.push( opts );
-      return { connect: connectMock, ping: pingMock, quit: quitMock };
-    } );
-
-    const { getRedisClient } = await loadModule();
-
-    const c1 = await getRedisClient();
-    const c2 = await getRedisClient();
-    expect( c1 ).toBe( c2 );
-    expect( created ).toHaveLength( 1 );
-
-    const c3 = await getRedisClient();
-    expect( quitMock ).toHaveBeenCalledTimes( 1 );
-    expect( created ).toHaveLength( 2 );
-    expect( c3 ).not.toBe( c1 );
-  } );
-
-  it( 'reconnects successfully even when quit() on stale client rejects', async () => {
-    getVarsMock.mockReturnValue( { redisUrl: 'redis://localhost:6379' } );
-
-    const quitMock = vi.fn().mockRejectedValue( new Error( 'Quit failed' ) );
-    const connectMock = vi.fn().mockResolvedValue();
-    const pingMock = vi.fn()
-      .mockResolvedValueOnce( 'PONG' )
-      .mockRejectedValueOnce( new Error( 'Connection lost' ) )
-      .mockResolvedValueOnce( 'PONG' );
-
-    const created = [];
-    createClientImpl.mockImplementation( opts => {
-      created.push( opts );
-      return { connect: connectMock, ping: pingMock, quit: quitMock };
-    } );
-
-    const { getRedisClient } = await loadModule();
-
-    const c1 = await getRedisClient();
-    const c1again = await getRedisClient();
-    expect( c1 ).toBe( c1again );
-    expect( created ).toHaveLength( 1 );
-
-    const c2 = await getRedisClient();
-    expect( quitMock ).toHaveBeenCalledTimes( 1 );
-    expect( created ).toHaveLength( 2 );
-    expect( c2 ).not.toBe( c1 );
-  } );
-
-  it( 'wraps connect() errors with code and cleans up failed client', async () => {
-    getVarsMock.mockReturnValue( { redisUrl: 'redis://localhost:6379' } );
-
-    const connectErr = new Error( 'Connection refused' );
-    connectErr.code = 'ECONNREFUSED';
-    const connectMock = vi.fn().mockRejectedValue( connectErr );
-    const quitMock = vi.fn().mockResolvedValue();
-
-    createClientImpl.mockImplementation( () => ( {
-      connect: connectMock,
-      quit: quitMock
-    } ) );
-
-    const { getRedisClient } = await loadModule();
-
-    try {
-      await getRedisClient();
-      expect.fail( 'Should have thrown' );
-    } catch ( err ) {
-      expect( err.message ).toBe( 'Failed to connect to Redis: Connection refused (ECONNREFUSED)' );
-      expect( err.cause ).toBe( connectErr );
-    }
-
-    expect( quitMock ).toHaveBeenCalledTimes( 1 );
-  } );
-
-  it( 'logs ping failures with error level', async () => {
-    getVarsMock.mockReturnValue( { redisUrl: 'redis://localhost:6379' } );
-
-    const pingErr = new Error( 'Connection reset' );
-    pingErr.code = 'ECONNRESET';
-    const pingMock = vi.fn()
-      .mockResolvedValueOnce( 'PONG' )
-      .mockRejectedValueOnce( pingErr )
-      .mockResolvedValueOnce( 'PONG' );
-    const connectMock = vi.fn().mockResolvedValue();
-    const quitMock = vi.fn().mockResolvedValue();
-
-    createClientImpl.mockImplementation( () => ( {
-      connect: connectMock,
-      ping: pingMock,
-      quit: quitMock
-    } ) );
-
-    const { getRedisClient } = await loadModule();
-
-    // First call: state.client is null, creates client (no ping)
-    await getRedisClient();
-    // Second call: pings existing client, returns PONG
-    await getRedisClient();
-    // Third call: pings existing client, fails with pingErr, reconnects
-    await getRedisClient();
-
-    expect( logCalls.error ).toContainEqual( [
-      'Redis ping failed',
-      { error: 'Connection reset', code: 'ECONNRESET' }
-    ] );
-  } );
-} );

package/src/tracing/processors/s3/s3_client.spec.js

@@ -1,68 +0,0 @@
-import { describe, it, expect, vi, beforeEach } from 'vitest';
-
-const getVarsMock = vi.fn();
-vi.mock( './configs', () => ( { getVars: () => getVarsMock() } ) );
-
-const ctorState = { args: null };
-class S3ClientMock {
-  constructor( args ) {
-    ctorState.args = args;
-  }
-}
-vi.mock( '@aws-sdk/client-s3', () => ( {
-  S3Client: S3ClientMock
-} ) );
-
-const uploadDoneMock = vi.fn();
-const uploadCtorState = { args: [] };
-class UploadMock {
-  constructor( args ) {
-    uploadCtorState.args.push( args );
-  }
-
-  done = uploadDoneMock;
-}
-vi.mock( '@aws-sdk/lib-storage', () => ( { Upload: UploadMock } ) );
-
-async function loadModule() {
-  vi.resetModules();
-  return import( './s3_client.js' );
-}
-
-describe( 'tracing/processors/s3/s3_client', () => {
-  beforeEach( () => {
-    vi.clearAllMocks();
-    ctorState.args = null;
-    uploadCtorState.args = [];
-    uploadDoneMock.mockResolvedValue( undefined );
-    getVarsMock.mockReturnValue( {
-      awsRegion: 'us-east-1',
-      awsAccessKeyId: 'id',
-      awsSecretAccessKey: 'sek',
-      remoteS3Bucket: 'bucket'
-    } );
-  } );
-
-  it( 'creates client once with config and uploads with bucket/key/content', async () => {
-    const { upload } = await loadModule();
-
-    await upload( { key: 'wf/key.json', content: '{"a":1}' } );
-
-    expect( ctorState.args ).toEqual( {
-      region: 'us-east-1',
-      credentials: { secretAccessKey: 'sek', accessKeyId: 'id' }
-    } );
-    expect( uploadCtorState.args ).toHaveLength( 1 );
-    expect( uploadCtorState.args[0] ).toEqual( {
-      client: expect.any( S3ClientMock ),
-      params: { Bucket: 'bucket', Key: 'wf/key.json', Body: '{"a":1}' }
-    } );
-    expect( uploadDoneMock ).toHaveBeenCalledTimes( 1 );
-
-    // subsequent upload uses cached client
-    await upload( { key: 'wf/key2.json', content: '{}' } );
-    expect( uploadCtorState.args ).toHaveLength( 2 );
-    expect( uploadDoneMock ).toHaveBeenCalledTimes( 2 );
-  } );
-} );
-

package/src/tracing/tools/build_trace_tree.spec.js

@@ -1,216 +0,0 @@
-import { describe, it, expect } from 'vitest';
-import { EventAction } from '../trace_consts.js';
-import { Attribute } from '#trace_attribute';
-import buildTraceTree from './build_trace_tree.js';
-
-describe( 'build_trace_tree', () => {
-  it( 'returns null when entries is empty', () => {
-    expect( buildTraceTree( [] ) ).toBeNull();
-  } );
-
-  it( 'sets root output with a fixed message when workflow has no end/error action yet', () => {
-    const entries = [
-      { kind: 'workflow', id: 'wf', parentId: undefined, action: EventAction.START, name: 'wf', details: {}, timestamp: 1000 }
-    ];
-    const result = buildTraceTree( entries );
-    expect( result ).not.toBeNull();
-    expect( result.output ).toBe( '<<Workflow did not finish yet. If this workflows is supposed to have been completed already, \
-this can indicate it timed out or was interrupted.>>' );
-    expect( result.endedAt ).toBeNull();
-  } );
-
-  it( 'returns null when there is no root (all entries have parentId)', () => {
-    const entries = [
-      { id: 'a', parentId: 'x', action: EventAction.START, name: 'a', timestamp: 1 },
-      { id: 'b', parentId: 'a', action: EventAction.START, name: 'b', timestamp: 2 }
-    ];
-    expect( buildTraceTree( entries ) ).toBeNull();
-  } );
-
-  it( 'add_attr action stores attribute details by type on node.attributes', () => {
-    const requestCount = {
-      type: Attribute.HTTPRequestCount.TYPE,
-      url: 'https://api.example.test',
-      requestId: 'req-1'
-    };
-    const requestCost = {
-      type: Attribute.HTTPRequestCost.TYPE,
-      url: 'https://api.example.test',
-      requestId: 'req-1',
-      total: 0.2
-    };
-    const entries = [
-      { kind: 'workflow', id: 'wf', parentId: undefined, action: EventAction.START, name: 'wf', details: {}, timestamp: 100 },
-      { kind: 'step', id: 's', parentId: 'wf', action: EventAction.START, name: 'step', details: {}, timestamp: 200 },
-      { id: 's', action: EventAction.ADD_ATTR, details: requestCount, timestamp: 250 },
-      { id: 's', action: EventAction.ADD_ATTR, details: requestCost, timestamp: 260 },
-      { id: 'wf', action: EventAction.END, details: {}, timestamp: 300 }
-    ];
-    const result = buildTraceTree( entries );
-    expect( result ).not.toBeNull();
-    expect( result.children[0].attributes ).toEqual( {
-      [Attribute.HTTPRequestCount.TYPE]: requestCount,
-      [Attribute.HTTPRequestCost.TYPE]: requestCost
-    } );
-  } );
-
-  it( 'add_attr action overwrites prior value for the same attribute type', () => {
-    const firstCost = {
-      type: Attribute.HTTPRequestCost.TYPE,
-      url: 'https://api.example.test',
-      requestId: 'req-1',
-      total: 1
-    };
-    const secondCost = {
-      type: Attribute.HTTPRequestCost.TYPE,
-      url: 'https://api.example.test',
-      requestId: 'req-1',
-      total: 2
-    };
-    const entries = [
-      { kind: 'workflow', id: 'wf', parentId: undefined, action: EventAction.START, name: 'wf', details: {}, timestamp: 1 },
-      { id: 'wf', action: EventAction.ADD_ATTR, details: firstCost, timestamp: 2 },
-      { id: 'wf', action: EventAction.ADD_ATTR, details: secondCost, timestamp: 3 },
-      { id: 'wf', action: EventAction.END, details: {}, timestamp: 4 }
-    ];
-    const result = buildTraceTree( entries );
-    expect( result.attributes ).toEqual( { [Attribute.HTTPRequestCost.TYPE]: secondCost } );
-  } );
-
-  it( 'add_attr does not attach nodes as children (only start does)', () => {
-    const entries = [
-      { kind: 'workflow', id: 'wf', parentId: undefined, action: EventAction.START, name: 'wf', details: {}, timestamp: 1 },
-      {
-        id: 'orphan',
-        parentId: 'wf',
-        action: EventAction.ADD_ATTR,
-        details: { type: Attribute.HTTPRequestCount.TYPE, url: 'https://api.example.test', requestId: 'req-1' },
-        timestamp: 2
-      },
-      { id: 'wf', action: EventAction.END, details: {}, timestamp: 3 }
-    ];
-    const result = buildTraceTree( entries );
-    expect( result.children ).toHaveLength( 0 );
-    expect( result.attributes ).toEqual( {} );
-  } );
-
-  it( 'error action sets error and endedAt on node', () => {
-    const entries = [
-      { kind: 'wf', id: 'r', parentId: undefined, action: EventAction.START, name: 'root', details: {}, timestamp: 100 },
-      { kind: 'step', id: 's', parentId: 'r', action: EventAction.START, name: 'step', details: {}, timestamp: 200 },
-      { id: 's', action: EventAction.ERROR, details: { message: 'failed' }, timestamp: 300 }
-    ];
-    const result = buildTraceTree( entries );
-    expect( result ).not.toBeNull();
-    expect( result.children ).toHaveLength( 1 );
-    expect( result.children[0].error ).toEqual( { message: 'failed' } );
-    expect( result.children[0].endedAt ).toBe( 300 );
-  } );
-
-  it( 'builds a tree from workflow/step/IO entries with grouping and sorting', () => {
-    const stepAttribute = {
-      type: Attribute.HTTPRequestCount.TYPE,
-      url: 'https://api.example.test/step-1',
-      requestId: 'req-step-1'
-    };
-    const entries = [
-      // workflow start
-      { kind: 'workflow', action: EventAction.START, name: 'wf', id: 'wf', parentId: undefined, details: { a: 1 }, timestamp: 1000 },
-      // evaluator start/stop
-      { kind: 'evaluator', action: EventAction.START, name: 'eval', id: 'eval', parentId: 'wf', details: { z: 0 }, timestamp: 1500 },
-      { id: 'eval', action: EventAction.END, details: { z: 1 }, timestamp: 1600 },
-      // step1 start
-      { kind: 'step', action: EventAction.START, name: 'step-1', id: 's1', parentId: 'wf', details: { x: 1 }, timestamp: 2000 },
-      { id: 's1', action: EventAction.ADD_ATTR, details: stepAttribute, timestamp: 2050 },
-      // IO under step1
-      { kind: 'IO', action: EventAction.START, name: 'test-1', id: 'io1', parentId: 's1', details: { y: 2 }, timestamp: 2300 },
-      // step2 start
-      { kind: 'step', action: EventAction.START, name: 'step-2', id: 's2', parentId: 'wf', details: { x: 2 }, timestamp: 2400 },
-      // IO under step2
-      { kind: 'IO', action: EventAction.START, name: 'test-2', id: 'io2', parentId: 's2', details: { y: 3 }, timestamp: 2500 },
-      { id: 'io2', action: EventAction.END, details: { y: 4 }, timestamp: 2600 },
-      // IO under step1 ends
-      { id: 'io1', action: EventAction.END, details: { y: 5 }, timestamp: 2700 },
-      // step1 end
-      { id: 's1', action: EventAction.END, details: { done: true }, timestamp: 2800 },
-      // step2 end
-      { id: 's2', action: EventAction.END, details: { done: true }, timestamp: 2900 },
-      // workflow end
-      { id: 'wf', action: EventAction.END, details: { ok: true }, timestamp: 3000 }
-    ];
-
-    const result = buildTraceTree( entries );
-
-    const expected = {
-      id: 'wf',
-      kind: 'workflow',
-      name: 'wf',
-      startedAt: 1000,
-      endedAt: 3000,
-      input: { a: 1 },
-      output: { ok: true },
-      attributes: {},
-      children: [
-        {
-          id: 'eval',
-          kind: 'evaluator',
-          name: 'eval',
-          startedAt: 1500,
-          endedAt: 1600,
-          input: { z: 0 },
-          output: { z: 1 },
-          attributes: {},
-          children: []
-        },
-        {
-          id: 's1',
-          kind: 'step',
-          name: 'step-1',
-          startedAt: 2000,
-          endedAt: 2800,
-          input: { x: 1 },
-          output: { done: true },
-          attributes: { [Attribute.HTTPRequestCount.TYPE]: stepAttribute },
-          children: [
-            {
-              id: 'io1',
-              kind: 'IO',
-              name: 'test-1',
-              startedAt: 2300,
-              endedAt: 2700,
-              input: { y: 2 },
-              output: { y: 5 },
-              attributes: {},
-              children: []
-            }
-          ]
-        },
-        {
-          id: 's2',
-          kind: 'step',
-          name: 'step-2',
-          startedAt: 2400,
-          endedAt: 2900,
-          input: { x: 2 },
-          output: { done: true },
-          attributes: {},
-          children: [
-            {
-              id: 'io2',
-              kind: 'IO',
-              name: 'test-2',
-              startedAt: 2500,
-              endedAt: 2600,
-              input: { y: 3 },
-              output: { y: 4 },
-              attributes: {},
-              children: []
-            }
-          ]
-        }
-      ]
-    };
-
-    expect( result ).toMatchObject( expected );
-  } );
-} );