import { mock } from 'vitest-mock-extended';

import type { AgentChatAttachmentService } from '../agent-chat-attachment.service';
import { AgentTestChatService, chatThreadId } from '../agent-test-chat.service';
import type { N8nMemory } from '../integrations/n8n-memory';

const agentId = 'agent-1';
const userId = 'user-1';
type MemoryImplementation = ReturnType<N8nMemory['getImplementation']>;

function makeService() {
	const n8nMemory = mock<N8nMemory>();
	const memory = mock<MemoryImplementation>();
	const attachmentService = mock<AgentChatAttachmentService>();
	n8nMemory.getImplementation.mockReturnValue(memory);

	return {
		service: new AgentTestChatService(n8nMemory, attachmentService),
		n8nMemory,
		memory,
		attachmentService,
	};
}

describe('AgentTestChatService', () => {
	it('derives stable agent and user-scoped test-chat thread ids', () => {
		expect(chatThreadId(agentId)).toBe(`test-${agentId}`);
		expect(chatThreadId(agentId, userId)).toBe(`test-${agentId}:${userId}`);
	});

	it('loads messages from the user-scoped thread and memory resource', async () => {
		const { service, memory } = makeService();
		const messages = [{ id: 'message-1' }];
		memory.getMessages.mockResolvedValue(messages as never);

		await expect(service.getTestChatMessages(agentId, userId)).resolves.toBe(messages);
		expect(memory.getMessages).toHaveBeenCalledWith(`test-${agentId}:${userId}`, {
			resourceId: `draft-chat:${userId}`,
		});
	});

	it('clears one user thread without changing all-agent cleanup', async () => {
		const { service, memory, attachmentService } = makeService();

		await service.clearTestChatMessages(agentId, userId);
		expect(memory.deleteThread).toHaveBeenCalledWith(`test-${agentId}:${userId}`);
		expect(attachmentService.deleteByThread).toHaveBeenCalledWith(`test-${agentId}:${userId}`, {
			agentId,
		});

		await service.clearAllTestChatMessages(agentId);
		expect(memory.deleteThreadsByPrefix).toHaveBeenCalledWith(`test-${agentId}`);
		expect(memory.deleteMessagesByThread).toHaveBeenCalledWith(`test-${agentId}`);
		expect(memory.deleteThread).toHaveBeenCalledWith(`test-${agentId}`);
	});
});