site: add tests

mafredri · mafredri · commit ecfe483f7485 · 2025-06-13T19:05:59.000Z
diff --git a/site/src/modules/resources/AgentDevcontainerCard.stories.tsx b/site/src/modules/resources/AgentDevcontainerCard.stories.tsx
@@ -1,10 +1,13 @@
 import type { Meta, StoryObj } from "@storybook/react";
 import type { WorkspaceAgentDevcontainer } from "api/typesGenerated";
 import {
+	MockTemplate,
 	MockWorkspace,
 	MockWorkspaceAgent,
 	MockWorkspaceAgentContainer,
 	MockWorkspaceAgentContainerPorts,
+	MockWorkspaceApp,
+	MockWorkspaceSubAgent,
 } from "testHelpers/entities";
 import { AgentDevcontainerCard } from "./AgentDevcontainerCard";
 
@@ -17,9 +20,9 @@ const MockWorkspaceAgentDevcontainer: WorkspaceAgentDevcontainer = {
 	dirty: false,
 	container: MockWorkspaceAgentContainer,
 	agent: {
-		id: "test-agent-id",
-		name: "test-devcontainer-agent",
-		directory: "/workspace/test",
+		id: MockWorkspaceSubAgent.id,
+		name: MockWorkspaceSubAgent.name,
+		directory: MockWorkspaceSubAgent?.directory ?? "/workspace/test",
 	},
 };
 
@@ -31,6 +34,8 @@ const meta: Meta<typeof AgentDevcontainerCard> = {
 		workspace: MockWorkspace,
 		wildcardHostname: "*.wildcard.hostname",
 		parentAgent: MockWorkspaceAgent,
+		template: MockTemplate,
+		subAgents: [MockWorkspaceSubAgent],
 	},
 };
 
@@ -48,6 +53,7 @@ export const WithPorts: Story = {
 				ports: MockWorkspaceAgentContainerPorts,
 			},
 		},
+		subAgents: [MockWorkspaceSubAgent],
 	},
 };
 
@@ -61,6 +67,7 @@ export const Dirty: Story = {
 				ports: MockWorkspaceAgentContainerPorts,
 			},
 		},
+		subAgents: [MockWorkspaceSubAgent],
 	},
 };
 
@@ -75,5 +82,51 @@ export const Recreating: Story = {
 				ports: MockWorkspaceAgentContainerPorts,
 			},
 		},
+		subAgents: [],
+	},
+};
+
+export const NoSubAgent: Story = {
+	args: {
+		devcontainer: {
+			...MockWorkspaceAgentDevcontainer,
+			agent: undefined,
+		},
+		subAgents: [],
+	},
+};
+
+export const SubAgentConnecting: Story = {
+	args: {
+		devcontainer: {
+			...MockWorkspaceAgentDevcontainer,
+			container: {
+				...MockWorkspaceAgentContainer,
+			},
+		},
+		subAgents: [
+			{
+				...MockWorkspaceSubAgent,
+				status: "connecting",
+			},
+		],
+	},
+};
+
+export const WithAppsAndPorts: Story = {
+	args: {
+		devcontainer: {
+			...MockWorkspaceAgentDevcontainer,
+			container: {
+				...MockWorkspaceAgentContainer,
+				ports: MockWorkspaceAgentContainerPorts,
+			},
+		},
+		subAgents: [
+			{
+				...MockWorkspaceSubAgent,
+				apps: [MockWorkspaceApp],
+			},
+		],
 	},
 };
diff --git a/site/src/pages/WorkspacePage/Workspace.stories.tsx b/site/src/pages/WorkspacePage/Workspace.stories.tsx
@@ -97,7 +97,7 @@ export const RunningWithChildAgent: Story = {
 								lifecycle_state: "ready",
 							},
 							{
-								...Mocks.MockWorkspaceChildAgent,
+								...Mocks.MockWorkspaceSubAgent,
 								lifecycle_state: "ready",
 							},
 						],
diff --git a/site/src/testHelpers/entities.ts b/site/src/testHelpers/entities.ts
@@ -970,38 +970,15 @@ export const MockWorkspaceAgent: TypesGen.WorkspaceAgent = {
 	],
 };
 
-export const MockWorkspaceChildAgent: TypesGen.WorkspaceAgent = {
+export const MockWorkspaceSubAgent: TypesGen.WorkspaceAgent = {
+	...MockWorkspaceAgent,
 	apps: [],
-	architecture: "amd64",
-	created_at: "",
-	environment_variables: {},
-	id: "test-workspace-child-agent",
+	id: "test-workspace-sub-agent",
 	parent_id: "test-workspace-agent",
-	name: "a-workspace-child-agent",
-	operating_system: "linux",
-	resource_id: "",
-	status: "connected",
-	updated_at: "",
-	version: MockBuildInfo.version,
-	api_version: MockBuildInfo.agent_api_version,
-	latency: {
-		"Coder Embedded DERP": {
-			latency_ms: 32.55,
-			preferred: true,
-		},
-	},
-	connection_timeout_seconds: 120,
-	troubleshooting_url: "https://coder.com/troubleshoot",
-	lifecycle_state: "starting",
-	logs_length: 0,
-	logs_overflowed: false,
-	log_sources: [MockWorkspaceAgentLogSource],
+	name: "a-workspace-sub-agent",
+	log_sources: [],
 	scripts: [],
-	startup_script_behavior: "non-blocking",
-	subsystems: ["envbox", "exectrace"],
-	health: {
-		healthy: true,
-	},
+	directory: "/workspace/test",
 	display_apps: [
 		"ssh_helper",
 		"port_forwarding_helper",

Original file line number	Diff line number	Diff line change
`@@ -97,7 +97,7 @@ export const RunningWithChildAgent: Story = {`
`97`	`97`	`lifecycle_state: "ready",`
`98`	`98`	`},`
`99`	`99`	`{`
`100`		`- ...Mocks.MockWorkspaceChildAgent,`
	`100`	`+ ...Mocks.MockWorkspaceSubAgent,`
`101`	`101`	`lifecycle_state: "ready",`
`102`	`102`	`},`
`103`	`103`	`],`