From 3a49222d1202c5284e9840ea76ef3fb3ebfb35ff Mon Sep 17 00:00:00 2001 From: Ryan Miller Date: Thu, 4 Jul 2024 11:14:37 +1000 Subject: [PATCH] fix: add feature flag mock data test to open node view --- .../app/stake/node/[nodeId]/NodeStaking.tsx | 16 +++++++++++++++- 1 file changed, 15 insertions(+), 1 deletion(-) diff --git a/apps/staking/app/stake/node/[nodeId]/NodeStaking.tsx b/apps/staking/app/stake/node/[nodeId]/NodeStaking.tsx index a256175b..708ed907 100644 --- 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 ? (