Allow fork rpc and block number to be specified

ourzora · Nov 14, 2023 · faeff94 · faeff94
1 parent 293e2c0
commit faeff94
Showing 1 changed file with 16 additions and 3 deletions.
diff --git a/packages/protocol-sdk/src/anvil.ts b/packages/protocol-sdk/src/anvil.ts
@@ -28,7 +28,10 @@ async function waitForAnvilInit(anvil: any) {
   });
 }
 
-export const anvilTest = test.extend<AnvilViemClientsTest>({
+export const makeAnvilTest = ({ forkUrl, forkBlockNumber }:{
+  forkUrl: string,
+  forkBlockNumber: number,
+}) => test.extend<AnvilViemClientsTest>({
   viemClients: async ({task}, use) => {
     console.log('setting up clients for ', task.name);
     const port = Math.floor(Math.random() * 2000) + 4000;
@@ -38,9 +41,9 @@ export const anvilTest = test.extend<AnvilViemClientsTest>({
         "--port",
         `${port}`,
         "--fork-url",
-        "https://rpc.zora.co/",
+        forkUrl,
         "--fork-block-number",
-        "6133407",
+        `${forkBlockNumber}`,
         "--chain-id",
         "31337",
       ],
@@ -82,3 +85,13 @@ export const anvilTest = test.extend<AnvilViemClientsTest>({
     anvil.kill("SIGINT");
   },
 });
+
+export const forkUrls = {
+  zoraMainnet: 'https://rpc.zora.co/',
+  zoraGoerli: 'https://testnet.rpc.zora.co'
+}
+
+export const anvilTest = makeAnvilTest({
+  forkUrl: forkUrls.zoraMainnet,
+  forkBlockNumber: 6133407
+});