diff --git a/app/components/UI/Stake/components/StakeButton/index.tsx b/app/components/UI/Stake/components/StakeButton/index.tsx index 9b13d2fe4bf..9e18af410f0 100644 --- a/app/components/UI/Stake/components/StakeButton/index.tsx +++ b/app/components/UI/Stake/components/StakeButton/index.tsx @@ -25,6 +25,8 @@ import { RootState } from '../../../../../reducers'; import useStakingEligibility from '../../hooks/useStakingEligibility'; import { StakeSDKProvider } from '../../sdk/stakeSdkProvider'; import { EVENT_LOCATIONS } from '../../constants/events'; +import useStakingChain from '../../hooks/useStakingChain'; +import Engine from '../../../../../core/Engine'; interface StakeButtonProps { asset: TokenI; @@ -37,11 +39,14 @@ const StakeButtonContent = ({ asset }: StakeButtonProps) => { const browserTabs = useSelector((state: RootState) => state.browser.tabs); const chainId = useSelector(selectChainId); - - const { refreshPooledStakingEligibility } = useStakingEligibility(); + const { isEligible } = useStakingEligibility(); + const { isStakingSupportedChain } = useStakingChain(); const onStakeButtonPress = async () => { - const { isEligible } = await refreshPooledStakingEligibility(); + if (!isStakingSupportedChain) { + const { NetworkController } = Engine.context; + await NetworkController.setActiveNetwork('mainnet'); + } if (isEligible) { navigation.navigate('StakeScreens', { screen: Routes.STAKING.STAKE }); } else { diff --git a/app/components/UI/Tokens/index.test.tsx b/app/components/UI/Tokens/index.test.tsx index e8a0df5103e..838143c7b21 100644 --- a/app/components/UI/Tokens/index.test.tsx +++ b/app/components/UI/Tokens/index.test.tsx @@ -248,10 +248,14 @@ jest.mock('../../UI/Stake/hooks/useStakingEligibility', () => ({ })), })); -jest.mock('../Stake/hooks/useStakingChain', () => ({ - useStakingChainByChainId: () => ({ +jest.mock('../../UI/Stake/hooks/useStakingChain', () => ({ + __esModule: true, + default: jest.fn(() => ({ isStakingSupportedChain: true, - }), + })), + useStakingChainByChainId: jest.fn(() => ({ + isStakingSupportedChain: true, + })), })); const Stack = createStackNavigator();