@zofai/zo-sdk 0.2.17 → 0.2.18

package/tests/storkClient.test.ts

@@ -1,11 +1,13 @@
 import { describe, expect, it, vi } from 'vitest'
 
 import {
+  apiBaseUrlToStorkWsUrl,
   buildStorkUpdateDataFromSignedPrice,
   fetchStorkUpdateDataFromKronos,
   getLatestStorkPrices,
   hexToBytes,
   parseStorkUpdateDataRaw,
+  StorkStreamClient,
 } from '../src/storkClient'
 import { appendStorkUpdatesToTransaction, estimateStorkUpdateFeeMist } from '../src/storkOracle'
 import { Transaction } from '@mysten/sui/transactions'
@@ -314,3 +316,89 @@ describe('storkOracle', () => {
     expect(tx.getData().commands.length).toBe(3)
   })
 })
+
+describe('StorkStreamClient', () => {
+  class MockWebSocket {
+    static OPEN = 1
+    readyState = MockWebSocket.OPEN
+    sent: string[] = []
+    private listeners: Record<string, Array<(event: { data?: string }) => void>> = {}
+
+    constructor(public url: string) {
+      queueMicrotask(() => {
+        this.listeners.open?.forEach(cb => cb({}))
+        this.emit({ data: JSON.stringify({ type: 'connected' }) })
+      })
+    }
+
+    addEventListener(event: string, cb: (event: { data?: string }) => void, opts?: { once?: boolean }) {
+      if (!this.listeners[event]) {
+        this.listeners[event] = []
+      }
+      if (opts?.once) {
+        const onceWrapper = (ev: { data?: string }) => {
+          this.listeners[event] = this.listeners[event].filter(h => h !== onceWrapper)
+          cb(ev)
+        }
+        this.listeners[event].push(onceWrapper)
+        return
+      }
+      this.listeners[event].push(cb)
+    }
+
+    send(data: string) {
+      this.sent.push(data)
+    }
+
+    close() {
+      this.readyState = 3
+      this.listeners.close?.forEach(cb => cb({}))
+    }
+
+    emit(event: { data?: string }) {
+      this.listeners.message?.forEach(cb => cb(event))
+    }
+  }
+
+  it('derives evm and stark ws urls from kronos base url', () => {
+    expect(apiBaseUrlToStorkWsUrl('https://api.zofinance.io')).toBe('wss://api.zofinance.io/stork/stream/evm')
+    expect(apiBaseUrlToStorkWsUrl('http://localhost:8080', 'stark')).toBe('ws://localhost:8080/stork/stream/stark')
+  })
+
+  it('subscribes and forwards oracle_prices', async () => {
+    vi.stubGlobal('WebSocket', MockWebSocket)
+
+    const client = new StorkStreamClient('http://localhost:8080', {
+      WebSocketImpl: MockWebSocket as unknown as typeof WebSocket,
+      pingIntervalMs: 0,
+    })
+
+    const updates: string[] = []
+    client.subscribe(['BTCUSD'], (update) => {
+      updates.push(update.prices.BTCUSD?.asset_id ?? '')
+    })
+
+    await client.connect()
+    await new Promise<void>(resolve => setTimeout(resolve, 0))
+
+    expect(JSON.parse((client as any).ws.sent[0]).type).toBe('subscribe')
+    expect(JSON.parse((client as any).ws.sent[0]).data).toEqual(['BTCUSD'])
+
+    ;(client as any).ws.emit({
+      data: JSON.stringify({
+        type: 'oracle_prices',
+        trace_id: 'trace-1',
+        data: {
+          BTCUSD: {
+            asset_id: 'BTCUSD',
+            price: '100000000000000000000',
+            stork_signed_price: { price: '100000000000000000000' },
+          },
+        },
+      }),
+    })
+
+    expect(updates).toEqual(['BTCUSD'])
+    client.disconnect()
+  })
+})