Pass refetch funtion up the context stack

Emyrk · Emyrk · commit d52d179898c1 · 2023-06-01T09:00:12.000-04:00
diff --git a/site/src/components/AppLink/AppLink.stories.tsx b/site/src/components/AppLink/AppLink.stories.tsx
@@ -29,6 +29,9 @@ const Template: Story<AppLinkProps> = (args) => (
       clearProxy: () => {
         return
       },
+      refetchProxyLatencies: () => {
+        return
+      },
     }}
   >
     <AppLink {...args} />
diff --git a/site/src/components/Navbar/NavbarView.test.tsx b/site/src/components/Navbar/NavbarView.test.tsx
@@ -19,6 +19,7 @@ const proxyContextValue: ProxyContextValue = {
   isFetched: true,
   setProxy: jest.fn(),
   clearProxy: action("clearProxy"),
+  refetchProxyLatencies: jest.fn(),
   proxyLatencies: {},
 }
 
diff --git a/site/src/components/Resources/AgentRow.stories.tsx b/site/src/components/Resources/AgentRow.stories.tsx
@@ -68,6 +68,9 @@ const TemplateFC = (
         clearProxy: () => {
           return
         },
+        refetchProxyLatencies: () => {
+          return
+        },
       }}
     >
       <AgentRow {...args} />
diff --git a/site/src/components/Resources/ResourceCard.stories.tsx b/site/src/components/Resources/ResourceCard.stories.tsx
@@ -33,6 +33,9 @@ Example.args = {
         clearProxy: () => {
           return
         },
+        refetchProxyLatencies: () => {
+          return
+        },
       }}
     >
       <AgentRow
@@ -103,6 +106,9 @@ BunchOfMetadata.args = {
         clearProxy: () => {
           return
         },
+        refetchProxyLatencies: () => {
+          return
+        },
       }}
     >
       <AgentRow
diff --git a/site/src/components/Workspace/Workspace.stories.tsx b/site/src/components/Workspace/Workspace.stories.tsx
@@ -53,6 +53,9 @@ const Template: Story<WorkspaceProps> = (args) => (
         setProxy: () => {
           return
         },
+        refetchProxyLatencies: () => {
+          return
+        },
       }}
     >
       <Workspace {...args} />
diff --git a/site/src/contexts/ProxyContext.tsx b/site/src/contexts/ProxyContext.tsx
@@ -45,6 +45,9 @@ export interface ProxyContextValue {
   // then the latency has not been fetched yet. Calculations happen async for each proxy in the list.
   // Refer to the returned report for a given proxy for more information.
   proxyLatencies: Record<string, ProxyLatencyReport>
+  // refetchProxyLatencies will trigger refreshing of the proxy latencies. By default the latencies
+  // are loaded once.
+  refetchProxyLatencies: () => void
   // setProxy is a function that sets the user's selected proxy. This function should
   // only be called if the user is manually selecting a proxy. This value is stored in local
   // storage and will persist across reloads and tabs.
@@ -101,7 +104,8 @@ export const ProxyProvider: FC<PropsWithChildren> = ({ children }) => {
 
   // Every time we get a new proxiesResponse, update the latency check
   // to each workspace proxy.
-  const { proxyLatencies } = useProxyLatency(proxiesResp)
+  const { proxyLatencies, refetch: refetchProxyLatencies } =
+    useProxyLatency(proxiesResp)
 
   // updateProxy is a helper function that when called will
   // update the proxy being used.
@@ -128,6 +132,7 @@ export const ProxyProvider: FC<PropsWithChildren> = ({ children }) => {
     <ProxyContext.Provider
       value={{
         proxyLatencies,
+        refetchProxyLatencies,
         userProxy: userSavedProxy,
         proxy: experimentEnabled
           ? proxy
diff --git a/site/src/pages/TerminalPage/TerminalPage.test.tsx b/site/src/pages/TerminalPage/TerminalPage.test.tsx
@@ -55,6 +55,7 @@ const renderTerminal = () => {
               isLoading: false,
               setProxy: jest.fn(),
               clearProxy: jest.fn(),
+              refetchProxyLatencies: jest.fn(),
             }}
           >
             <TerminalPage renderer="dom" />

Original file line number	Diff line number	Diff line change
`@@ -19,6 +19,7 @@ const proxyContextValue: ProxyContextValue = {`
`19`	`19`	`isFetched: true,`
`20`	`20`	`setProxy: jest.fn(),`
`21`	`21`	`clearProxy: action("clearProxy"),`
	`22`	`+ refetchProxyLatencies: jest.fn(),`
`22`	`23`	`proxyLatencies: {},`
`23`	`24`	`}`
`24`	`25`
Original file line number	Diff line number	Diff line change
`@@ -55,6 +55,7 @@ const renderTerminal = () => {`
`55`	`55`	`isLoading: false,`
`56`	`56`	`setProxy: jest.fn(),`
`57`	`57`	`clearProxy: jest.fn(),`
	`58`	`+ refetchProxyLatencies: jest.fn(),`
`58`	`59`	`}}`
`59`	`60`	`>`
`60`	`61`	`<TerminalPage renderer="dom" />`