diff --git a/src/jumpstart/jumpstartLinkHandler.test.ts b/src/jumpstart/jumpstartLinkHandler.test.ts index 7066832813d..5ae61f73ee8 100644 --- a/src/jumpstart/jumpstartLinkHandler.test.ts +++ b/src/jumpstart/jumpstartLinkHandler.test.ts @@ -1,6 +1,7 @@ import { getDynamicConfigParams } from 'src/statsig' import Logger from 'src/utils/Logger' import { fetchWithTimeout } from 'src/utils/fetchWithTimeout' +import networkConfig from 'src/web3/networkConfig' import { mockAccount, mockAccount2 } from 'test/values' import { jumpstartLinkHandler } from './jumpstartLinkHandler' @@ -43,7 +44,9 @@ describe('jumpstartLinkHandler', () => { ;(fetchWithTimeout as jest.Mock).mockImplementation(() => ({ ok: true, })) - jest.mocked(getDynamicConfigParams).mockReturnValue({ celo: { contractAddress: '0xTEST' } }) + jest.mocked(getDynamicConfigParams).mockReturnValue({ + jumpstartContracts: { [networkConfig.defaultNetworkId]: { contractAddress: '0xTEST' } }, + }) await jumpstartLinkHandler(privateKey, mockAccount2) diff --git a/src/jumpstart/jumpstartLinkHandler.ts b/src/jumpstart/jumpstartLinkHandler.ts index afa3ae136b3..4b3519b3e62 100644 --- a/src/jumpstart/jumpstartLinkHandler.ts +++ b/src/jumpstart/jumpstartLinkHandler.ts @@ -4,7 +4,6 @@ import jumpstartAbi from 'src/abis/WalletJumpStart.json' import { getDynamicConfigParams } from 'src/statsig' import { DynamicConfigs } from 'src/statsig/constants' import { StatsigDynamicConfigs } from 'src/statsig/types' -import { Network } from 'src/transactions/types' import Logger from 'src/utils/Logger' import { fetchWithTimeout } from 'src/utils/fetchWithTimeout' import { getWeb3Async } from 'src/web3/contracts' @@ -16,7 +15,7 @@ const TAG = 'WalletJumpstart' export async function jumpstartLinkHandler(privateKey: string, userAddress: string) { const contractAddress = getDynamicConfigParams( DynamicConfigs[StatsigDynamicConfigs.WALLET_JUMPSTART_CONFIG] - )?.[Network.Celo]?.contractAddress + ).jumpstartContracts?.[networkConfig.defaultNetworkId]?.contractAddress if (!contractAddress) { Logger.error(TAG, 'Contract address is not provided in dynamic config') diff --git a/src/statsig/constants.ts b/src/statsig/constants.ts index 58089f5c935..92c08ef7dca 100644 --- a/src/statsig/constants.ts +++ b/src/statsig/constants.ts @@ -1,5 +1,5 @@ import { StatsigDynamicConfigs, StatsigExperiments, StatsigFeatureGates } from 'src/statsig/types' -import { Network, NetworkId } from 'src/transactions/types' +import { NetworkId } from 'src/transactions/types' import networkConfig from 'src/web3/networkConfig' export const FeatureGates = { @@ -101,7 +101,9 @@ export const DynamicConfigs = { }, [StatsigDynamicConfigs.WALLET_JUMPSTART_CONFIG]: { configName: StatsigDynamicConfigs.WALLET_JUMPSTART_CONFIG, - defaultValues: {} as { [key in Network]?: { contractAddress?: string } }, + defaultValues: { + jumpstartContracts: {} as { [key in NetworkId]?: { contractAddress?: string } }, + }, }, [StatsigDynamicConfigs.NFT_CELEBRATION_CONFIG]: { configName: StatsigDynamicConfigs.NFT_CELEBRATION_CONFIG,