Fix mocks

Emyrk · Emyrk · commit ed4aebf6a231 · 2023-06-01T09:21:25.000-04:00
diff --git a/site/jest.setup.ts b/site/jest.setup.ts
@@ -19,7 +19,7 @@ jest.mock("contexts/useProxyLatency", () => ({
   useProxyLatency: (proxies?: RegionsResponse) => {
     // Must use `useMemo` here to avoid infinite loop.
     // Mocking the hook with a hook.
-    const latencies = useMemo(() => {
+    const proxyLatencies = useMemo(() => {
       if (!proxies) {
         return {} as Record<string, ProxyLatencyReport>
       }
@@ -35,7 +35,7 @@ jest.mock("contexts/useProxyLatency", () => ({
       }, {} as Record<string, ProxyLatencyReport>)
     }, [proxies])
 
-    return latencies
+    return {proxyLatencies, refetch: jest.fn() }
   },
 }))
 
diff --git a/site/src/contexts/ProxyContext.test.tsx b/site/src/contexts/ProxyContext.test.tsx
@@ -26,7 +26,7 @@ import userEvent from "@testing-library/user-event"
 // here and not inside a unit test.
 jest.mock("contexts/useProxyLatency", () => ({
   useProxyLatency: () => {
-    return hardCodedLatencies
+    return { proxyLatencies: hardCodedLatencies, refetch: jest.fn() }
   },
 }))
 
