test(site): Update agent mocks

mafredri · mafredri · commit bd1a87f584c8 · 2023-01-23T14:32:11.000Z
diff --git a/site/src/testHelpers/entities.ts b/site/src/testHelpers/entities.ts
@@ -301,6 +301,7 @@ export const MockWorkspaceAgent: TypesGen.WorkspaceAgent = {
   },
   connection_timeout_seconds: 120,
   troubleshooting_url: "https://coder.com/troubleshoot",
+  lifecycle_state: "starting",
 }
 
 export const MockWorkspaceAgentDisconnected: TypesGen.WorkspaceAgent = {
@@ -310,6 +311,7 @@ export const MockWorkspaceAgentDisconnected: TypesGen.WorkspaceAgent = {
   status: "disconnected",
   version: "",
   latency: {},
+  lifecycle_state: "ready",
 }
 
 export const MockWorkspaceAgentOutdated: TypesGen.WorkspaceAgent = {
@@ -333,6 +335,7 @@ export const MockWorkspaceAgentOutdated: TypesGen.WorkspaceAgent = {
       latency_ms: 221.66,
     },
   },
+  lifecycle_state: "ready",
 }
 
 export const MockWorkspaceAgentConnecting: TypesGen.WorkspaceAgent = {
@@ -342,6 +345,7 @@ export const MockWorkspaceAgentConnecting: TypesGen.WorkspaceAgent = {
   status: "connecting",
   version: "",
   latency: {},
+  lifecycle_state: "created",
 }
 
 export const MockWorkspaceAgentTimeout: TypesGen.WorkspaceAgent = {
@@ -351,6 +355,28 @@ export const MockWorkspaceAgentTimeout: TypesGen.WorkspaceAgent = {
   status: "timeout",
   version: "",
   latency: {},
+  lifecycle_state: "created",
+}
+
+export const MockWorkspaceAgentStarting: TypesGen.WorkspaceAgent = {
+  ...MockWorkspaceAgent,
+  id: "test-workspace-agent-starting",
+  name: "a-starting-workspace-agent",
+  lifecycle_state: "starting",
+}
+
+export const MockWorkspaceAgentStartTimeout: TypesGen.WorkspaceAgent = {
+  ...MockWorkspaceAgent,
+  id: "test-workspace-agent-start-timeout",
+  name: "a-workspace-agent-timed-out-while-running-startup-script",
+  lifecycle_state: "start_timeout",
+}
+
+export const MockWorkspaceAgentStartError: TypesGen.WorkspaceAgent = {
+  ...MockWorkspaceAgent,
+  id: "test-workspace-agent-start-error",
+  name: "a-workspace-agent-errored-while-running-startup-script",
+  lifecycle_state: "start_error",
 }
 
 export const MockWorkspaceResource: TypesGen.WorkspaceResource = {
