fixup! feat(eth-staking): update account staked balance ui

everstake · Feb 28, 2024 · 55eb99e · 55eb99e
1 parent f6685e4
commit 55eb99e
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/packages/suite/src/components/wallet/WalletLayout/AccountBanners/StakeEthBanner.tsx b/packages/suite/src/components/wallet/WalletLayout/AccountBanners/StakeEthBanner.tsx
@@ -1,4 +1,3 @@
-import { useLocation } from 'react-router-dom';
 import styled, { useTheme } from 'styled-components';
 import { Button, Card, Icon, Paragraph, variables, IconButton } from '@trezor/components';
 import { spacingsPx } from '@trezor/theme';
@@ -53,7 +52,7 @@ export const StakeEthBanner = () => {
     const isDebug = useSelector(selectIsDebugModeActive);
     const { stakeEthBannerClosed } = useSelector(state => state.suite.flags);
     const hasSufficientEthForStaking = useSelector(selectSelectedAccountHasSufficientEthForStaking);
-    const { pathname } = useLocation();
+    const { pathname } = useSelector(state => state.router);
     const isShown = !stakeEthBannerClosed && pathname === '/accounts' && hasSufficientEthForStaking;
     const { ethApy } = useEverstakePoolStats();
 

diff --git a/packages/suite/src/hooks/wallet/__fixtures__/useSendForm.ts b/packages/suite/src/hooks/wallet/__fixtures__/useSendForm.ts
@@ -181,6 +181,7 @@ export const getRootReducer = (selectedAccount = BTC_ACCOUNT, fees = DEFAULT_FEE
                 online: true,
                 settings: { debug: {}, theme: { variant: 'light' } },
                 evmSettings: { confirmExplanationModalClosed: {} },
+                flags: { stakeEthBannerClosed: false },
             },
             () => ({}),
         ),

diff --git a/suite-common/wallet-utils/src/__tests__/accountUtils.test.ts b/suite-common/wallet-utils/src/__tests__/accountUtils.test.ts
@@ -254,6 +254,7 @@ describe('account utils', () => {
             'sign-verify',
             'tokens',
             'token-definitions',
+            'staking',
         ]);
         expect(getNetworkFeatures(coinjoinAcc)).toEqual(['rbf', 'amount-unit']);
     });