feat: add mock-qa-rewards-address for rewards table

lidofinance · Aug 6, 2024 · 9c3484d · 9c3484d
1 parent 368dc6c
commit 9c3484d
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/features/rewards/hooks/useGetCurrentAddress.ts b/features/rewards/hooks/useGetCurrentAddress.ts
@@ -6,6 +6,7 @@ import { useSDK } from '@lido-sdk/react';
 
 import { resolveEns, isValidEns, isValidAddress } from 'features/rewards/utils';
 import { useCurrentStaticRpcProvider } from 'shared/hooks/use-current-static-rpc-provider';
+import { overrideWithQAMockString } from 'utils/qa';
 
 type UseGetCurrentAddress = () => {
   address: string;
@@ -86,7 +87,11 @@ export const useGetCurrentAddress: UseGetCurrentAddress = () => {
         return;
       }
       // From a connected wallet
-      if (account) setInputValue(account);
+      if (account) {
+        setInputValue(
+          overrideWithQAMockString(account, 'mock-qa-rewards-address'),
+        );
+      }
     }
   }, [account, query.address, isReady, setInputValue]);
 

diff --git a/utils/qa.ts b/utils/qa.ts
@@ -20,6 +20,16 @@ export const overrideWithQAMockNumber = (value: number, key: string) => {
   return value;
 };
 
+export const overrideWithQAMockString = (value: string, key: string) => {
+  if (config.enableQaHelpers && typeof window !== 'undefined') {
+    const mock = localStorage.getItem(key);
+    if (mock) {
+      return mock;
+    }
+  }
+  return value;
+};
+
 export const overrideWithQAMockArray = <TArrayElement>(
   value: TArrayElement[],
   key: string,