Add mock latencies to tests/storybook

coder · Emyrk · May 11, 2023 · May 9, 2023 · May 9, 2023 · May 10, 2023
commit b90eda0250a8b6c0030cd74bae8c2ac7d57fbd0a
diff --git a/site/src/components/AppLink/AppLink.stories.tsx b/site/src/components/AppLink/AppLink.stories.tsx
@@ -5,6 +5,7 @@ import {
   MockWorkspace,
   MockWorkspaceAgent,
   MockWorkspaceApp,
+  MockProxyLatencies,
 } from "testHelpers/entities"
 import { AppLink, AppLinkProps } from "./AppLink"
 import { ProxyContext, getPreferredProxy } from "contexts/ProxyContext"
@@ -17,6 +18,7 @@ export default {
 const Template: Story<AppLinkProps> = (args) => (
   <ProxyContext.Provider
     value={{
+      proxyLatencies: MockProxyLatencies,
       proxy: getPreferredProxy(MockWorkspaceProxies, MockPrimaryWorkspaceProxy),
       proxies: MockWorkspaceProxies,
       isLoading: false,

diff --git a/site/src/components/Resources/AgentRow.stories.tsx b/site/src/components/Resources/AgentRow.stories.tsx
@@ -16,6 +16,7 @@ import {
   MockWorkspaceAgentStartTimeout,
   MockWorkspaceAgentTimeout,
   MockWorkspaceApp,
+  MockProxyLatencies,
 } from "testHelpers/entities"
 import { AgentRow, AgentRowProps } from "./AgentRow"
 import { ProxyContext, getPreferredProxy } from "contexts/ProxyContext"
@@ -56,6 +57,7 @@ const TemplateFC = (
   return (
     <ProxyContext.Provider
       value={{
+        proxyLatencies: MockProxyLatencies,
         proxy: getPreferredProxy(proxies, selectedProxy),
         proxies: proxies,
         isLoading: false,

diff --git a/site/src/components/Resources/ResourceCard.stories.tsx b/site/src/components/Resources/ResourceCard.stories.tsx
@@ -1,6 +1,10 @@
 import { action } from "@storybook/addon-actions"
 import { Story } from "@storybook/react"
-import { MockWorkspace, MockWorkspaceResource } from "testHelpers/entities"
+import {
+  MockProxyLatencies,
+  MockWorkspace,
+  MockWorkspaceResource,
+} from "testHelpers/entities"
 import { AgentRow } from "./AgentRow"
 import { ResourceCard, ResourceCardProps } from "./ResourceCard"
 import { ProxyContext, getPreferredProxy } from "contexts/ProxyContext"
@@ -18,6 +22,7 @@ Example.args = {
   agentRow: (agent) => (
     <ProxyContext.Provider
       value={{
+        proxyLatencies: MockProxyLatencies,
         proxy: getPreferredProxy([], undefined),
         proxies: [],
         isLoading: false,
@@ -84,6 +89,7 @@ BunchOfMetadata.args = {
   agentRow: (agent) => (
     <ProxyContext.Provider
       value={{
+        proxyLatencies: MockProxyLatencies,
         proxy: getPreferredProxy([], undefined),
         proxies: [],
         isLoading: false,

diff --git a/site/src/components/Workspace/Workspace.stories.tsx b/site/src/components/Workspace/Workspace.stories.tsx
@@ -7,6 +7,7 @@ import { Workspace, WorkspaceErrors, WorkspaceProps } from "./Workspace"
 import { withReactContext } from "storybook-react-context"
 import EventSource from "eventsourcemock"
 import { ProxyContext, getPreferredProxy } from "contexts/ProxyContext"
+import { MockProxyLatencies } from "../../testHelpers/entities"
 
 export default {
   title: "components/Workspace",
@@ -26,6 +27,7 @@ export default {
 const Template: Story<WorkspaceProps> = (args) => (
   <ProxyContext.Provider
     value={{
+      proxyLatencies: MockProxyLatencies,
       proxy: getPreferredProxy([], undefined),
       proxies: [],
       isLoading: false,

diff --git a/site/src/pages/TerminalPage/TerminalPage.test.tsx b/site/src/pages/TerminalPage/TerminalPage.test.tsx
@@ -4,6 +4,7 @@ import WS from "jest-websocket-mock"
 import { rest } from "msw"
 import {
   MockPrimaryWorkspaceProxy,
+  MockProxyLatencies,
   MockWorkspace,
   MockWorkspaceAgent,
   MockWorkspaceProxies,
@@ -43,6 +44,7 @@ const renderTerminal = () => {
         element={
           <ProxyContext.Provider
             value={{
+              proxyLatencies: MockProxyLatencies,
               proxy: {
                 selectedProxy: MockPrimaryWorkspaceProxy,
                 preferredPathAppURL: "",

diff --git a/site/src/pages/UserSettingsPage/WorkspaceProxyPage/WorspaceProxyView.stories.tsx b/site/src/pages/UserSettingsPage/WorkspaceProxyPage/WorspaceProxyView.stories.tsx
@@ -4,6 +4,7 @@ import {
   MockWorkspaceProxies,
   MockPrimaryWorkspaceProxy,
   MockHealthyWildWorkspaceProxy,
+  MockProxyLatencies,
 } from "testHelpers/entities"
 import {
   WorkspaceProxyView,
@@ -27,6 +28,7 @@ PrimarySelected.args = {
   isLoading: false,
   hasLoaded: true,
   proxies: MockWorkspaceProxies,
+  proxyLatencies: MockProxyLatencies,
   preferredProxy: MockPrimaryWorkspaceProxy,
   onSelect: () => {
     return Promise.resolve()
@@ -38,6 +40,7 @@ Example.args = {
   isLoading: false,
   hasLoaded: true,
   proxies: MockWorkspaceProxies,
+  proxyLatencies: MockProxyLatencies,
   preferredProxy: MockHealthyWildWorkspaceProxy,
   onSelect: () => {
     return Promise.resolve()

diff --git a/site/src/testHelpers/entities.ts b/site/src/testHelpers/entities.ts
@@ -7,6 +7,7 @@ import range from "lodash/range"
 import { Permissions } from "xServices/auth/authXService"
 import { TemplateVersionFiles } from "utils/templateVersion"
 import { FileTree } from "utils/filetree"
+import { ProxyLatencyReport } from "contexts/useProxyLatency"
 
 export const MockOrganization: TypesGen.Organization = {
   id: "fc0774ce-cc9e-48d4-80ae-88f7a4d4a8b0",
@@ -113,6 +114,34 @@ export const MockWorkspaceProxies: TypesGen.Region[] = [
   },
 ]
 
+export const MockProxyLatencies: Record<string, ProxyLatencyReport> = {
+  ...MockWorkspaceProxies.reduce((acc, proxy) => {
+    if (!proxy.healthy) {
+      return acc
+    }
+    acc[proxy.id] = {
+      // Make one of them inaccurate.
+      accurate: proxy.id !== "26e84c16-db24-4636-a62d-aa1a4232b858",
+      // This is a deterministic way to generate a latency to for each proxy.
+      // It will be the same for each run as long as the IDs don't change.
+      latencyMS:
+        (Number(
+          Array.from(proxy.id).reduce(
+            // Multiply each char code by some large prime number to increase the
+            // size of the number and allow use to get some decimal points.
+            (acc, char) => acc + char.charCodeAt(0) * 37,
+            0,
+          ),
+        ) /
+          // Cap at 250ms
+          100) %
+        250,
+      at: new Date(),
+    }
+    return acc
+  }, {} as Record<string, ProxyLatencyReport>),
+}
+
 export const MockBuildInfo: TypesGen.BuildInfoResponse = {
   external_url: "file:///mock-url",
   version: "v99.999.9999+c9cdf14",