coder
diff --git a/‎site/src/modules/resources/AgentLogs/useAgentLogs.test.tsx
Lines changed: 0 additions & 115 deletions b/‎site/src/modules/resources/AgentLogs/useAgentLogs.test.tsx
Lines changed: 0 additions & 115 deletions
diff --git a/‎site/src/modules/resources/AgentRow.tsx
Lines changed: 1 addition & 1 deletion b/‎site/src/modules/resources/AgentRow.tsx
Lines changed: 1 addition & 1 deletion
diff --git a/‎site/src/modules/resources/useAgentLogs.test.ts
Lines changed: 54 additions & 0 deletions b/‎site/src/modules/resources/useAgentLogs.test.ts
Lines changed: 54 additions & 0 deletions
diff --git a/‎site/src/modules/resources/AgentLogs/useAgentLogs.ts renamed to ‎site/src/modules/resources/useAgentLogs.ts b/‎site/src/modules/resources/AgentLogs/useAgentLogs.ts renamed to ‎site/src/modules/resources/useAgentLogs.ts
diff --git a/‎site/src/pages/WorkspaceBuildPage/WorkspaceBuildPageView.tsx
Lines changed: 1 addition & 1 deletion b/‎site/src/pages/WorkspaceBuildPage/WorkspaceBuildPageView.tsx
Lines changed: 1 addition & 1 deletion
@@ -30,7 +30,7 @@ import { AgentDevcontainerCard } from "./AgentDevcontainerCard";
 import { AgentLatency } from "./AgentLatency";
 import { AGENT_LOG_LINE_HEIGHT } from "./AgentLogs/AgentLogLine";
 import { AgentLogs } from "./AgentLogs/AgentLogs";
-import { useAgentLogs } from "./AgentLogs/useAgentLogs";
+import { useAgentLogs } from "./useAgentLogs";
 import { AgentMetadata } from "./AgentMetadata";
 import { AgentStatus } from "./AgentStatus";
 import { AgentVersion } from "./AgentVersion";
 
@@ -0,0 +1,54 @@
+import { renderHook } from "@testing-library/react";
+import type { WorkspaceAgentLog } from "api/typesGenerated";
+import WS from "jest-websocket-mock";
+import { MockWorkspaceAgent } from "testHelpers/entities";
+import { useAgentLogs } from "./useAgentLogs";
+
+/**
+ * TODO: WS does not support multiple tests running at once in isolation so we
+ * have one single test that test the most common scenario.
+ * Issue: https://github.com/romgain/jest-websocket-mock/issues/172
+ */
+
+describe("useAgentLogs", () => {
+	afterEach(() => {
+		WS.clean();
+	});
+
+	it("clear logs when disabled to avoid duplicates", async () => {
+		const server = new WS(
+			`ws://localhost/api/v2/workspaceagents/${
+				MockWorkspaceAgent.id
+			}/logs?follow&after=0`,
+		);
+		const { result, rerender } = renderHook(
+			({ enabled }) => useAgentLogs(MockWorkspaceAgent, enabled),
+			{ initialProps: { enabled: true } },
+		);
+		await server.connected;
+
+		// Send 3 logs
+		server.send(JSON.stringify(generateLogs(3)));
+		expect(result.current).toHaveLength(3);
+
+		// Disable the hook
+		rerender({ enabled: false });
+		expect(result.current).toBeUndefined();
+
+		// Enable the hook again
+		rerender({ enabled: true });
+		await server.connected;
+		server.send(JSON.stringify(generateLogs(3)));
+		expect(result.current).toHaveLength(3);
+	});
+});
+
+function generateLogs(count: number): WorkspaceAgentLog[] {
+	return Array.from({ length: count }, (_, i) => ({
+		id: i,
+		created_at: new Date().toISOString(),
+		level: "info",
+		output: `Log ${i}`,
+		source_id: "",
+	}));
+}
@@ -21,7 +21,7 @@ import { useSearchParamsKey } from "hooks/useSearchParamsKey";
 import { BuildAvatar } from "modules/builds/BuildAvatar/BuildAvatar";
 import { DashboardFullPage } from "modules/dashboard/DashboardLayout";
 import { AgentLogs } from "modules/resources/AgentLogs/AgentLogs";
-import { useAgentLogs } from "modules/resources/AgentLogs/useAgentLogs";
+import { useAgentLogs } from "modules/resources/useAgentLogs";
 import {
 	WorkspaceBuildData,
 	WorkspaceBuildDataSkeleton,