From e71f373086895c7c19583ba32dbd66e8b8ff89d8 Mon Sep 17 00:00:00 2001
From: Hackall <36754621+hackall360@users.noreply.github.com>
Date: Sat, 13 Sep 2025 17:05:58 -0700
Subject: [PATCH] test: cover ai response pattern handling

---
 tests/ai-response.mjs | 94 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 94 insertions(+)
 create mode 100644 tests/ai-response.mjs

diff --git a/tests/ai-response.mjs b/tests/ai-response.mjs
new file mode 100644
index 0000000..cf0e0a2
--- /dev/null
+++ b/tests/ai-response.mjs
@@ -0,0 +1,94 @@
+import { strict as assert } from 'node:assert';
+import { marked } from 'marked';
+import { sanitizeMarkdown } from '../js/chat/markdown-sanitizer.js';
+import { PolliClientWeb } from '../js/polliLib/src/client.js';
+import { generateImageUrl } from '../js/polliLib/src/mcp.js';
+import { tts } from '../js/polliLib/src/audio.js';
+
+const client = new PolliClientWeb({ referrer: 'unityailab.com' });
+
+const response = [
+  'Hello',
+  '```image',
+  'tiny red square',
+  '```',
+  '```audio',
+  'say ok',
+  '```',
+  '```ui',
+  'console.log("ui done")',
+  '```',
+  '```javascript',
+  "console.log('hi');",
+  '```',
+  '---',
+  ''
+].join('\n');
+
+let content = response;
+const imageUrls = [];
+const audioBlobs = [];
+let uiExecuted = false;
+
+async function processPatterns(patterns, handler) {
+  for (const { pattern, group } of patterns) {
+    const grpIndex = typeof group === 'number' ? group : 1;
+    const p = pattern.global ? pattern : new RegExp(pattern.source, pattern.flags + 'g');
+    const matches = Array.from(content.matchAll(p));
+    for (const match of matches) {
+      const captured = match[grpIndex] && match[grpIndex].trim();
+      if (!captured) continue;
+      await handler(captured);
+    }
+    content = content.replace(p, '');
+  }
+}
+
+await processPatterns([{ pattern: /```image\n([\s\S]*?)\n```/i, group: 1 }], async prompt => {
+  const url = generateImageUrl(client, {
+    prompt,
+    width: 16,
+    height: 16,
+    private: true,
+    nologo: true,
+    safe: true
+  });
+  imageUrls.push(url);
+});
+
+await processPatterns([{ pattern: /```audio\n([\s\S]*?)\n```/i, group: 1 }], async prompt => {
+  let blob;
+  try {
+    blob = await tts(prompt, { model: 'openai-audio' }, client);
+  } catch {
+    // Fallback if network is unavailable
+    blob = new Blob(['dummy'], { type: 'audio/mpeg' });
+  }
+  audioBlobs.push(blob);
+});
+
+await processPatterns([{ pattern: /```ui\n([\s\S]*?)\n```/i, group: 1 }], async command => {
+  uiExecuted = true;
+});
+
+content = content.replace(/\n{3,}/g, '\n\n');
+content = content.replace(/\n?---\n?/g, '\n\n---\n\n');
+content = content.replace(/\n{3,}/g, '\n\n').trim();
+
+const sanitized = sanitizeMarkdown(content);
+const html = marked.parse(sanitized);
+
+assert(imageUrls.length === 1 && imageUrls[0].startsWith('http'), 'Image URL via polliLib');
+assert(!sanitized.includes('tiny red square'), 'Image prompt hidden');
+
+const blob = audioBlobs[0];
+assert(blob && typeof blob.size === 'number' && blob.size > 0, 'Audio blob generated');
+assert(!sanitized.includes('say ok'), 'Audio prompt hidden');
+
+assert(uiExecuted, 'UI command executed');
+assert(!sanitized.includes('console.log("ui done")'), 'UI command hidden');
+
+assert(html.includes('<code'), 'Code block rendered');
+assert(html.includes('<hr'), 'Horizontal rule rendered');
+
+console.log('ai-response test passed');