Trilium/spec/search/value_extractor.spec.ts

import becca_mocking from "./becca_mocking.js";
import ValueExtractor from "../../src/services/search/value_extractor.js";
import becca from "../../src/becca/becca.js";
import SearchContext from "../../src/services/search/search_context.js";

const dsc = new SearchContext();

describe("Value extractor", () => {
    beforeEach(() => {
        becca.reset();
    });

    it("simple title extraction", async () => {
        const europe = becca_mocking.note("Europe").note;

        const valueExtractor = new ValueExtractor(dsc, ["note", "title"]);

        expect(valueExtractor.validate()).toBeFalsy();
        expect(valueExtractor.extract(europe)).toEqual("Europe");
    });

    it("label extraction", async () => {
        const austria = becca_mocking.note("Austria").label("Capital", "Vienna").note;

        let valueExtractor = new ValueExtractor(dsc, ["note", "labels", "capital"]);

        expect(valueExtractor.validate()).toBeFalsy();
        expect(valueExtractor.extract(austria)).toEqual("Vienna");

        valueExtractor = new ValueExtractor(dsc, ["#capital"]);

        expect(valueExtractor.validate()).toBeFalsy();
        expect(valueExtractor.extract(austria)).toEqual("Vienna");
    });

    it("parent/child property extraction", async () => {
        const vienna = becca_mocking.note("Vienna");
        const europe = becca_mocking.note("Europe").child(becca_mocking.note("Austria").child(vienna));

        let valueExtractor = new ValueExtractor(dsc, ["note", "children", "children", "title"]);

        expect(valueExtractor.validate()).toBeFalsy();
        expect(valueExtractor.extract(europe.note)).toEqual("Vienna");

        valueExtractor = new ValueExtractor(dsc, ["note", "parents", "parents", "title"]);

        expect(valueExtractor.validate()).toBeFalsy();
        expect(valueExtractor.extract(vienna.note)).toEqual("Europe");
    });

    it("extract through relation", async () => {
        const czechRepublic = becca_mocking.note("Czech Republic").label("capital", "Prague");
        const slovakia = becca_mocking.note("Slovakia").label("capital", "Bratislava");
        const austria = becca_mocking.note("Austria").relation("neighbor", czechRepublic.note).relation("neighbor", slovakia.note);

        let valueExtractor = new ValueExtractor(dsc, ["note", "relations", "neighbor", "labels", "capital"]);

        expect(valueExtractor.validate()).toBeFalsy();
        expect(valueExtractor.extract(austria.note)).toEqual("Prague");

        valueExtractor = new ValueExtractor(dsc, ["~neighbor", "labels", "capital"]);

        expect(valueExtractor.validate()).toBeFalsy();
        expect(valueExtractor.extract(austria.note)).toEqual("Prague");
    });
});

describe("Invalid value extractor property path", () => {
    it('each path must start with "note" (or label/relation)', () => expect(new ValueExtractor(dsc, ["neighbor"]).validate()).toBeTruthy());

    it("extra path element after terminal label", () => expect(new ValueExtractor(dsc, ["~neighbor", "labels", "capital", "noteId"]).validate()).toBeTruthy());

    it("extra path element after terminal title", () => expect(new ValueExtractor(dsc, ["note", "title", "isProtected"]).validate()).toBeTruthy());

    it("relation name and note property is missing", () => expect(new ValueExtractor(dsc, ["note", "relations"]).validate()).toBeTruthy());

    it("relation is specified but target note property is not specified", () => expect(new ValueExtractor(dsc, ["note", "relations", "myrel"]).validate()).toBeTruthy());
});