1 files changed, 46 insertions, 0 deletions
diff --git a/test/websocket/ping-pong.js b/test/websocket/ping-pong.js
new file mode 100644
index 0000000..b7c4694
--- /dev/null
+++ b/test/websocket/ping-pong.js
@@ -0,0 +1,46 @@
+'use strict'
+
+const { test } = require('tap')
+const { WebSocketServer } = require('ws')
+const diagnosticsChannel = require('diagnostics_channel')
+const { WebSocket } = require('../..')
+
+test('Receives ping and parses body', (t) => {
+  t.plan(1)
+
+  const server = new WebSocketServer({ port: 0 })
+
+  server.on('connection', (ws) => {
+    ws.ping('Hello, world')
+  })
+
+  t.teardown(server.close.bind(server))
+
+  const ws = new WebSocket(`ws://localhost:${server.address().port}`)
+  ws.onerror = ws.onmessage = t.fail
+
+  diagnosticsChannel.channel('undici:websocket:ping').subscribe(({ payload }) => {
+    t.same(payload, Buffer.from('Hello, world'))
+    ws.close()
+  })
+})
+
+test('Receives pong and parses body', (t) => {
+  t.plan(1)
+
+  const server = new WebSocketServer({ port: 0 })
+
+  server.on('connection', (ws) => {
+    ws.pong('Pong')
+  })
+
+  t.teardown(server.close.bind(server))
+
+  const ws = new WebSocket(`ws://localhost:${server.address().port}`)
+  ws.onerror = ws.onmessage = t.fail
+
+  diagnosticsChannel.channel('undici:websocket:pong').subscribe(({ payload }) => {
+    t.same(payload, Buffer.from('Pong'))
+    ws.close()
+  })
+})