diff --git a/src/__tests__/retryMode.test.ts b/src/__tests__/retryMode.test.ts index 21cb927..039a562 100644 --- a/src/__tests__/retryMode.test.ts +++ b/src/__tests__/retryMode.test.ts @@ -132,20 +132,20 @@ describe('buildRetryTemplateVars', () => { expect(vars.movementDetails).toContain('Architect'); }); - it('should set hasPreviousOrder=false and empty previousOrderContent when previousOrderContent is null', () => { + it('should set hasOrderContent=false and empty orderContent when previousOrderContent is null (via ctx)', () => { const ctx = createRetryContext({ previousOrderContent: null }); const vars = buildRetryTemplateVars(ctx, 'en'); - expect(vars.hasPreviousOrder).toBe(false); - expect(vars.previousOrderContent).toBe(''); + expect(vars.hasOrderContent).toBe(false); + expect(vars.orderContent).toBe(''); }); - it('should set hasPreviousOrder=true and populate previousOrderContent when provided', () => { - const ctx = createRetryContext({ previousOrderContent: '# Order content' }); - const vars = buildRetryTemplateVars(ctx, 'en'); + it('should set hasOrderContent=true and populate orderContent when provided via parameter', () => { + const ctx = createRetryContext(); + const vars = buildRetryTemplateVars(ctx, 'en', '# Order content'); - expect(vars.hasPreviousOrder).toBe(true); - expect(vars.previousOrderContent).toBe('# Order content'); + expect(vars.hasOrderContent).toBe(true); + expect(vars.orderContent).toBe('# Order content'); }); it('should include retryNote when present', () => { diff --git a/src/__tests__/retrySlashCommand.test.ts b/src/__tests__/retrySlashCommand.test.ts index 5f36ac2..3a2f4cd 100644 --- a/src/__tests__/retrySlashCommand.test.ts +++ b/src/__tests__/retrySlashCommand.test.ts @@ -8,7 +8,7 @@ */ import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest'; -import { mkdirSync, writeFileSync, rmSync } from 'node:fs'; +import { mkdirSync, rmSync } from 'node:fs'; import { join } from 'node:path'; import { tmpdir } from 'node:os'; import { @@ -153,7 +153,7 @@ describe('/retry slash command', () => { setupProvider([]); const retryContext = buildRetryContext({ previousOrderContent: orderContent }); - const result = await runRetryMode(tmpDir, retryContext); + const result = await runRetryMode(tmpDir, retryContext, orderContent); expect(result.action).toBe('execute'); expect(result.task).toBe(orderContent); @@ -164,7 +164,7 @@ describe('/retry slash command', () => { setupProvider([]); const retryContext = buildRetryContext({ previousOrderContent: null }); - const result = await runRetryMode(tmpDir, retryContext); + const result = await runRetryMode(tmpDir, retryContext, null); expect(mockInfo).toHaveBeenCalledWith('No previous order found.'); expect(result.action).toBe('cancel'); @@ -176,7 +176,7 @@ describe('/retry slash command', () => { const capture = setupProvider(['I see the order content.']); const retryContext = buildRetryContext({ previousOrderContent: orderContent }); - await runRetryMode(tmpDir, retryContext); + await runRetryMode(tmpDir, retryContext, orderContent); expect(capture.systemPrompts.length).toBeGreaterThan(0); const systemPrompt = capture.systemPrompts[0]!; @@ -189,7 +189,7 @@ describe('/retry slash command', () => { const capture = setupProvider(['No order found.']); const retryContext = buildRetryContext({ previousOrderContent: null }); - await runRetryMode(tmpDir, retryContext); + await runRetryMode(tmpDir, retryContext, null); expect(capture.systemPrompts.length).toBeGreaterThan(0); const systemPrompt = capture.systemPrompts[0]!;