test: test gsoc and pss onClose handlers (#1083)

* test: test gsoc and pss onClose handlers

* test: update test coverage

* test: update test coverage

---------

Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>

Author: Cafe137

test/coverage/coverage-summary.json

@@ -1,6 +1,6 @@
-{"total": {"lines":{"total":2455,"covered":2008,"skipped":0,"pct":81.79},"statements":{"total":2498,"covered":2047,"skipped":0,"pct":81.94},"functions":{"total":595,"covered":472,"skipped":0,"pct":79.32},"branches":{"total":558,"covered":346,"skipped":0,"pct":62},"branchesTrue":{"total":0,"covered":0,"skipped":0,"pct":100}}
+{"total": {"lines":{"total":2455,"covered":2009,"skipped":0,"pct":81.83},"statements":{"total":2498,"covered":2048,"skipped":0,"pct":81.98},"functions":{"total":595,"covered":473,"skipped":0,"pct":79.49},"branches":{"total":558,"covered":346,"skipped":0,"pct":62},"branchesTrue":{"total":0,"covered":0,"skipped":0,"pct":100}}
 ,"/home/runner/work/bee-js/bee-js/src/bee-dev.ts": {"lines":{"total":14,"covered":5,"skipped":0,"pct":35.71},"functions":{"total":2,"covered":0,"skipped":0,"pct":0},"statements":{"total":14,"covered":5,"skipped":0,"pct":35.71},"branches":{"total":0,"covered":0,"skipped":0,"pct":100}}
-,"/home/runner/work/bee-js/bee-js/src/bee.ts": {"lines":{"total":445,"covered":369,"skipped":0,"pct":82.92},"functions":{"total":127,"covered":103,"skipped":0,"pct":81.1},"statements":{"total":448,"covered":372,"skipped":0,"pct":83.03},"branches":{"total":115,"covered":67,"skipped":0,"pct":58.26}}
+,"/home/runner/work/bee-js/bee-js/src/bee.ts": {"lines":{"total":445,"covered":370,"skipped":0,"pct":83.14},"functions":{"total":127,"covered":104,"skipped":0,"pct":81.88},"statements":{"total":448,"covered":373,"skipped":0,"pct":83.25},"branches":{"total":115,"covered":67,"skipped":0,"pct":58.26}}
 ,"/home/runner/work/bee-js/bee-js/src/index.ts": {"lines":{"total":16,"covered":16,"skipped":0,"pct":100},"functions":{"total":10,"covered":7,"skipped":0,"pct":70},"statements":{"total":25,"covered":25,"skipped":0,"pct":100},"branches":{"total":0,"covered":0,"skipped":0,"pct":100}}
 ,"/home/runner/work/bee-js/bee-js/src/chunk/bmt.ts": {"lines":{"total":16,"covered":15,"skipped":0,"pct":93.75},"functions":{"total":3,"covered":3,"skipped":0,"pct":100},"statements":{"total":17,"covered":16,"skipped":0,"pct":94.11},"branches":{"total":1,"covered":0,"skipped":0,"pct":0}}
 ,"/home/runner/work/bee-js/bee-js/src/chunk/cac.ts": {"lines":{"total":21,"covered":19,"skipped":0,"pct":90.47},"functions":{"total":2,"covered":2,"skipped":0,"pct":100},"statements":{"total":21,"covered":19,"skipped":0,"pct":90.47},"branches":{"total":7,"covered":5,"skipped":0,"pct":71.42}}

test/integration/on-close.spec.ts

@@ -0,0 +1,108 @@
+import { Dates, System } from 'cafe-utility'
+import { createServer } from 'http'
+import { WebSocketServer } from 'ws'
+import { Bee, NULL_IDENTIFIER, NULL_TOPIC } from '../../src'
+
+interface HandledEvents {
+  message: any[]
+  error: any[]
+  close: any[]
+}
+
+test('pss onClose handler', async () => {
+  const events: HandledEvents = {
+    message: [],
+    error: [],
+    close: [],
+  }
+  const bee = new Bee('http://localhost:8081')
+
+  const httpServer = createServer()
+  const socketServer = new WebSocketServer({ noServer: true })
+
+  httpServer.on('upgrade', (request, socket, head) => {
+    socketServer.handleUpgrade(request, socket, head, ws => {
+      socketServer.emit('connection', ws, request)
+    })
+  })
+
+  httpServer.listen(8081)
+
+  bee.pssSubscribe(NULL_TOPIC, {
+    onMessage: event => {
+      events.message.push(event)
+    },
+    onError: error => {
+      events.error.push(error)
+    },
+    onClose: () => {
+      events.close.push('closed')
+    },
+  })
+
+  await System.waitFor(async () => socketServer.clients.size === 1, {
+    attempts: 10,
+    waitMillis: Dates.seconds(1),
+  })
+
+  socketServer.clients.forEach(client => client.close())
+  socketServer.close()
+  httpServer.close()
+
+  await System.waitFor(async () => events.close.length > 0, {
+    attempts: 60,
+    waitMillis: Dates.seconds(1),
+  })
+
+  expect(events.message.length).toBe(0)
+  expect(events.error.length).toBe(0)
+})
+
+test('gsoc onClose handler', async () => {
+  const events: HandledEvents = {
+    message: [],
+    error: [],
+    close: [],
+  }
+  const bee = new Bee('http://localhost:8082')
+
+  const httpServer = createServer()
+  const socketServer = new WebSocketServer({ noServer: true })
+
+  httpServer.on('upgrade', (request, socket, head) => {
+    socketServer.handleUpgrade(request, socket, head, ws => {
+      socketServer.emit('connection', ws, request)
+    })
+  })
+
+  httpServer.listen(8082)
+
+  bee.gsocSubscribe('0x0123401234012340123401234012340123401234', NULL_IDENTIFIER, {
+    onMessage: event => {
+      events.message.push(event)
+    },
+    onError: error => {
+      events.error.push(error)
+    },
+    onClose: () => {
+      events.close.push('closed')
+    },
+  })
+
+  await System.waitFor(async () => socketServer.clients.size === 1, {
+    attempts: 10,
+    waitMillis: Dates.seconds(1),
+  })
+
+  socketServer.clients.forEach(client => client.close())
+  socketServer.close()
+  httpServer.close()
+
+  await System.waitFor(async () => events.close.length > 0, {
+    attempts: 60,
+    waitMillis: Dates.seconds(1),
+  })
+
+  expect(events.message.length).toBe(0)
+  expect(events.error.length).toBe(0)
+})