feat: add grandpa justifications mock rpc

packages/core/src/blockchain/head-state.ts

@@ -45,7 +45,7 @@ export class HeadState {
     delete this.#storageListeners[id]
   }
 
-  async subscrubeRuntimeVersion(cb: (block: Block) => void) {
+  async subscribeRuntimeVersion(cb: (block: Block) => void) {
     const id = randomId()
     const codeKey = stringToHex(':code')
     this.#storageListeners[id] = [[codeKey], cb]

packages/core/src/rpc/substrate/grandpa.ts

@@ -0,0 +1,25 @@
+import { Handler } from '../shared.js'
+
+export const grandpa_subscribeJustifications: Handler<void, string> = async (context, _params, { subscribe }) => {
+  let update = () => {}
+
+  const id = context.chain.headState.subscribeHead(() => update())
+  const callback = subscribe('grandpa_justifications', id, () => context.chain.headState.unsubscribeHead(id))
+
+  update = async () => {
+    const mockProof = '0x' + '7'.repeat(64)
+    callback(mockProof)
+  }
+
+  setTimeout(update, 50)
+
+  return id
+}
+
+export const grandpa_unsubscribeJustifications: Handler<[string], void> = async (
+  _context,
+  [subid],
+  { unsubscribe },
+) => {
+  unsubscribe(subid)
+}

packages/core/src/rpc/substrate/index.ts

@@ -1,18 +1,21 @@
 import * as AuthorRPC from './author.js'
 import * as ChainRPC from './chain.js'
+import * as GrandpaRPC from './grandpa.js'
 import * as PaymentRPC from './payment.js'
 import * as StateRPC from './state.js'
 import * as SystemRPC from './system.js'
 
 export { AuthorRPC }
 export { ChainRPC }
+export { GrandpaRPC }
 export { PaymentRPC }
 export { StateRPC }
 export { SystemRPC }
 
 const handlers = {
   ...AuthorRPC,
   ...ChainRPC,
+  ...GrandpaRPC,
   ...PaymentRPC,
   ...StateRPC,
   ...SystemRPC,

packages/core/src/rpc/substrate/state.ts

@@ -56,6 +56,24 @@ export const state_getKeysPaged: Handler<[string, number, string, HexString], st
   return block?.getKeysPaged({ prefix, pageSize, startKey })
 }
 
+/**
+ * Mock get proof of the runtime storage value.
+ *
+ * @param context
+ * @param params - [`keys`, `blockhash`]
+ *
+ * @return mock proof
+ */
+export const state_getReadProof: Handler<
+  [HexString[], HexString],
+  { at: HexString; proof: HexString[] } | null
+> = async (_context, [keys, hash]) => {
+  return {
+    at: hash,
+    proof: keys.map(() => ('0x' + '7'.repeat(64)) as HexString),
+  }
+}
+
 /**
  * @param context
  * @param params - [`keys`, `blockhash`]
@@ -110,7 +128,7 @@ export const state_call: Handler<[HexString, HexString, HexString], HexString> =
  */
 export const state_subscribeRuntimeVersion: Handler<[], string> = async (context, _params, { subscribe }) => {
   let update = (_block: Block) => {}
-  const id = await context.chain.headState.subscrubeRuntimeVersion((block) => update(block))
+  const id = await context.chain.headState.subscribeRuntimeVersion((block) => update(block))
   const callback = subscribe('state_runtimeVersion', id)
   update = async (block) => callback(await block.runtimeVersion)
   setTimeout(() => {

packages/e2e/src/__snapshots__/grandpa.test.ts.snap

@@ -0,0 +1,11 @@
+// Vitest Snapshot v1, https://vitest.dev/guide/snapshot.html
+
+exports[`grandpa rpc > subscribeJustifications 1`] = `
+[
+  [
+    "0x7777777777777777777777777777777777777777777777777777777777777777",
+  ],
+]
+`;
+
+exports[`grandpa rpc > subscribeJustifications 2`] = `"0x7ba5ba52f5b88fa5bcab5c1a0f7e158d845d93e1c70e9a5b42a7c3a9af681ee7"`;

packages/e2e/src/grandpa.test.ts

@@ -0,0 +1,36 @@
+import { describe, expect, it } from 'vitest'
+
+import { api, delay, dev, mockCallback, setupApi } from './helper.js'
+
+setupApi({
+  endpoint: ['wss://rpc.ibp.network/polkadot'],
+  blockHash: '0xb012d04c56b65cfa1f47cb1f884d920f95d0097b1ed42f5da18d5e2a436c2f3e',
+})
+
+describe('grandpa rpc', () => {
+  it('subscribeJustifications', async () => {
+    const { callback, next } = mockCallback()
+    const unsub = await api.rpc.grandpa.subscribeJustifications(callback)
+
+    await next()
+    expect(callback.mock.calls).toMatchSnapshot()
+
+    callback.mockClear()
+
+    expect(await dev.newBlock()).toMatchSnapshot()
+
+    await next()
+
+    expect(callback.mock.calls).toMatchSnapshot()
+
+    callback.mockClear()
+
+    unsub()
+
+    expect(await dev.newBlock()).toMatchSnapshot()
+
+    await delay(100)
+
+    expect(callback).not.toHaveBeenCalled()
+  })
+})

packages/e2e/src/state.test.ts

@@ -10,6 +10,7 @@ describe('state rpc', () => {
     await checkHex(api.rpc.state.getMetadata(env.acala.blockHash)).toMatchSnapshot()
     const genesisHash = await api.rpc.chain.getBlockHash(0)
     expect(await api.rpc.state.getMetadata(genesisHash)).to.not.be.eq(await api.rpc.state.getMetadata())
+    expect(await api.rpc.state.getReadProof(['0x123', '0x123'], env.acala.blockHash)).toMatchSnapshot()
   })
 
   it.todo('subscribeRuntimeVersion')