mynode/packages/lib/services/search/SearchEngine.semantic.test.ts

import Note from '../../models/Note';
import { setUpSemanticSearch, tearDownSemanticSearch, updateSearchIndex } from '../../testing/ai/semanticSearch';
import { setupDatabaseAndSynchronizer, switchClient } from '../../testing/test-utils';
import SearchEngine, { SearchType } from './SearchEngine';

describe('SearchEngine.semantic', () => {
	let engine: SearchEngine = null;

	beforeEach(async () => {
		await setupDatabaseAndSynchronizer(1);
		await switchClient(1);

		await setUpSemanticSearch();
		engine = SearchEngine.instance();
	});

	afterEach(async () => {
		await tearDownSemanticSearch();
	});

	it('should include semantic results in general search output', async () => {
		await Note.save({ title: 'test', body: 'letter letter letter' });

		await updateSearchIndex();

		const rows = await engine.search('letters letter letter');
		expect(rows).toHaveLength(1);
		expect(rows[0]).toMatchObject({ searchType: [SearchType.Semantic] });
	});

	it('should not use semantic search when the user has specified a field to search in', async () => {
		await Note.save({ title: 'test', body: 'letter' });

		await updateSearchIndex();

		const rows = await engine.search('body:letters letter letter');
		expect(rows).toHaveLength(0);
	});

	it('should not use semantic search when searching for an item ID-like pattern', async () => {
		const testUuid = '5d17d3649e0449209db61fdbf6b27d14';
		await Note.save({ title: 'test', body: testUuid });

		await updateSearchIndex();

		expect(await engine.search(testUuid)).toHaveLength(1);
		expect(await engine.search(testUuid.replace(/.$/, 'a'))).toHaveLength(0);
	});
});