@zofai/zo-sdk 0.2.14 → 0.2.15

package/tests/storkClient.test.ts

@@ -0,0 +1,316 @@
+import { describe, expect, it, vi } from 'vitest'
+
+import {
+  buildStorkUpdateDataFromSignedPrice,
+  fetchStorkUpdateDataFromKronos,
+  getLatestStorkPrices,
+  hexToBytes,
+  parseStorkUpdateDataRaw,
+} from '../src/storkClient'
+import { appendStorkUpdatesToTransaction, estimateStorkUpdateFeeMist } from '../src/storkOracle'
+import { Transaction } from '@mysten/sui/transactions'
+
+const SAMPLE_ID = `0x${'ab'.repeat(32)}`
+
+describe('storkClient', () => {
+  it('parses kronos update_data payload', () => {
+    const update = parseStorkUpdateDataRaw({
+      id: SAMPLE_ID,
+      temporal_numeric_value_timestamp_ns: 123,
+      temporal_numeric_value_magnitude: '1000000000000000000',
+      temporal_numeric_value_negative: false,
+      publisher_merkle_root: '0x01',
+      value_compute_alg_hash: '0x02',
+      r: '0x03',
+      s: '0x04',
+      v: 27,
+    })
+    expect(update.id.length).toBe(32)
+    expect(update.temporalNumericValueMagnitude).toBe(10n ** 18n)
+    expect(update.v).toBe(27)
+  })
+
+  it('getLatestStorkPrices calls kronos REST', async () => {
+    vi.stubGlobal('fetch', vi.fn().mockResolvedValue({
+      ok: true,
+      json: async () => ({
+        success: true,
+        data: {
+          prices: {
+            BTCUSD: {
+              asset_id: 'BTCUSD',
+              price: '100000000000000000000',
+              update_data: {
+                id: SAMPLE_ID,
+                temporal_numeric_value_timestamp_ns: 1,
+                temporal_numeric_value_magnitude: '1',
+                temporal_numeric_value_negative: false,
+                publisher_merkle_root: '0x01',
+                value_compute_alg_hash: '0x02',
+                r: '0x03',
+                s: '0x04',
+                v: 27,
+              },
+            },
+          },
+        },
+      }),
+    }))
+
+    const data = await getLatestStorkPrices('http://localhost:8080', { assets: ['BTCUSD'] })
+    expect(data.prices.BTCUSD.asset_id).toBe('BTCUSD')
+  })
+
+  it('fetchStorkUpdateDataFromKronos returns parsed updates', async () => {
+    vi.stubGlobal('fetch', vi.fn().mockResolvedValue({
+      ok: true,
+      json: async () => ({
+        success: true,
+        data: {
+          prices: {
+            BTCUSD: {
+              asset_id: 'BTCUSD',
+              price: '1',
+              update_data: {
+                id: SAMPLE_ID,
+                temporal_numeric_value_timestamp_ns: 1,
+                temporal_numeric_value_magnitude: '1',
+                temporal_numeric_value_negative: false,
+                publisher_merkle_root: '0x01',
+                value_compute_alg_hash: '0x02',
+                r: '0x03',
+                s: '0x04',
+                v: 27,
+              },
+            },
+          },
+        },
+      }),
+    }))
+
+    const updates = await fetchStorkUpdateDataFromKronos('http://localhost:8080', ['BTCUSD'])
+    expect(updates).toHaveLength(1)
+  })
+
+  it('accepts string timestamp from stork_signed_price (no float64 truncation)', () => {
+    const entry = {
+      asset_id: 'BTCUSD',
+      price: '64319215652687504000000',
+      stork_signed_price: {
+        encoded_asset_id: '0x7404e3d104ea7841c3d9e6fd20adfe99b4ad586bc08d8f3bd3afef894cf184de',
+        price: '64319215652687504000000',
+        publisher_merkle_root: '0xcbe0ea2a350df5e5a386587ecf1764ada3e744b67564f0146e12f25bd8b4aebd',
+        calculation_alg: {
+          checksum: '9be7e9f9ed459417d96112a7467bd0b27575a2c7847195c68f805b70ce1795ba',
+        },
+        timestamped_signature: {
+          // String preserves exact u64 — the correct on-chain value
+          timestamp: '1782014930584673519',
+          signature: {
+            r: '0x668ea843c69fe1eb8dd1d0c8275b3acac2eea7fe9947b2dbac2aceb288195359',
+            s: '0x3c3d2236f2b175e26d5191185eac31b6a95ce99bad6cd1bdee1586cef40b3766',
+            v: '0x1b',
+          },
+        },
+      },
+    }
+
+    const update = buildStorkUpdateDataFromSignedPrice(entry)
+    expect(update?.temporalNumericValueTimestampNs).toBe(1782014930584673519n)
+    expect(update?.temporalNumericValueMagnitude).toBe(64319215652687504000000n)
+    expect(update?.v).toBe(27)
+  })
+
+  it('falls back to update_data string timestamp when stork_signed_price timestamp is a number', () => {
+    // Kronos returns stork_signed_price.timestamp as a JSON number (float64 truncated)
+    // and update_data.temporal_numeric_value_timestamp_ns as a string (exact u64).
+    // The SDK must use the string from update_data to avoid EInvalidSignature on-chain.
+    const truncatedNumberTs = 1782148271583801600 // float64 of real ts 1782148271583801715
+    const exactStringTs = '1782148271583801715'
+
+    const entry = {
+      asset_id: 'BTCUSD',
+      price: '64594662603625009000000',
+      update_data: {
+        id: '0x7404e3d104ea7841c3d9e6fd20adfe99b4ad586bc08d8f3bd3afef894cf184de',
+        temporal_numeric_value_timestamp_ns: exactStringTs,
+        temporal_numeric_value_magnitude: '64594662603625009000000',
+        temporal_numeric_value_negative: false,
+        publisher_merkle_root: '0xd3bbe3d69002f7db6e14da284506c5071539bd47ce4134fa48fc805a2c40f29e',
+        value_compute_alg_hash: '9be7e9f9ed459417d96112a7467bd0b27575a2c7847195c68f805b70ce1795ba',
+        r: '0x5e2a8d7c1a62842924a7abda81c65024544a65f3447d77dd1686d150bc7527a0',
+        s: '0x0bf5d68c5705335ab40ba63f31d03b54547376f7bb3ec6551a91dc03e39d7b7b',
+        v: 27,
+      },
+      stork_signed_price: {
+        public_key: '0x0a803F9b1CCe32e2773e0d2e98b37E0775cA5d44',
+        encoded_asset_id: '0x7404e3d104ea7841c3d9e6fd20adfe99b4ad586bc08d8f3bd3afef894cf184de',
+        price: '64594662603625009000000',
+        publisher_merkle_root: '0xd3bbe3d69002f7db6e14da284506c5071539bd47ce4134fa48fc805a2c40f29e',
+        calculation_alg: {
+          checksum: '9be7e9f9ed459417d96112a7467bd0b27575a2c7847195c68f805b70ce1795ba',
+        },
+        timestamped_signature: {
+          timestamp: truncatedNumberTs, // float64-truncated — wrong for on-chain use
+          msg_hash: '0x57b69c9ea90ee0492d4ab0149c817b16d95645121837bf031cb0cec3a8ba7d83',
+          signature: {
+            r: '0x5e2a8d7c1a62842924a7abda81c65024544a65f3447d77dd1686d150bc7527a0',
+            s: '0x0bf5d68c5705335ab40ba63f31d03b54547376f7bb3ec6551a91dc03e39d7b7b',
+            v: '0x1b',
+          },
+        },
+      },
+    }
+
+    const update = buildStorkUpdateDataFromSignedPrice(entry)
+    // Must use the exact string timestamp, not the float64-truncated number
+    expect(update?.temporalNumericValueTimestampNs).toBe(1782148271583801715n)
+    expect(update?.temporalNumericValueTimestampNs).not.toBe(BigInt(truncatedNumberTs))
+  })
+
+  it('uses stork_signed_price over update_data', () => {
+    const update = buildStorkUpdateDataFromSignedPrice({
+      asset_id: 'BTCUSD',
+      price: '999',
+      update_data: {
+        id: SAMPLE_ID,
+        temporal_numeric_value_timestamp_ns: 1,
+        temporal_numeric_value_magnitude: '1',
+        temporal_numeric_value_negative: false,
+        publisher_merkle_root: '0x01',
+        value_compute_alg_hash: '0x02',
+        r: '0x03',
+        s: '0x04',
+        v: 27,
+      },
+      stork_signed_price: {
+        encoded_asset_id: '0x7404e3d104ea7841c3d9e6fd20adfe99b4ad586bc08d8f3bd3afef894cf184de',
+        price: '64340643513099996000000',
+        publisher_merkle_root: '0xce95788915171eab496e1f6318b4f94065ca94d49b121627deaa5087e433fd8b',
+        calculation_alg: {
+          checksum: '9be7e9f9ed459417d96112a7467bd0b27575a2c7847195c68f805b70ce1795ba',
+        },
+        timestamped_signature: {
+          timestamp: '1782012523783284700',
+          signature: {
+            r: '0x6abb61d61751c5f533833260bcf15f076498ef0fabdfdd9ce73ff4bf55f1e8a6',
+            s: '0x6a1747bc57e098950c683457e0e8e20707b71be704a648eca16477f82ad29151',
+            v: '0x1b',
+          },
+        },
+      },
+    })
+
+    expect(update?.temporalNumericValueTimestampNs).toBe(1782012523783284700n)
+    expect(update?.temporalNumericValueMagnitude).toBe(64340643513099996000000n)
+  })
+
+  it('buildStorkUpdateDataFromSignedPrice matches Go ToUpdateData conversion', () => {
+    const update = buildStorkUpdateDataFromSignedPrice({
+      asset_id: 'BTCUSD',
+      price: '64340643513099996000000',
+      stork_signed_price: {
+        public_key: '0x0a803F9b1CCe32e2773e0d2e98b37E0775cA5d44',
+        encoded_asset_id: '0x7404e3d104ea7841c3d9e6fd20adfe99b4ad586bc08d8f3bd3afef894cf184de',
+        price: '64340643513099996000000',
+        timestamped_signature: {
+          signature: {
+            r: '0x6abb61d61751c5f533833260bcf15f076498ef0fabdfdd9ce73ff4bf55f1e8a6',
+            s: '0x6a1747bc57e098950c683457e0e8e20707b71be704a648eca16477f82ad29151',
+            v: '0x1b',
+          },
+          timestamp: '1782012523783284700',
+          msg_hash: '0x88e8f9e4d5036ef248b2c1a7fae0b00c39095b2bab955bb1149332b4d90c821d',
+        },
+        publisher_merkle_root: '0xce95788915171eab496e1f6318b4f94065ca94d49b121627deaa5087e433fd8b',
+        calculation_alg: {
+          type: 'median',
+          version: 'v1',
+          checksum: '9be7e9f9ed459417d96112a7467bd0b27575a2c7847195c68f805b70ce1795ba',
+        },
+      },
+    })
+
+    expect(update?.id).toEqual(hexToBytes('0x7404e3d104ea7841c3d9e6fd20adfe99b4ad586bc08d8f3bd3afef894cf184de'))
+    expect(update?.id.length).toBe(32)
+    expect(update?.temporalNumericValueTimestampNs).toBe(1782012523783284700n)
+    expect(update?.temporalNumericValueMagnitude).toBe(64340643513099996000000n)
+    expect(update?.temporalNumericValueNegative).toBe(false)
+    expect(update?.valueComputeAlgHash).toEqual(hexToBytes('9be7e9f9ed459417d96112a7467bd0b27575a2c7847195c68f805b70ce1795ba'))
+    expect(update?.v).toBe(27)
+  })
+
+  it('buildStorkUpdateDataFromSignedPrice uses feederId when encoded_asset_id absent', () => {
+    const feederId = hexToBytes(`0x${'74'.repeat(32)}`)
+
+    const update = buildStorkUpdateDataFromSignedPrice(
+      {
+        asset_id: 'BTCUSD',
+        price: '1000000000000000000',
+        stork_signed_price: {
+          price: '1000000000000000000',
+          publisher_merkle_root: '0x0101',
+          calculation_alg: { checksum: '0x0202' },
+          timestamped_signature: {
+            timestamp: 999,
+            signature: { r: '0x0303', s: '0x0404', v: '27' },
+          },
+        },
+      },
+      feederId,
+    )
+
+    expect(update?.id).toEqual(feederId)
+    expect(update?.id.length).toBe(32)
+  })
+
+  it('buildStorkUpdateDataFromSignedPrice decodes hex encoded_asset_id', () => {
+    const update = buildStorkUpdateDataFromSignedPrice({
+      asset_id: 'BTCUSD',
+      price: '1000000000000000000',
+      stork_signed_price: {
+        encoded_asset_id: `0x${'07'.repeat(32)}`,
+        price: '1000000000000000000',
+        publisher_merkle_root: '0x0101',
+        calculation_alg: { checksum: '0x0202' },
+        timestamped_signature: {
+          timestamp: 999,
+          signature: { r: '0x0303', s: '0x0404', v: '27' },
+        },
+      },
+    })
+
+    expect(update?.id.length).toBe(32)
+    expect(update?.temporalNumericValueTimestampNs).toBe(999n)
+  })
+})
+
+describe('storkOracle', () => {
+  it('estimateStorkUpdateFeeMist scales per update', () => {
+    expect(estimateStorkUpdateFeeMist(2)).toBe(2n)
+  })
+
+  it('appendStorkUpdatesToTransaction adds move calls', () => {
+    const tx = new Transaction()
+    appendStorkUpdatesToTransaction(
+      tx,
+      {
+        storkPackage: '0x664fcc7d42abf7583938e6e64f51aa1ede063a49f445a6a9c0206638af0a53d9',
+        storkState: '0x88ca8f0ce0f6b30a39bdf71be27ffd0c5571adba28c49a1a3bd530e8fcffb29a',
+      },
+      [{
+        id: hexToBytes(SAMPLE_ID),
+        temporalNumericValueTimestampNs: 1n,
+        temporalNumericValueMagnitude: 1n,
+        temporalNumericValueNegative: false,
+        publisherMerkleRoot: hexToBytes('0x01'),
+        valueComputeAlgHash: hexToBytes('0x02'),
+        r: hexToBytes('0x03'),
+        s: hexToBytes('0x04'),
+        v: 27,
+      }],
+    )
+
+    expect(tx.getData().commands.length).toBe(3)
+  })
+})