From 664900c69eb7bd404b3ca6b96d291ff88d1cdd9f Mon Sep 17 00:00:00 2001
From: iamacook <aaron@safe.global>
Date: Tue, 10 Oct 2023 14:35:46 +0200
Subject: [PATCH] fix: WC search param test

---
 .../useWalletConnectSearchParamUri.test.ts    | 26 ++++++++++++-------
 1 file changed, 16 insertions(+), 10 deletions(-)

diff --git a/src/services/walletconnect/__tests__/useWalletConnectSearchParamUri.test.ts b/src/services/walletconnect/__tests__/useWalletConnectSearchParamUri.test.ts
index 3fc8d34822..584515b4cf 100644
--- a/src/services/walletconnect/__tests__/useWalletConnectSearchParamUri.test.ts
+++ b/src/services/walletconnect/__tests__/useWalletConnectSearchParamUri.test.ts
@@ -5,11 +5,15 @@ import { useWalletConnectSearchParamUri } from '../useWalletConnectSearchParamUr
 
 describe('useWalletConnectSearchParamUri', () => {
   const mockRouter = {
+    pathname: '/',
     query: {},
     replace: jest.fn(),
   } as unknown as router.NextRouter
 
   beforeEach(() => {
+    mockRouter.pathname = '/'
+    mockRouter.query = {}
+
     jest.spyOn(router, 'useRouter').mockReturnValue(mockRouter)
   })
 
@@ -34,6 +38,9 @@ describe('useWalletConnectSearchParamUri', () => {
   })
 
   it('should update the wc uri search param value when setWcUri is called', () => {
+    mockRouter.pathname = '/test'
+    mockRouter.query = { test: 'example', wc: 'wc:123' }
+
     const { result } = renderHook(() => useWalletConnectSearchParamUri())
     const [wcUri, setWcUri] = result.current
 
@@ -44,15 +51,15 @@ describe('useWalletConnectSearchParamUri', () => {
     })
 
     expect(mockRouter.replace).toHaveBeenCalledWith({
-      pathname: mockRouter.pathname,
-      query: { wc: 'wc:456' },
+      pathname: '/test',
+      // Preserves other query params
+      query: { test: 'example', wc: 'wc:456' },
     })
-
-    expect(wcUri).toBe('wc:456')
   })
 
-  it('should remove the wc uri search param when setWcUri is called with null', () => {
-    mockRouter.query = { wc: 'wc:123' }
+  it('should remove the wc uri search param when setWcUri is called with null', async () => {
+    mockRouter.pathname = '/test'
+    mockRouter.query = { test: 'example', wc: 'wc:123' }
 
     const { result } = renderHook(() => useWalletConnectSearchParamUri())
     const [wcUri, setWcUri] = result.current
@@ -64,10 +71,9 @@ describe('useWalletConnectSearchParamUri', () => {
     })
 
     expect(mockRouter.replace).toHaveBeenCalledWith({
-      pathname: mockRouter.pathname,
-      query: {},
+      pathname: '/test',
+      // Preserves other query params
+      query: { test: 'example' },
     })
-
-    expect(wcUri).toBeNull()
   })
 })