xahau-patch

packages/xahau/test/setupClient.ts

@@ -0,0 +1,78 @@
+import { Client } from '../src/client'
+
+import createMockRippled, {
+  type MockedWebSocketServer,
+} from './createMockRippled'
+import xahaud from './fixtures/xahaud'
+import { destroyServer, getFreePort } from './testUtils'
+
+export interface XrplTestContext {
+  client: Client
+  _mockedServerPort?: number
+  mockRippled?: MockedWebSocketServer
+  mocks?: MockedWebSocketServer[]
+  servers?: number[]
+}
+
+async function setupMockRippledConnection(
+  port: number,
+): Promise<XrplTestContext> {
+  const context: XrplTestContext = {
+    mockRippled: createMockRippled(port),
+    _mockedServerPort: port,
+    client: new Client(`ws://localhost:${port}`),
+    servers: [port],
+  }
+
+  context.client.on('error', () => {
+    // We must have an error listener attached for reconnect errors
+  })
+  context.mockRippled?.addResponse(
+    'server_info',
+    xahaud.server_info.withNetworkId,
+  )
+
+  return context.client.connect().then(() => context)
+}
+
+async function setupClient(): Promise<XrplTestContext> {
+  return getFreePort().then(async (port) => {
+    return setupMockRippledConnection(port)
+  })
+}
+
+async function teardownClient(
+  incomingContext: XrplTestContext,
+  done?: () => void,
+): Promise<void> {
+  return incomingContext.client
+    .disconnect()
+    .then(async () => {
+      return new Promise<void>((resolve) => {
+        // eslint-disable-next-line no-negated-condition -- Easier to read with negation
+        if (incomingContext.mockRippled != null) {
+          incomingContext.mockRippled.close(() => {
+            resolve()
+          })
+        } else {
+          resolve()
+        }
+      })
+    })
+    .then(async () => {
+      await Promise.all(
+        incomingContext.servers?.map(async (port) => destroyServer(port)) ?? [],
+      )
+    })
+    .catch((err) => {
+      // eslint-disable-next-line no-console -- console.error is fine in tests
+      console.error(err)
+      if (done) {
+        done()
+      } else {
+        throw err
+      }
+    })
+}
+
+export { setupClient, teardownClient, createMockRippled }