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

tests/server/onDisconnect.ts

@@ -0,0 +1,114 @@
+import test from 'ava'
+import { newHocuspocus, newHocuspocusProvider } from '../utils/index.ts'
+
+test('executes the onDisconnect callback', async t => {
+  await new Promise(async resolve => {
+    const server = await newHocuspocus(t, {
+      async onDisconnect() {
+        t.pass()
+        resolve('done')
+      },
+    })
+
+    const provider = newHocuspocusProvider(t, server, {
+      onConnect() {
+        provider.configuration.websocketProvider.disconnect()
+        provider.disconnect()
+      },
+    })
+  })
+})
+
+test('executes the onDisconnect callback from an extension', async t => {
+  await new Promise(async resolve => {
+    const server = await newHocuspocus(t)
+
+    class CustomExtension {
+      async onDisconnect() {
+        t.pass()
+        resolve('done')
+      }
+    }
+
+    server.configure({
+      extensions: [
+        new CustomExtension(),
+      ],
+    })
+
+    const provider = newHocuspocusProvider(t, server, {
+
+      onConnect() {
+        provider.configuration.websocketProvider.disconnect()
+        provider.disconnect()
+      },
+    })
+  })
+})
+
+test('passes the context to the onLoadDocument callback', async t => {
+  await new Promise(async resolve => {
+    const server = await newHocuspocus(t)
+
+    const mockContext = {
+      user: 123,
+    }
+
+    server.configure({
+      async onConnect() {
+        return mockContext
+      },
+      async onDisconnect({ context }) {
+        t.deepEqual(context, mockContext)
+
+        resolve('done')
+      },
+    })
+
+    const provider = newHocuspocusProvider(t, server, {
+
+      onConnect() {
+        provider.configuration.websocketProvider.disconnect()
+        provider.disconnect()
+      },
+    })
+  })
+})
+
+test('has the server instance', async t => {
+  await new Promise(async resolve => {
+    const server = await newHocuspocus(t, {
+      async onDisconnect({ instance }) {
+        t.is(instance, server)
+
+        resolve('done')
+      },
+    })
+
+    const provider = newHocuspocusProvider(t, server, {
+
+      onConnect() {
+        provider.configuration.websocketProvider.disconnect()
+        provider.disconnect()
+      },
+    })
+  })
+})
+
+test('the connections count is correct', async t => {
+  await new Promise(async resolve => {
+    const server = await newHocuspocus(t, {
+      async onDisconnect() {
+        t.is(server.getConnectionsCount(), 0)
+        resolve('done')
+      },
+    })
+
+    const provider = newHocuspocusProvider(t, server, {
+      onConnect() {
+        provider.configuration.websocketProvider.disconnect()
+        provider.disconnect()
+      },
+    })
+  })
+})