search-engine.test.js

const { describe, it, beforeEach, afterEach } = require('node:test');
const assert = require('node:assert');
const { createSearchEngine } = require('./search-engine');
const { createCacheDb } = require('./cache/db');
const { createMigrationRunner } = require('./cache/migrate');

describe('search-engine boundary', () => {
  let cache;

  beforeEach(() => {
    cache = createCacheDb({ cacheDbPath: ':memory:' });
    cache.open();

    const migrationRunner = createMigrationRunner({ db: cache.getDatabase() });
    migrationRunner.runMigrations();
  });

  afterEach(() => {
    if (cache) {
      cache.close();
    }
  });

  function insertSession(id, title, directory, timeCreated, timeUpdated) {
    cache.run(
      `INSERT INTO session (id, title, directory, parent_id, time_created, time_updated, model, agent)
       VALUES (?, ?, ?, ?, ?, ?, ?, ?)`,
      [id, title, directory, null, timeCreated, timeUpdated, '', '']
    );
  }

  function insertMessage(id, sessionId, timeCreated, role) {
    cache.run(
      `INSERT INTO message (id, session_id, time_created, data, role)
       VALUES (?, ?, ?, ?, ?)`,
      [id, sessionId, timeCreated, JSON.stringify({ role }), role]
    );
  }

  function insertPart(id, messageId, timeCreated, data) {
    const textContent = (() => {
      try {
        const parsed = JSON.parse(data || '{}');
        if (parsed.type === 'text' && typeof parsed.text === 'string') {
          return parsed.text;
        }
        return '';
      } catch (e) {
        return '';
      }
    })();

    cache.run(
      `INSERT INTO part (id, message_id, time_created, data, text_content)
       VALUES (?, ?, ?, ?, ?)`,
      [id, messageId, timeCreated, data, textContent]
    );
  }

  it('returns empty results for empty query', async () => {
    const engine = createSearchEngine({ cacheDb: cache.getDatabase() });
    const result = await engine.searchSessions('');
    assert.deepStrictEqual(result.results, []);
  });

  it('returns empty results for whitespace-only query', async () => {
    const engine = createSearchEngine({ cacheDb: cache.getDatabase() });
    const result = await engine.searchSessions('   ');
    assert.deepStrictEqual(result.results, []);
  });

  it('returns sessions matching title with score 1.0', async () => {
    insertSession('sess-1', 'Auth Bug Fix', '/home/dsp/project', 1000, 2000);

    const engine = createSearchEngine({ cacheDb: cache.getDatabase() });
    const result = await engine.searchSessions('auth');

    assert.strictEqual(result.results.length, 1);
    assert.strictEqual(result.results[0].id, 'sess-1');
    assert.strictEqual(result.results[0].title, 'Auth Bug Fix');
    assert.strictEqual(result.results[0].project, 'project');
    assert.strictEqual(result.results[0].score, 1.0);
  });

  it('returns sessions matching directory with score 1.0', async () => {
    insertSession('sess-1', 'Some Session', '/home/dsp/dashboard', 1000, 2000);

    const engine = createSearchEngine({ cacheDb: cache.getDatabase() });
    const result = await engine.searchSessions('dashboard');

    assert.strictEqual(result.results.length, 1);
    assert.strictEqual(result.results[0].project, 'dashboard');
    assert.strictEqual(result.results[0].score, 1.0);
  });

  it('returns sessions matching message content with score 0.8', async () => {
    insertSession('sess-1', 'My Session', '/home/dsp/project', 1000, 2000);
    insertMessage('msg-1', 'sess-1', 1100, 'user');
    insertPart('part-1', 'msg-1', 1110, JSON.stringify({ type: 'text', text: 'The authentication token is invalid' }));

    const engine = createSearchEngine({ cacheDb: cache.getDatabase() });
    const result = await engine.searchSessions('authentication');

    assert.strictEqual(result.results.length, 1);
    assert.strictEqual(result.results[0].id, 'sess-1');
    assert.strictEqual(result.results[0].title, 'My Session');
    assert.strictEqual(result.results[0].score, 0.8);
    assert.ok(result.results[0].snippet.includes('authentication'));
  });

  it('deduplicates session and message matches, boosting score', async () => {
    insertSession('sess-1', 'Auth Fix', '/home/dsp/project', 1000, 2000);
    insertMessage('msg-1', 'sess-1', 1100, 'user');
    insertPart('part-1', 'msg-1', 1110, JSON.stringify({ type: 'text', text: 'Fix the auth bug' }));

    const engine = createSearchEngine({ cacheDb: cache.getDatabase() });
    const result = await engine.searchSessions('auth');

    assert.strictEqual(result.results.length, 1);
    assert.strictEqual(result.results[0].id, 'sess-1');
    assert.ok(result.results[0].score >= 0.85);
    assert.ok(result.results[0].snippet.includes('auth'));
  });

  it('extracts snippet with context around match', async () => {
    insertSession('sess-1', 'Session', null, 1000, 2000);
    insertMessage('msg-1', 'sess-1', 1100, 'user');
    insertPart('part-1', 'msg-1', 1110, JSON.stringify({
      type: 'text',
      text: 'prefix text here is the keyword and more suffix text after'
    }));

    const engine = createSearchEngine({ cacheDb: cache.getDatabase() });
    const result = await engine.searchSessions('keyword');

    assert.strictEqual(result.results.length, 1);
    assert.ok(result.results[0].snippet.includes('keyword'));
    assert.ok(result.results[0].snippet.includes('prefix'));
    assert.ok(result.results[0].snippet.includes('suffix'));
  });

  it('truncates long snippets', async () => {
    const longText = 'xyzzy '.repeat(100);
    insertSession('sess-1', 'Session', null, 1000, 2000);
    insertMessage('msg-1', 'sess-1', 1100, 'user');
    insertPart('part-1', 'msg-1', 1110, JSON.stringify({ type: 'text', text: longText }));

    const engine = createSearchEngine({ cacheDb: cache.getDatabase() });
    const result = await engine.searchSessions('xyzzy');

    assert.ok(result.results[0].snippet.length < 200);
  });

  it('falls back to Untitled when title is missing', async () => {
    insertSession('sess-1', null, '/home/dsp/project', 1000, 2000);

    const engine = createSearchEngine({ cacheDb: cache.getDatabase() });
    const result = await engine.searchSessions('project');

    assert.strictEqual(result.results[0].title, 'Untitled');
  });

  it('abbreviates directory to project name', async () => {
    insertSession('sess-1', 'Test', '/home/dsp/my-project', 1000, 2000);

    const engine = createSearchEngine({ cacheDb: cache.getDatabase() });
    const result = await engine.searchSessions('test');

    assert.strictEqual(result.results[0].project, 'my-project');
  });

  it('handles tilde abbreviation for /home/dsp', async () => {
    insertSession('sess-1', 'Test', '/home/dsp', 1000, 2000);

    const engine = createSearchEngine({ cacheDb: cache.getDatabase() });
    const result = await engine.searchSessions('test');

    assert.strictEqual(result.results[0].project, '~');
  });

  it('ranks title matches above content matches', async () => {
    insertSession('sess-title', 'Bug Report', null, 1000, 2000);
    insertSession('sess-content', 'Other', null, 1000, 2000);
    insertMessage('msg-1', 'sess-content', 1100, 'user');
    insertPart('part-1', 'msg-1', 1110, JSON.stringify({ type: 'text', text: 'bug in the code' }));

    const engine = createSearchEngine({ cacheDb: cache.getDatabase() });
    const result = await engine.searchSessions('bug');

    assert.strictEqual(result.results.length, 2);
    assert.strictEqual(result.results[0].id, 'sess-title');
    assert.strictEqual(result.results[0].score, 1.0);
    assert.strictEqual(result.results[1].id, 'sess-content');
    assert.strictEqual(result.results[1].score, 0.8);
  });

  it('escapes single quotes in search query safely', async () => {
    insertSession('sess-1', "It's a Bug", '/home/dsp/project', 1000, 2000);
    insertMessage('msg-1', 'sess-1', 1100, 'user');
    insertPart('part-1', 'msg-1', 1110, JSON.stringify({ type: 'text', text: "It's a bug" }));

    const engine = createSearchEngine({ cacheDb: cache.getDatabase() });
    // Should not throw — the adapter uses parameterized queries
    const result = await engine.searchSessions("it's a bug");

    assert.strictEqual(result.results.length, 1);
    assert.strictEqual(result.results[0].id, 'sess-1');
  });

  it('handles tool part data gracefully', async () => {
    insertSession('sess-1', 'Session', null, 1000, 2000);
    insertMessage('msg-1', 'sess-1', 1100, 'user');
    insertPart('part-1', 'msg-1', 1110, JSON.stringify({ type: 'tool', tool: 'read_file', state: { input: { path: '/tmp/test' } } }));

    const engine = createSearchEngine({ cacheDb: cache.getDatabase() });
    // Searching for session title still works; tool parts have no text_content for FTS5
    const result = await engine.searchSessions('session');

    assert.strictEqual(result.results.length, 1);
    assert.strictEqual(result.results[0].title, 'Session');
  });

  it('handles invalid JSON in part data gracefully', async () => {
    insertSession('sess-1', 'Session', null, 1000, 2000);
    insertMessage('msg-1', 'sess-1', 1100, 'user');
    insertPart('part-1', 'msg-1', 1110, 'not json at all');

    const engine = createSearchEngine({ cacheDb: cache.getDatabase() });
    // Searching for session title still works; invalid JSON has no text_content for FTS5
    const result = await engine.searchSessions('session');

    assert.strictEqual(result.results.length, 1);
    assert.strictEqual(result.results[0].title, 'Session');
  });
});