upstream-hocuspocus/tests/server/onDisconnect.ts

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()
      },
    })
  })
})