Testing

import { test } from 'node:test';
import assert from 'node:assert/strict';
import OpenAI from 'openai';
import { Shimmy, testing } from '@rfa-labs/shimmy';

const shimmy = new Shimmy();
const openai = shimmy.wrap(new OpenAI());

/** The code under test: two Steps, in order. */
async function triage(ticket: string): Promise<string> {
  return shimmy.run('triage', async () => {
    const label = await shimmy.step('classify', () =>
      openai.chat.completions.create({
        model: 'gpt-5.5',
        messages: [{ role: 'user', content: `Classify: ${ticket}` }],
      }),
    );
    const reply = await shimmy.step('draft_reply', () =>
      openai.chat.completions.create({
        model: 'gpt-5.5',
        messages: [
          { role: 'user', content: `Reply to a ${label.choices[0]?.message.content} ticket.` },
        ],
      }),
    );
    return reply.choices[0]?.message.content ?? '';
  });
}

test('triage classifies, then drafts', async () => {
  testing.reset();
  const reply = await triage('The export button does nothing');

  assert.ok(reply.length > 0);
  testing.assertStepsInOrder(['classify', 'draft_reply']);
  testing.assertStepCalled('classify', 1);
});

test('triage surfaces a rate limit instead of hiding it', async () => {
  testing.reset();
  // The next `classify` call fails the way the provider does: openai's own
  // RateLimitError, so this exercises your real error handling.
  await testing.withFault('classify', 'rate_limit', () =>
    assert.rejects(triage('The export button does nothing'), OpenAI.RateLimitError),
  );
  testing.assertStepNotCalled('draft_reply');
});

// vitest.config.ts — resets the call log and faults before every test.
// (jest: setupFilesAfterEnv. node --test: call testing.reset() yourself.)
import { defineConfig } from 'vitest/config';

export default defineConfig({
  test: { setupFiles: ['@rfa-labs/shimmy/testing-setup'] },
});