fix test code

coulsontl · May 19, 2024 · 725a1de · 725a1de
1 parent 2217644
commit 725a1de
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/src/features/User/__tests__/useMenu.test.tsx b/src/features/User/__tests__/useMenu.test.tsx
@@ -2,6 +2,7 @@ import { act, renderHook } from '@testing-library/react';
 import { describe, expect, it, vi } from 'vitest';
 
 import { useUserStore } from '@/store/user';
+import { ServerConfigStoreProvider } from '@/store/serverConfig';
 
 import { useMenu } from '../UserPanel/useMenu';
 
@@ -69,7 +70,7 @@ describe('useMenu', () => {
     enableAuth = true;
     enableClerk = false;
 
-    const { result } = renderHook(() => useMenu());
+    const { result } = renderHook(() => useMenu(), { wrapper: ({children})=> <ServerConfigStoreProvider>{children}</ServerConfigStoreProvider> });
 
     act(() => {
       const { mainItems, logoutItems } = result.current;
@@ -89,7 +90,7 @@ describe('useMenu', () => {
     enableAuth = true;
     enableClerk = true;
 
-    const { result } = renderHook(() => useMenu());
+    const { result } = renderHook(() => useMenu(), { wrapper: ({children})=> <ServerConfigStoreProvider>{children}</ServerConfigStoreProvider> });
 
     act(() => {
       const { mainItems, logoutItems } = result.current;
@@ -108,7 +109,7 @@ describe('useMenu', () => {
     });
     enableAuth = false;
 
-    const { result } = renderHook(() => useMenu());
+    const { result } = renderHook(() => useMenu(), { wrapper: ({children})=> <ServerConfigStoreProvider>{children}</ServerConfigStoreProvider> });
 
     act(() => {
       const { mainItems, logoutItems } = result.current;
@@ -127,7 +128,7 @@ describe('useMenu', () => {
     });
     enableAuth = true;
 
-    const { result } = renderHook(() => useMenu());
+    const { result } = renderHook(() => useMenu(), { wrapper: ({children})=> <ServerConfigStoreProvider>{children}</ServerConfigStoreProvider> });
 
     act(() => {
       const { mainItems, logoutItems } = result.current;

diff --git a/src/store/serverConfig/selectors.test.ts b/src/store/serverConfig/selectors.test.ts
@@ -24,6 +24,8 @@ describe('featureFlagsSelectors', () => {
       showLLM: false,
       showOpenAIApiKey: true,
       showOpenAIProxyUrl: true,
+      enableCheckUpdates: true,
+      showWelcomeSuggest: true,
     });
   });
 });