Dockery/Trilium
1
0
Fork 0
mirror of https://github.com/zadam/trilium.git synced 2026-05-07 06:57:16 +02:00
Code Issues Packages Projects Releases Activity

test(ocr): broken due to change in architecture

Browse Source
This commit is contained in:
Elian Doran
2026-04-09 17:53:06 +03:00
parent acf9aa8b41
commit bfb9df48b1
1 changed files with 62 additions and 79 deletions
Download Patch File Download Diff File Expand all files Collapse all files

141
packages/trilium-core/src/services/search/search_result_ocr.spec.ts
View File

@@ -1,92 +1,75 @@
import { beforeEach, describe, expect, it, vi } from 'vitest';
import { beforeEach, describe, expect, it } from "vitest";
const mockBecca = {
notes: {} as Record<string, any>,
getNote: vi.fn()
};
import becca from "../../becca/becca.js";
import BNote from "../../becca/entities/bnote.js";
import { buildNote } from "../../test/becca_easy_mocking.js";
import SearchResult from "./search_result.js";
const mockBeccaService = {
getNoteTitleForPath: vi.fn()
};
describe("SearchResult", () => {
let note: BNote;
vi.mock('../../becca/becca.js', () => ({
default: mockBecca
}));
vi.mock('../../becca/becca_service.js', () => ({
default: mockBeccaService
}));
let SearchResult: any;
beforeEach(async () => {
vi.clearAllMocks();
mockBeccaService.getNoteTitleForPath.mockReturnValue('Test Note Title');
mockBecca.notes['test123'] = {
noteId: 'test123',
title: 'Test Note',
isInHiddenSubtree: vi.fn().mockReturnValue(false)
};
const module = await import('./search_result.js');
SearchResult = module.default;
});
describe('SearchResult', () => {
describe('constructor', () => {
it('should initialize with note path array', () => {
const searchResult = new SearchResult(['root', 'folder', 'test123']);
expect(searchResult.notePathArray).toEqual(['root', 'folder', 'test123']);
expect(searchResult.noteId).toBe('test123');
expect(searchResult.notePath).toBe('root/folder/test123');
expect(searchResult.score).toBe(0);
expect(mockBeccaService.getNoteTitleForPath).toHaveBeenCalledWith(['root', 'folder', 'test123']);
beforeEach(() => {
becca.reset();
note = buildNote({
id: "test123",
title: "Test Note"
});
});
describe('computeScore', () => {
let searchResult: any;
describe("constructor", () => {
it("should initialize with note path array", () => {
const searchResult = new SearchResult([note.noteId]);
expect(searchResult.notePathArray).toEqual(["test123"]);
expect(searchResult.noteId).toBe("test123");
expect(searchResult.notePath).toBe("test123");
expect(searchResult.score).toBe(0);
expect(searchResult.notePathTitle).toBe("Test Note");
});
});
describe("computeScore", () => {
let searchResult: SearchResult;
beforeEach(() => {
searchResult = new SearchResult(['root', 'test123']);
searchResult = new SearchResult([note.noteId]);
});
describe('basic scoring', () => {
it('should give highest score for exact note ID match', () => {
searchResult.computeScore('test123', ['test123']);
describe("basic scoring", () => {
it("should give highest score for exact note ID match", () => {
searchResult.computeScore("test123", ["test123"]);
expect(searchResult.score).toBeGreaterThanOrEqual(1000);
});
it('should give high score for exact title match', () => {
searchResult.computeScore('test note', ['test', 'note']);
it("should give high score for exact title match", () => {
searchResult.computeScore("test note", ["test", "note"]);
expect(searchResult.score).toBeGreaterThan(2000);
});
it('should give medium score for title prefix match', () => {
searchResult.computeScore('test', ['test']);
it("should give medium score for title prefix match", () => {
searchResult.computeScore("test", ["test"]);
expect(searchResult.score).toBeGreaterThan(500);
});
it('should give lower score for title word match', () => {
mockBecca.notes['test123'].title = 'This is a test note';
searchResult.computeScore('test', ['test']);
it("should give lower score for title word match", () => {
note.title = "This is a test note";
searchResult.computeScore("test", ["test"]);
expect(searchResult.score).toBeGreaterThan(300);
});
});
describe('hidden notes penalty', () => {
it('should apply penalty for hidden notes', () => {
mockBecca.notes['test123'].isInHiddenSubtree.mockReturnValue(true);
describe("hidden notes penalty", () => {
it("should apply penalty for hidden notes", () => {
const hiddenNote = buildNote({
id: "_hidden",
title: "Test Note"
});
const hiddenSearchResult = new SearchResult([hiddenNote.noteId]);
searchResult.computeScore('test', ['test']);
const hiddenScore = searchResult.score;
hiddenSearchResult.computeScore("test", ["test"]);
const hiddenScore = hiddenSearchResult.score;
mockBecca.notes['test123'].isInHiddenSubtree.mockReturnValue(false);
searchResult.score = 0;
searchResult.computeScore('test', ['test']);
searchResult.computeScore("test", ["test"]);
const normalScore = searchResult.score;
expect(normalScore).toBeGreaterThan(hiddenScore);
@@ -95,51 +78,51 @@ describe('SearchResult', () => {
});
});
describe('addScoreForStrings', () => {
let searchResult: any;
describe("addScoreForStrings", () => {
let searchResult: SearchResult;
beforeEach(() => {
searchResult = new SearchResult(['root', 'test123']);
searchResult = new SearchResult([note.noteId]);
});
it('should give highest score for exact token match', () => {
searchResult.addScoreForStrings(['sample'], 'sample text', 1.0);
it("should give highest score for exact token match", () => {
searchResult.addScoreForStrings(["sample"], "sample text", 1.0);
const exactScore = searchResult.score;
searchResult.score = 0;
searchResult.addScoreForStrings(['sample'], 'sampling text', 1.0);
searchResult.addScoreForStrings(["sample"], "sampling text", 1.0);
const prefixScore = searchResult.score;
searchResult.score = 0;
searchResult.addScoreForStrings(['sample'], 'text sample text', 1.0);
searchResult.addScoreForStrings(["sample"], "text sample text", 1.0);
const partialScore = searchResult.score;
expect(exactScore).toBeGreaterThan(prefixScore);
expect(exactScore).toBeGreaterThanOrEqual(partialScore);
});
it('should apply factor multiplier correctly', () => {
searchResult.addScoreForStrings(['sample'], 'sample text', 2.0);
it("should apply factor multiplier correctly", () => {
searchResult.addScoreForStrings(["sample"], "sample text", 2.0);
const doubleFactorScore = searchResult.score;
searchResult.score = 0;
searchResult.addScoreForStrings(['sample'], 'sample text', 1.0);
searchResult.addScoreForStrings(["sample"], "sample text", 1.0);
const singleFactorScore = searchResult.score;
expect(doubleFactorScore).toBe(singleFactorScore * 2);
});
it('should handle multiple tokens', () => {
searchResult.addScoreForStrings(['hello', 'world'], 'hello world test', 1.0);
it("should handle multiple tokens", () => {
searchResult.addScoreForStrings(["hello", "world"], "hello world test", 1.0);
expect(searchResult.score).toBeGreaterThan(0);
});
it('should be case insensitive', () => {
searchResult.addScoreForStrings(['sample'], 'sample text', 1.0);
it("should be case insensitive", () => {
searchResult.addScoreForStrings(["sample"], "sample text", 1.0);
const lowerCaseScore = searchResult.score;
searchResult.score = 0;
searchResult.addScoreForStrings(['sample'], 'SAMPLE text', 1.0);
searchResult.addScoreForStrings(["sample"], "SAMPLE text", 1.0);
const upperCaseScore = searchResult.score;
expect(upperCaseScore).toEqual(lowerCaseScore);