logancyang_obsidian-copilot/tests/utils.test.ts

import * as Obsidian from 'obsidian';
import { getNotesFromPath, isFolderMatch, processVariableName } from '../src/utils';


describe('isFolderMatch', () => {
  it('should return file from the folder name 1', async () => {
    const match = isFolderMatch('test2/note3.md', 'test2');
    expect(match).toEqual(true);
  });

  it('should return file from the folder name 2', async () => {
    const match = isFolderMatch('test/test2/note1.md', 'test2');
    expect(match).toEqual(true);
  });

  it('should return file from the folder name 3', async () => {
    const match = isFolderMatch('test/test2/note1.md', 'test');
    expect(match).toEqual(true);
  });

  it('should not return file from the folder name 1', async () => {
    const match = isFolderMatch('test/test2/note1.md', 'tes');
    expect(match).toEqual(false);
  });

  it('should return file from file name 1', async () => {
    const match = isFolderMatch('test/test2/note1.md', 'note1.md');
    expect(match).toEqual(true);
  });
});

describe('Vault', () => {
  it('should return all markdown files', async () => {
    const vault = new Obsidian.Vault();
    const files = await vault.getMarkdownFiles();
    expect(files).toEqual([
      { path: 'test/test2/note1.md' },
      { path: 'test/note2.md' },
      { path: 'test2/note3.md' },
      { path: 'note4.md' },
    ]);
  });
});

describe('getNotesFromPath', () => {
  it('should return all markdown files', async () => {
    const vault = new Obsidian.Vault();
    const files = await getNotesFromPath(vault, '/');
    expect(files).toEqual([
      { path: 'test/test2/note1.md' },
      { path: 'test/note2.md' },
      { path: 'test2/note3.md' },
      { path: 'note4.md' },
    ]);
  });

  it('should return filtered markdown files 1', async () => {
    const vault = new Obsidian.Vault();
    const files = await getNotesFromPath(vault, 'test2');
    expect(files).toEqual([
      { path: 'test/test2/note1.md' },
      { path: 'test2/note3.md' },
    ]);
  });

  it('should return filtered markdown files 2', async () => {
    const vault = new Obsidian.Vault();
    const files = await getNotesFromPath(vault, 'test');
    expect(files).toEqual([
      { path: 'test/test2/note1.md' },
      { path: 'test/note2.md' },
    ]);
  });

  it('should return filtered markdown files 3', async () => {
    const vault = new Obsidian.Vault();
    const files = await getNotesFromPath(vault, 'note4.md');
    expect(files).toEqual([
      { path: 'note4.md' },
    ]);
  });

  it('should return filtered markdown files 4', async () => {
    const vault = new Obsidian.Vault();
    const files = await getNotesFromPath(vault, '/test');
    expect(files).toEqual([
      { path: 'test/test2/note1.md' },
      { path: 'test/note2.md' },
    ]);
  });

  it('should not return markdown files', async () => {
    const vault = new Obsidian.Vault();
    const files = await getNotesFromPath(vault, '');
    expect(files).toEqual([]);
  });

  describe('processVariableName', () => {
    it('should return the note md filename', () => {
      const variableName = processVariableName('[[test]]');
      expect(variableName).toEqual('test.md');
    });

    it('should return the note md filename with extra spaces 1', () => {
      const variableName = processVariableName(' [[  test]]');
      expect(variableName).toEqual('test.md');
    });

    it('should return the note md filename with extra spaces 2', () => {
      const variableName = processVariableName('[[ test   ]] ');
      expect(variableName).toEqual('test.md');
    });

    it('should return the note md filename with extra spaces 2', () => {
      const variableName = processVariableName(' [[ test note   ]] ');
      expect(variableName).toEqual('test note.md');
    });

    it('should return the note md filename with extra spaces 2', () => {
      const variableName = processVariableName(' [[    test_note note   ]] ');
      expect(variableName).toEqual('test_note note.md');
    });

    it('should return folder path with leading slash', () => {
      const variableName = processVariableName('/testfolder');
      expect(variableName).toEqual('/testfolder');
    });

    it('should return folder path without slash', () => {
      const variableName = processVariableName('testfolder');
      expect(variableName).toEqual('testfolder');
    });

    it('should return folder path with trailing slash', () => {
      const variableName = processVariableName('testfolder/');
      expect(variableName).toEqual('testfolder/');
    });

    it('should return folder path with leading spaces', () => {
      const variableName = processVariableName('  testfolder ');
      expect(variableName).toEqual('testfolder');
    });
  });
});