feat: test client resetFork decorator

packages/react/src/hooks/useBlockNumber.test.ts

@@ -5,6 +5,8 @@ import { expect, test } from 'vitest'
 import { useBlockNumber } from './useBlockNumber.js'
 
 test('mounts', async () => {
+ await testClient.mainnet.resetFork()
+
  const { result } = renderHook(() => useBlockNumber())
 
  await waitFor(() => expect(result.current.isSuccess).toBeTruthy())

packages/test/src/chains.ts

@@ -9,7 +9,7 @@ import { getRpcUrls } from './utils.js'
 
 type Fork = { blockNumber: bigint; url: string }
 
-type Chain = Evaluate<
+export type Chain = Evaluate<
  viem_Chain & {
  fork: Fork
  port: number

packages/test/src/clients.ts

@@ -1,30 +1,57 @@
-import { createTestClient, http } from 'viem'
+import {
+ type Account,
+ type Client,
+ type TestActions,
+ type TestRpcSchema,
+ type Transport,
+ createTestClient,
+ http,
+} from 'viem'
 
-import { mainnet, mainnet2, optimism } from './chains.js'
+import { type Chain, mainnet, mainnet2, optimism } from './chains.js'
 
 export const mainnetTestClient = createTestClient({
  mode: 'anvil',
  cacheTime: 0,
  chain: mainnet,
  transport: http(),
-})
+}).extend(decorator)
 
 export const mainnet2TestClient = createTestClient({
  mode: 'anvil',
  cacheTime: 0,
  chain: mainnet2,
  transport: http(),
-})
+}).extend(decorator)
 
 export const optimismTestClient = createTestClient({
  mode: 'anvil',
  cacheTime: 0,
  chain: optimism,
  transport: http(),
-})
+}).extend(decorator)
 
 export const testClient = {
  mainnet: mainnetTestClient,
  mainnet2: mainnet2TestClient,
  optimism: optimismTestClient,
 }
+
+function decorator(
+ client: Client<
+ Transport,
+ Chain,
+ Account | undefined,
+ TestRpcSchema<'anvil'>,
+ TestActions
+ >,
+) {
+ return {
+ resetFork() {
+ return client.reset({
+ jsonRpcUrl: client.chain.fork.url,
+ blockNumber: client.chain.fork.blockNumber,
+ })
+ },
+ }
+}

packages/test/src/setup.ts

@@ -8,12 +8,7 @@ import { pool } from './constants.js'
 afterAll(async () => {
  // If you are using a fork, you can reset your anvil instance to the initial fork block.
  await Promise.all(
- Object.values(testClient).map((client) =>
- client.reset({
- jsonRpcUrl: client.chain.fork.url,
- blockNumber: client.chain.fork.blockNumber,
- }),
- ),
+ Object.values(testClient).map((client) => client.resetFork()),
  )
 })