Allow cost models in single network mode

packages/indexer-agent/src/commands/start.ts

@@ -43,6 +43,7 @@ const SUGGESTED_SUBGRAPH_MAX_BLOCK_DISTANCE_ON_L2 =
   50 + DEFAULT_SUBGRAPH_MAX_BLOCK_DISTANCE
 const DEFAULT_SUBGRAPH_FRESHNESS_SLEEP_MILLISECONDS = 5_000
 
+// NOTE: This is run only in single-network mode
 export const start = {
   command: 'start',
   describe: 'Start the agent',
@@ -237,7 +238,7 @@ export const start = {
         description:
           'Inject the GRT to DAI/USDC conversion rate into cost model variables',
         type: 'boolean',
-        default: true,
+        default: false,
         group: 'Cost Models',
       })
       .option('address-book', {
@@ -439,6 +440,12 @@ export async function createNetworkSpecification(
     }
   }
 
+  if (chainId !== 1 && dai.inject) {
+    throw new Error(
+      `The DAI injection feature for cost models is only supported on Ethereum Mainnet`,
+    )
+  }
+
   const tapAddressBook = loadFile(argv.tapAddressBook)
 
   try {

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', () => {

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)

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,
+      },
+    )
+  })
+})

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)

packages/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

packages/indexer-cli/src/__tests__/references/indexer-cost-variables-deployment-multinetworks.stdout

@@ -0,0 +1 @@
+[GraphQL] Must be in single network mode to set cost models

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: {

packages/indexer-common/src/indexer-management/resolvers/cost-models.ts

@@ -98,7 +98,17 @@ export default {
     { models, multiNetworks, dai }: IndexerManagementResolverContext,
   ): Promise<object> => {
     if (!multiNetworks) {
-      throw Error('IndexerManagementClient must be in `network` mode to set cost models')
+      throw new Error('No network configuration available')
+    }
+    if (Object.keys(multiNetworks.inner).length !== 1) {
+      throw Error('Must be in single network mode to set cost models')
+    }
+    const network = Object.values(multiNetworks.inner)[0]
+    const injectDai = network.specification.dai.inject
+    if (network.specification.networkIdentifier !== 'eip155:1' && injectDai) {
+      throw new Error(
+        `Can't set cost model: DAI injection enabled but not on Ethereum Mainnet`,
+      )
     }
 
     const update = parseGraphQLCostModel(costModel)
@@ -111,13 +121,6 @@ export default {
     } catch (err) {
       throw new Error(`Invalid cost model or variables: ${err.message}`)
     }
-    const network = multiNetworks.inner['eip155:1']
-    if (!network) {
-      throw new Error(
-        `Can't set cost model: Indexer Agent does not have Ethereum Mainnet network configured.`,
-      )
-    }
-    const injectDai = !!network.specification.dai.inject
     const [model] = await models.CostModel.findOrBuild({
       where: { deployment: update.deployment },
     })