From f5ecdd216449d7f16d09ca0d8f8c5218d5231f09 Mon Sep 17 00:00:00 2001
From: Stanislav Lysak <lysaksa@amofintech.com>
Date: Mon, 2 Dec 2024 12:10:49 +0200
Subject: [PATCH] lint fix

---
 src/components/ConnectButton.tsx | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/src/components/ConnectButton.tsx b/src/components/ConnectButton.tsx
index ce69c60a6..7efa864d7 100644
--- a/src/components/ConnectButton.tsx
+++ b/src/components/ConnectButton.tsx
@@ -90,6 +90,12 @@ const PersonOverlay = styled.div(
   `,
 )
 
+const getTestId = (isTabBar: boolean | undefined, inHeader: boolean | undefined) => {
+  if (isTabBar) return 'tabbar-connect-button'
+  if (!inHeader) return 'body-connect-button'
+  return 'connect-button'
+}
+
 type Props = {
   isTabBar?: boolean
   large?: boolean
@@ -104,9 +110,7 @@ export const ConnectButton = ({ isTabBar, large, inHeader }: Props) => {
   return (
     <StyledButtonWrapper $large={large} $isTabBar={isTabBar}>
       <Button
-        data-testid={
-          isTabBar ? 'tabbar-connect-button' : !inHeader ? 'body-connect-button' : 'connect-button'
-        }
+        data-testid={getTestId(isTabBar, !inHeader)}
         onClick={() => openConnectModal?.()}
         size={breakpoints.sm || large ? 'medium' : 'small'}
         width={inHeader ? '45' : undefined}