snapshot of https://github.com/ueberdosis/hocuspocus.git @ cf3cfa355b4d3bb61c6b6c80071f71783e32470e (2026-09-15)

tests/server/afterUnloadDocument.ts

@@ -0,0 +1,77 @@
+import test from 'ava'
+import type { HocuspocusProvider } from '@hocuspocus/provider'
+
+import { newHocuspocus, newHocuspocusProvider } from '../utils/index.ts'
+
+test('executes the afterUnloadDocument callback', async t => {
+  await new Promise(async resolve => {
+    const server = await newHocuspocus(t, {
+      async afterUnloadDocument() {
+        t.pass()
+        resolve('done')
+      },
+    })
+
+    const p = newHocuspocusProvider(t, server, {
+      onSynced(data) {
+        p.configuration.websocketProvider.disconnect()
+        p.disconnect()
+      },
+    })
+  })
+})
+
+test('executes the afterUnloadDocument callback when all clients disconnect after a document was loaded', async t => {
+  await new Promise(async resolve => {
+    // eslint-disable-next-line prefer-const
+    let provider: HocuspocusProvider
+
+    class CustomExtension {
+      async afterLoadDocument() {
+        provider.configuration.websocketProvider.disconnect()
+        provider.disconnect()
+      }
+
+      async afterUnloadDocument() {
+        t.pass()
+        resolve('done')
+      }
+    }
+
+    const server = await newHocuspocus(t, {
+      extensions: [new CustomExtension()],
+    })
+
+    provider = newHocuspocusProvider(t, server)
+  })
+})
+
+test('does not execute the afterUnloadDocument callback when document fails to load', async t => {
+  await new Promise(async (resolve, reject) => {
+    const server = await newHocuspocus(t)
+
+    class CustomExtension {
+      async onLoadDocument() {
+        throw new Error('oops!')
+      }
+
+      async afterUnloadDocument() {
+        t.fail()
+        reject('should not be called')
+      }
+    }
+
+    server.configure({
+      extensions: [
+        new CustomExtension(),
+      ],
+    })
+
+    newHocuspocusProvider(t, server)
+
+    setTimeout(() => {
+      t.pass()
+      resolve('done')
+    }, 500)
+  })
+})