fix: re-run the once function on every key change

src/ls.test.ts

@@ -10,7 +10,6 @@ describe("ls", () => {
   })
 
   afterEach(() => {
-    vi.restoreAllMocks()
     localStorage.clear()
     storage = null
   })

src/once.test.ts

@@ -12,16 +12,6 @@ describe("once", () => {
     mockQuery = vi.fn()
   })
 
-  afterEach(() => {
-    vi.restoreAllMocks()
-  })
-
-  it("should call the query function once", async () => {
-    once("123", mockQuery)
-    await delay()
-    expect(mockQuery).toHaveBeenCalledTimes(1)
-  })
-
   it("should only call the query function once with the same ID", async () => {
     Promise.all([
       once("123", mockQuery),
@@ -41,4 +31,16 @@ describe("once", () => {
     await delay()
     expect(mockQuery).toHaveBeenCalledTimes(3)
   })
+
+  it("should call the query function again if the key changes and then changes again", async () => {
+    once("123", mockQuery)
+    await delay()
+    expect(mockQuery).toHaveBeenCalledTimes(1)
+    once("234", mockQuery)
+    await delay()
+    expect(mockQuery).toHaveBeenCalledTimes(2)
+    once("123", mockQuery)
+    await delay()
+    expect(mockQuery).toHaveBeenCalledTimes(3)
+  })
 })

src/once.ts

@@ -1,8 +1,8 @@
 export const makeOnce = () => {
-  const calls = new Set()
-  return (id: string | number, query: () => void | Promise<void>) => {
-    if (!calls.has(id)) {
-      calls.add(id)
+  let latestCalledKey: string | number | undefined
+  return (key: string | number, query: () => void | Promise<void>) => {
+    if (latestCalledKey !== key) {
+      latestCalledKey = key
       setTimeout(query)
     }
   }

src/purity.test.ts

@@ -17,7 +17,6 @@ describe("purity", () => {
     // eslint-disable-next-line @typescript-eslint/ban-ts-comment
     // @ts-ignore
     app = undefined
-    vi.restoreAllMocks()
   })
 
   describe("counter app", () => {

src/purity.ts

@@ -116,6 +116,11 @@ export const init = <State extends Record<string, unknown>>(
    * Forces html re-rendering with the current state
    */
   function rerender() {
+    if (!rootComponent) {
+      throw new Error(
+        "Cannot rerender: app has not been mounted. Call mount() first."
+      )
+    }
     const newNodesMap = rootComponent()
     console.warn("🌀")
     for (const [id, domNode] of domNodesMap) {

src/router.test.ts

@@ -12,7 +12,6 @@ describe("router", () => {
   })
   afterEach(() => {
     document.body.innerHTML = ""
-    vi.restoreAllMocks()
   })
   it("switches between different views depending on the url state", async () => {
     const App = (): string => render`

test-utils.ts

@@ -43,9 +43,6 @@ export const setBeforeAndAfter = ({
   beforeEach(() => {
     console.warn = vi.fn()
   })
-  afterEach(() => {
-    vi.restoreAllMocks()
-  })
   afterAll(() => {
     document.body.innerHTML = ""
   })

vitest.config.ts

@@ -3,6 +3,7 @@ import {defineConfig} from "vitest/config"
 export default defineConfig({
   test: {
     environment: "jsdom",
+    restoreMocks: true,
     coverage: {
       reporter: ["text", "json-summary", "json"],
       extension: [".ts"],