cleanup

mafredri · mafredri · commit 8f12d26a2820 · 2025-06-13T20:06:03.000Z
diff --git a/site/src/modules/resources/AgentDevcontainerCard.stories.tsx b/site/src/modules/resources/AgentDevcontainerCard.stories.tsx
@@ -10,6 +10,7 @@ import {
 	MockWorkspaceAgent,
 	MockWorkspaceAgentContainer,
 	MockWorkspaceAgentContainerPorts,
+	MockWorkspaceAgentDevcontainer,
 	MockWorkspaceApp,
 	MockWorkspaceProxies,
 	MockWorkspaceSubAgent,
@@ -20,21 +21,6 @@ import {
 } from "testHelpers/storybook";
 import { AgentDevcontainerCard } from "./AgentDevcontainerCard";
 
-const MockWorkspaceAgentDevcontainer: WorkspaceAgentDevcontainer = {
-	id: "test-devcontainer-id",
-	name: "test-devcontainer",
-	workspace_folder: "/workspace/test",
-	config_path: "/workspace/test/.devcontainer/devcontainer.json",
-	status: "running",
-	dirty: false,
-	container: MockWorkspaceAgentContainer,
-	agent: {
-		id: MockWorkspaceSubAgent.id,
-		name: MockWorkspaceSubAgent.name,
-		directory: MockWorkspaceSubAgent?.directory ?? "/workspace/test",
-	},
-};
-
 const meta: Meta<typeof AgentDevcontainerCard> = {
 	title: "modules/resources/AgentDevcontainerCard",
 	component: AgentDevcontainerCard,
@@ -80,10 +66,6 @@ export const Dirty: Story = {
 		devcontainer: {
 			...MockWorkspaceAgentDevcontainer,
 			dirty: true,
-			container: {
-				...MockWorkspaceAgentContainer,
-				ports: MockWorkspaceAgentContainerPorts,
-			},
 		},
 	},
 };
@@ -94,10 +76,7 @@ export const Recreating: Story = {
 			...MockWorkspaceAgentDevcontainer,
 			dirty: true,
 			status: "starting",
-			container: {
-				...MockWorkspaceAgentContainer,
-				ports: MockWorkspaceAgentContainerPorts,
-			},
+			container: undefined,
 		},
 		subAgents: [],
 	},
@@ -115,12 +94,6 @@ export const NoSubAgent: Story = {
 
 export const SubAgentConnecting: Story = {
 	args: {
-		devcontainer: {
-			...MockWorkspaceAgentDevcontainer,
-			container: {
-				...MockWorkspaceAgentContainer,
-			},
-		},
 		subAgents: [
 			{
 				...MockWorkspaceSubAgent,
@@ -148,7 +121,7 @@ export const WithAppsAndPorts: Story = {
 	},
 };
 
-export const ShowingPortForward: Story = {
+export const WithPortForwarding: Story = {
 	decorators: [
 		withProxyProvider({
 			proxy: getPreferredProxy(MockWorkspaceProxies, MockPrimaryWorkspaceProxy),
diff --git a/site/src/testHelpers/entities.ts b/site/src/testHelpers/entities.ts
@@ -4369,6 +4369,22 @@ export const MockWorkspaceAgentContainer: TypesGen.WorkspaceAgentContainer = {
 	},
 };
 
+export const MockWorkspaceAgentDevcontainer: TypesGen.WorkspaceAgentDevcontainer =
+	{
+		id: "test-devcontainer-id",
+		name: "test-devcontainer",
+		workspace_folder: "/workspace/test",
+		config_path: "/workspace/test/.devcontainer/devcontainer.json",
+		status: "running",
+		dirty: false,
+		container: MockWorkspaceAgentContainer,
+		agent: {
+			id: MockWorkspaceSubAgent.id,
+			name: MockWorkspaceSubAgent.name,
+			directory: MockWorkspaceSubAgent?.directory ?? "/workspace/test",
+		},
+	};
+
 export const MockWorkspaceAppStatuses: TypesGen.WorkspaceAppStatus[] = [
 	{
 		// This is the latest status chronologically (15:04:38)
