diff --git a/frontend/__tests__/components/features/chat/plan-preview.test.tsx b/frontend/__tests__/components/features/chat/plan-preview.test.tsx index 00d0466919..3e53ae7b09 100644 --- a/frontend/__tests__/components/features/chat/plan-preview.test.tsx +++ b/frontend/__tests__/components/features/chat/plan-preview.test.tsx @@ -92,19 +92,21 @@ describe("PlanPreview", () => { }); it("should render nothing when planContent is null", () => { - renderPlanPreview(); + // Arrange & Act + const { container } = renderPlanPreview(); - const contentDiv = screen.getByTestId("plan-preview-content"); - expect(contentDiv).toBeInTheDocument(); - expect(contentDiv.textContent?.trim() || "").toBe(""); + // Assert + expect(container.firstChild).toBeNull(); }); it("should render nothing when planContent is undefined", () => { - renderPlanPreview(); + // Arrange & Act + const { container } = renderPlanPreview( + , + ); - const contentDiv = screen.getByTestId("plan-preview-content"); - expect(contentDiv).toBeInTheDocument(); - expect(contentDiv.textContent?.trim() || "").toBe(""); + // Assert + expect(container.firstChild).toBeNull(); }); it("should render markdown content when planContent is provided", () => { diff --git a/frontend/src/components/features/chat/plan-preview.tsx b/frontend/src/components/features/chat/plan-preview.tsx index 44cbedace3..9a69d8ad64 100644 --- a/frontend/src/components/features/chat/plan-preview.tsx +++ b/frontend/src/components/features/chat/plan-preview.tsx @@ -65,7 +65,7 @@ export function PlanPreview({ return `${planContent.slice(0, MAX_CONTENT_LENGTH)}...`; }, [planContent]); - if (!shouldUsePlanningAgent) { + if (!shouldUsePlanningAgent || !planContent) { return null; }