unbound-cli 1.3.2 → 1.4.0

package/test/policy-ai-assist-preflight.test.js

@@ -0,0 +1,66 @@
+const { test } = require('node:test');
+const assert = require('node:assert/strict');
+
+const {
+  validatePromptPreflight,
+  OUT_OF_SCOPE_KEYWORDS,
+  MAX_PROMPT_LEN,
+} = require('../src/lib/policy-ai-assist');
+
+// Spec §6.2 step 3 / §6.9: trim then check length; 1800 chars passes, 1801 rejects
+// with backend-equivalent wording.
+
+test('preflight: prompt of 1799 chars passes', () => {
+  const r = validatePromptPreflight({ prompt: 'a'.repeat(1799) });
+  assert.equal(r.warnings.length, 0);
+});
+
+test('preflight: prompt of 1800 chars passes (boundary)', () => {
+  const r = validatePromptPreflight({ prompt: 'a'.repeat(MAX_PROMPT_LEN) });
+  assert.equal(r.warnings.length, 0);
+});
+
+test('preflight: prompt of 1801 chars rejects with verbatim wording', () => {
+  assert.throws(
+    () => validatePromptPreflight({ prompt: 'a'.repeat(1801) }),
+    new RegExp(`^Error: Input is too long \\(max ${MAX_PROMPT_LEN} characters\\)\\.$`)
+  );
+});
+
+// §6.2 step 4: parametrized over the keyword list — every token surfaces a warning.
+for (const token of OUT_OF_SCOPE_KEYWORDS) {
+  test(`preflight: out-of-scope keyword "${token}" produces a warning`, () => {
+    const r = validatePromptPreflight({ prompt: `block X in ${token} mode` });
+    assert.ok(r.warnings.includes(token), `expected warning for ${token}, got ${JSON.stringify(r.warnings)}`);
+  });
+}
+
+// §6.2 step 5: collapse runs of >=2 newlines to a single newline.
+test('preflight: collapses three consecutive newlines to one', () => {
+  const r = validatePromptPreflight({ prompt: 'line1\n\n\nline2' });
+  assert.equal(r.sanitizedPrompt, 'line1\nline2');
+});
+
+test('preflight: collapses two consecutive newlines to one', () => {
+  const r = validatePromptPreflight({ prompt: 'a\n\nb' });
+  assert.equal(r.sanitizedPrompt, 'a\nb');
+});
+
+test('preflight: a single newline is left alone', () => {
+  const r = validatePromptPreflight({ prompt: 'a\nb' });
+  assert.equal(r.sanitizedPrompt, 'a\nb');
+});
+
+// §6.2: warning is informational only; under --yes the caller logs it and
+// continues. The helper itself never throws for warnings — it just reports.
+test('preflight: warnings do not throw', () => {
+  // Use a keyword that is NOT a substring of "production" (the keyword set
+  // contains "prod" which would also match): pick a non-overlapping one.
+  const r = validatePromptPreflight({ prompt: 'block weekend pushes' });
+  assert.ok(r.warnings.includes('weekend'));
+  // No throw — the function returned and we got here.
+});
+
+test('preflight: missing prompt throws', () => {
+  assert.throws(() => validatePromptPreflight({}), /--prompt is required/);
+});