test(cli): update cost model tests for single/multi network mode

Signed-off-by: Alexis Asseman <[email protected]>
graphprotocol · May 25, 2024 · 01934d3 · 01934d3
1 parent db0fd13
commit 01934d3
Show file tree

Hide file tree

Showing 7 changed files with 77 additions and 14 deletions.
diff --git a/packages/indexer-cli/src/__tests__/cli.test.ts b/packages/indexer-cli/src/__tests__/cli.test.ts
@@ -1,10 +1,10 @@
-import { cliTest, setup, teardown } from './util'
+import { cliTest, setupMultiNetworks, teardown } from './util'
 import path from 'path'
 
 const baseDir = path.join(__dirname)
 
 describe('Indexer cli tests', () => {
-  beforeEach(setup)
+  beforeEach(setupMultiNetworks)
   afterEach(teardown)
 
   describe('General', () => {

diff --git a/packages/indexer-cli/src/__tests__/indexer/actions.test.ts b/packages/indexer-cli/src/__tests__/indexer/actions.test.ts
@@ -1,10 +1,16 @@
-import { cliTest, deleteFromAllTables, seedActions, setup, teardown } from '../util'
+import {
+  cliTest,
+  deleteFromAllTables,
+  seedActions,
+  setupMultiNetworks,
+  teardown,
+} from '../util'
 import path from 'path'
 
 const baseDir = path.join(__dirname, '..')
 describe('Indexer actions tests', () => {
   describe('With indexer management server', () => {
-    beforeAll(setup)
+    beforeAll(setupMultiNetworks)
     afterAll(teardown)
     beforeEach(seedActions)
     afterEach(deleteFromAllTables)

diff --git a/packages/indexer-cli/src/__tests__/indexer/cost.test.ts b/packages/indexer-cli/src/__tests__/indexer/cost.test.ts
@@ -1,18 +1,19 @@
 import {
   cliTest,
-  setup,
   teardown,
   connect,
   deleteFromAllTables,
   seedCostModels,
+  setupSingleNetwork,
+  setupMultiNetworks,
 } from '../util'
 import path from 'path'
 
 const baseDir = path.join(__dirname, '..')
 
-describe('Indexer cost tests', () => {
+describe('Indexer cost tests singleNetwork', () => {
   describe('With indexer management server', () => {
-    beforeAll(setup)
+    beforeAll(setupSingleNetwork)
     afterAll(teardown)
     beforeEach(seedCostModels)
     afterEach(deleteFromAllTables)
@@ -231,3 +232,47 @@ describe('Indexer cost tests', () => {
     )
   })
 })
+
+describe('Indexer cost tests multiNetworks', () => {
+  beforeAll(setupMultiNetworks)
+  afterAll(teardown)
+  beforeEach(seedCostModels)
+  afterEach(deleteFromAllTables)
+
+  describe('Cost set...', () => {
+    cliTest(
+      'Indexer cost set model deployment id - reject multinetwork mode',
+      [
+        'indexer',
+        'cost',
+        'set',
+        'model',
+        'QmXRpJW3qBuYaiBYHdhv8DF4bHDZhXBmh91MtrnhJfQ5Lk',
+        'references/basic.agora',
+      ],
+      'references/indexer-cost-model-deployment-multinetworks',
+      {
+        expectedExitCode: 1,
+        cwd: baseDir,
+        timeout: 10000,
+      },
+    )
+    cliTest(
+      'Indexer cost set variable deployment id - reject multinetwork mode',
+      [
+        'indexer',
+        'cost',
+        'set',
+        'variables',
+        'QmQ44hgrWWt3Qf2X9XEX2fPyTbmQbChxwNm5c1t4mhKpGt',
+        `'{"DAI": "0.5"}'`,
+      ],
+      'references/indexer-cost-variables-deployment-multinetworks',
+      {
+        expectedExitCode: 1,
+        cwd: baseDir,
+        timeout: 10000,
+      },
+    )
+  })
+})
diff --git a/packages/indexer-cli/src/__tests__/indexer/rules.test.ts b/packages/indexer-cli/src/__tests__/indexer/rules.test.ts
@@ -1,18 +1,18 @@
 import {
   cliTest,
   connect,
-  setup,
   teardown,
   deleteFromAllTables,
   seedIndexingRules,
+  setupMultiNetworks,
 } from '../util'
 import path from 'path'
 
 const baseDir = path.join(__dirname, '..')
 
 describe('Indexer rules tests', () => {
   describe('With indexer management server', () => {
-    beforeAll(setup)
+    beforeAll(setupMultiNetworks)
     afterAll(teardown)
     beforeEach(seedIndexingRules)
     afterEach(deleteFromAllTables)

diff --git a/...s/indexer-cli/src/__tests__/references/indexer-cost-model-deployment-multinetworks.stdout b/...s/indexer-cli/src/__tests__/references/indexer-cost-model-deployment-multinetworks.stdout
@@ -0,0 +1 @@
+[GraphQL] Must be in single network mode to set cost models
diff --git a/...dexer-cli/src/__tests__/references/indexer-cost-variables-deployment-multinetworks.stdout b/...dexer-cli/src/__tests__/references/indexer-cost-variables-deployment-multinetworks.stdout
@@ -0,0 +1 @@
+[GraphQL] Must be in single network mode to set cost models
diff --git a/packages/indexer-cli/src/__tests__/util.ts b/packages/indexer-cli/src/__tests__/util.ts
@@ -103,7 +103,15 @@ export const testNetworkSpecification = specification.NetworkSpecification.parse
   },
 })
 
-export const setup = async () => {
+export const setupMultiNetworks = async () => {
+  return await setup(true)
+}
+
+export const setupSingleNetwork = async () => {
+  return await setup(false)
+}
+
+export const setup = async (multiNetworksEnabled: boolean) => {
   logger = createLogger({
     name: 'Setup',
     async: false,
@@ -141,10 +149,12 @@ export const setup = async () => {
   const fakeMainnetNetwork = cloneDeep(network) as Network
   fakeMainnetNetwork.specification.networkIdentifier = 'eip155:1'
 
-  const multiNetworks = new MultiNetworks(
-    [network, fakeMainnetNetwork],
-    (n: Network) => n.specification.networkIdentifier,
-  )
+  const multiNetworks = multiNetworksEnabled
+    ? new MultiNetworks(
+        [network, fakeMainnetNetwork],
+        (n: Network) => n.specification.networkIdentifier,
+      )
+    : new MultiNetworks([network], (n: Network) => n.specification.networkIdentifier)
 
   const defaults: IndexerManagementDefaults = {
     globalIndexingRule: {
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		[GraphQL] Must be in single network mode to set cost models