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

tests/server/onClose.ts

@@ -0,0 +1,70 @@
+import test from 'ava'
+import { WebSocketStatus } from '@hocuspocus/provider'
+import { newHocuspocus, newHocuspocusProvider, newHocuspocusProviderWebsocket } from '../utils/index.ts'
+import { retryableAssertion } from '../utils/retryableAssertion.ts'
+
+test('server closes connection when receiving close event from provider', async t => {
+  await new Promise(async resolve => {
+    const server = await newHocuspocus(t, {})
+    const socket = newHocuspocusProviderWebsocket(t, server, {})
+
+    const provider1 = newHocuspocusProvider(t, server, {
+      websocketProvider: socket,
+      name: 'hocuspocus-test',
+    })
+
+    await retryableAssertion(t, t2 => {
+      t2.is(server.getConnectionsCount(), 1)
+    })
+
+    await retryableAssertion(t, t2 => {
+      provider1.destroy()
+      t2.is(server.getConnectionsCount(), 0)
+    })
+
+    resolve('ok')
+  })
+})
+
+test('server doesnt close connection after receiving close event from all connections', async t => {
+  await new Promise(async resolve => {
+    const server = await newHocuspocus(t, {})
+    const socket = newHocuspocusProviderWebsocket(t, server, {})
+
+    const provider1 = newHocuspocusProvider(t, server, {
+      websocketProvider: socket,
+      name: 'hocuspocus-test',
+    })
+
+    const provider2 = newHocuspocusProvider(t, server, {
+      websocketProvider: socket,
+      name: 'hocuspocus-test2',
+    })
+
+    await retryableAssertion(t, t2 => {
+      t2.is(server.getConnectionsCount(), 1)
+    })
+
+    socket.shouldConnect = false
+    provider1.destroy()
+
+    t.is(provider1.configuration.websocketProvider.status, WebSocketStatus.Connected)
+    t.is(provider2.configuration.websocketProvider.status, WebSocketStatus.Connected)
+
+    setTimeout(async () => {
+      t.is(server.getConnectionsCount(), 1)
+      provider2.destroy()
+
+      t.is(provider1.configuration.websocketProvider.status, WebSocketStatus.Connected)
+      t.is(provider2.configuration.websocketProvider.status, WebSocketStatus.Connected)
+
+      await retryableAssertion(t, t2 => {
+        t2.is(server.getConnectionsCount(), 1)
+        t2.is(provider1.configuration.websocketProvider.status, WebSocketStatus.Connected)
+        t2.is(provider2.configuration.websocketProvider.status, WebSocketStatus.Connected)
+      })
+
+      resolve('ok')
+    }, 200)
+  })
+})