Jacky Zhao
2025-03-12 270a5dc14ac3654c3aac94dfca222bab0d091474
quartz/util/fileTrie.test.ts
@@ -5,6 +5,7 @@
interface TestData {
  title: string
  slug: string
  filePath: string
}
describe("FileTrie", () => {
@@ -26,6 +27,7 @@
      const data = {
        title: "Test Title",
        slug: "test",
        filePath: "test.md",
      }
      trie.add(data)
@@ -36,6 +38,7 @@
      const data = {
        title: "Test Title",
        slug: "test",
        filePath: "test.md",
      }
      trie.add(data)
@@ -49,6 +52,7 @@
      const data = {
        title: "Test",
        slug: "test",
        filePath: "test.md",
      }
      trie.add(data)
@@ -61,6 +65,7 @@
      const data = {
        title: "Index",
        slug: "index",
        filePath: "index.md",
      }
      trie.add(data)
@@ -72,11 +77,13 @@
      const data1 = {
        title: "Nested",
        slug: "folder/test",
        filePath: "folder/test.md",
      }
      const data2 = {
        title: "Really nested index",
        slug: "a/b/c/index",
        filePath: "a/b/c/index.md",
      }
      trie.add(data1)
@@ -103,8 +110,8 @@
  describe("filter", () => {
    test("should filter nodes based on condition", () => {
      const data1 = { title: "Test1", slug: "test1" }
      const data2 = { title: "Test2", slug: "test2" }
      const data1 = { title: "Test1", slug: "test1", filePath: "test1.md" }
      const data2 = { title: "Test2", slug: "test2", filePath: "test2.md" }
      trie.add(data1)
      trie.add(data2)
@@ -117,8 +124,8 @@
  describe("map", () => {
    test("should apply function to all nodes", () => {
      const data1 = { title: "Test1", slug: "test1" }
      const data2 = { title: "Test2", slug: "test2" }
      const data1 = { title: "Test1", slug: "test1", filePath: "test1.md" }
      const data2 = { title: "Test2", slug: "test2", filePath: "test2.md" }
      trie.add(data1)
      trie.add(data2)
@@ -134,8 +141,12 @@
    })
    test("map over folders should work", () => {
      const data1 = { title: "Test1", slug: "test1" }
      const data2 = { title: "Test2", slug: "a/b/test2" }
      const data1 = { title: "Test1", slug: "test1", filePath: "test1.md" }
      const data2 = {
        title: "Test2",
        slug: "a/b-with-space/test2",
        filePath: "a/b with space/test2.md",
      }
      trie.add(data1)
      trie.add(data2)
@@ -150,15 +161,19 @@
      assert.strictEqual(trie.children[0].displayName, "File: Test1")
      assert.strictEqual(trie.children[1].displayName, "Folder: a")
      assert.strictEqual(trie.children[1].children[0].displayName, "Folder: b")
      assert.strictEqual(trie.children[1].children[0].displayName, "Folder: b with space")
      assert.strictEqual(trie.children[1].children[0].children[0].displayName, "File: Test2")
    })
  })
  describe("entries", () => {
    test("should return all entries", () => {
      const data1 = { title: "Test1", slug: "test1" }
      const data2 = { title: "Test2", slug: "a/b/test2" }
      const data1 = { title: "Test1", slug: "test1", filePath: "test1.md" }
      const data2 = {
        title: "Test2",
        slug: "a/b-with-space/test2",
        filePath: "a/b with space/test2.md",
      }
      trie.add(data1)
      trie.add(data2)
@@ -170,8 +185,8 @@
          ["index", trie.data],
          ["test1", data1],
          ["a/index", null],
          ["a/b/index", null],
          ["a/b/test2", data2],
          ["a/b-with-space/index", null],
          ["a/b-with-space/test2", data2],
        ],
      )
    })
@@ -182,14 +197,17 @@
      const data1 = {
        title: "Root",
        slug: "index",
        filePath: "index.md",
      }
      const data2 = {
        title: "Test",
        slug: "folder/subfolder/test",
        filePath: "folder/subfolder/test.md",
      }
      const data3 = {
        title: "Folder Index",
        slug: "abc/index",
        filePath: "abc/index.md",
      }
      trie.add(data1)
@@ -208,9 +226,9 @@
  describe("sort", () => {
    test("should sort nodes according to sort function", () => {
      const data1 = { title: "A", slug: "a" }
      const data2 = { title: "B", slug: "b" }
      const data3 = { title: "C", slug: "c" }
      const data1 = { title: "A", slug: "a", filePath: "a.md" }
      const data2 = { title: "B", slug: "b", filePath: "b.md" }
      const data3 = { title: "C", slug: "c", filePath: "c.md" }
      trie.add(data3)
      trie.add(data1)