export signFake helper (#379)

* export sign fake

* Update packages/chopsticks/src/utils/signFake.ts

Co-authored-by: Xiliang Chen <xlchen1291@gmail.com>

* fix

* fix lint

---------

Co-authored-by: Xiliang Chen <xlchen1291@gmail.com>

packages/chopsticks/src/index.ts

@@ -1,3 +1,5 @@
+import '@polkadot/api-augment'
+
 export * from '@acala-network/chopsticks-core'
 export { setupWithServer } from './setup-with-server'
 export { fetchConfig } from './schema'

packages/chopsticks/src/utils/index.ts

@@ -0,0 +1,5 @@
+export * from './decoder'
+export * from './generate-html-diff'
+export * from './open-html'
+export * from './override'
+export * from './signFake'

packages/chopsticks/src/utils/signFake.ts

@@ -0,0 +1,33 @@
+import { ApiPromise } from '@polkadot/api'
+import { GenericExtrinsic } from '@polkadot/types'
+import { SignatureOptions } from '@polkadot/types/types'
+
+export type SignFakeOptions = Partial<SignatureOptions>
+
+export const signFakeWithApi = async (
+  api: ApiPromise,
+  tx: GenericExtrinsic,
+  addr: string,
+  options: SignFakeOptions = {},
+) => {
+  const nonce = options.nonce ?? (await api.query.system.account(addr)).nonce
+
+  return signFake(tx, addr, {
+    nonce,
+    genesisHash: api.genesisHash,
+    runtimeVersion: api.runtimeVersion,
+    blockHash: api.genesisHash,
+    ...options,
+  })
+}
+
+export const signFake = (tx: GenericExtrinsic, addr: string, options: SignatureOptions) => {
+  const mockSignature = new Uint8Array(64)
+  mockSignature.fill(0xcd)
+  mockSignature.set([0xde, 0xad, 0xbe, 0xef])
+  tx.signFake(addr, options)
+
+  tx.signature.set(mockSignature)
+
+  return tx
+}

packages/e2e/src/mock-signature.test.ts

@@ -1,4 +1,5 @@
 import { describe, expect, it } from 'vitest'
+import { signFake, signFakeWithApi } from '@acala-network/chopsticks/utils'
 
 import { api, dev, env, setupApi, testingPairs } from './helper'
 
@@ -39,7 +40,22 @@ describe('mock signature', () => {
     await expect(tx.send()).rejects.toThrow('1010: {"invalid":{"badProof":null}}')
   })
 
-  it('accept mock signature', async () => {
+  it('accept mock signature (with api)', async () => {
+    const { alice, bob } = testingPairs()
+    await dev.setStorage({
+      System: {
+        Account: [[[alice.address], { providers: 1, data: { free: 1000 * 1e12 } }]],
+      },
+    })
+
+    const tx = api.tx.balances.transfer(bob.address, 100)
+
+    await signFakeWithApi(api, tx, alice.address)
+
+    await expect(tx.send()).resolves.toBeTruthy()
+  })
+
+  it('accept mock signature (manually input options)', async () => {
     const { alice, bob } = testingPairs()
     await dev.setStorage({
       System: {
@@ -50,18 +66,13 @@ describe('mock signature', () => {
     const { nonce } = await api.query.system.account(alice.address)
     const tx = api.tx.balances.transfer(bob.address, 100)
 
-    tx.signFake(alice.address, {
+    signFake(tx, alice.address, {
       nonce,
       genesisHash: api.genesisHash,
       runtimeVersion: api.runtimeVersion,
       blockHash: api.genesisHash,
     })
 
-    const mockSignature = new Uint8Array(64)
-    mockSignature.fill(0xcd)
-    mockSignature.set([0xde, 0xad, 0xbe, 0xef])
-    tx.signature.set(mockSignature)
-
     await expect(tx.send()).resolves.toBeTruthy()
   })
 })