fix: add feature flag mock data test to open node view

oxen-io · Jul 4, 2024 · 3a49222 · 3a49222
1 parent 1e8e991
commit 3a49222
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/apps/staking/app/stake/node/[nodeId]/NodeStaking.tsx b/apps/staking/app/stake/node/[nodeId]/NodeStaking.tsx
@@ -16,12 +16,26 @@ import { useMemo, useState } from 'react';
 import { ActionModuleDivider, ActionModuleRow } from '../../ActionModule';
 
 export default function NodeStaking({ nodeId }: { nodeId: string }) {
+  const showMockNodes = useFeatureFlag(FEATURE_FLAG.MOCK_OPEN_NODES);
+  const showNoNodes = useFeatureFlag(FEATURE_FLAG.MOCK_NO_OPEN_NODES);
+
+  if (showMockNodes && showNoNodes) {
+    console.error('Cannot show mock nodes and no nodes at the same time');
+  }
+
   const { data, isLoading } = useSessionStakingQuery({
     query: 'getOpenNodes',
     args: undefined,
   });
 
-  const node = useMemo(() => data?.nodes.find((node) => node.pubKey === nodeId), [data, nodeId]);
+  const node = useMemo(() => {
+    if (showMockNodes) {
+      return generateOpenNodes({ userAddress: address })[0];
+    } else if (showNoNodes) {
+      return {} as OpenNode;
+    }
+    return data?.nodes.find((node) => node.pubKey === nodeId);
+  }, [data, nodeId, showMockNodes, showNoNodes]);
 
   return isLoading ? (
     <Loading />