From 0baf45dc359214853d5f85cdcab505d375a58166 Mon Sep 17 00:00:00 2001
From: Roo Code <roomote@roocode.com>
Date: Fri, 25 Jul 2025 09:27:49 +0000
Subject: [PATCH] fix: update experiments tests to include new
 SHOW_ENHANCE_PROMPT_BUTTON experiment

---
 src/shared/__tests__/experiments.spec.ts | 30 ++++++++++++++++++++++++
 1 file changed, 30 insertions(+)

diff --git a/src/shared/__tests__/experiments.spec.ts b/src/shared/__tests__/experiments.spec.ts
index 4a8f06d62a..b36bf21530 100644
--- a/src/shared/__tests__/experiments.spec.ts
+++ b/src/shared/__tests__/experiments.spec.ts
@@ -23,11 +23,21 @@ describe("experiments", () => {
 		})
 	})
 
+	describe("SHOW_ENHANCE_PROMPT_BUTTON", () => {
+		it("is configured correctly", () => {
+			expect(EXPERIMENT_IDS.SHOW_ENHANCE_PROMPT_BUTTON).toBe("showEnhancePromptButton")
+			expect(experimentConfigsMap.SHOW_ENHANCE_PROMPT_BUTTON).toMatchObject({
+				enabled: true,
+			})
+		})
+	})
+
 	describe("isEnabled", () => {
 		it("returns false when POWER_STEERING experiment is not enabled", () => {
 			const experiments: Record<ExperimentId, boolean> = {
 				powerSteering: false,
 				multiFileApplyDiff: false,
+				showEnhancePromptButton: true,
 			}
 			expect(Experiments.isEnabled(experiments, EXPERIMENT_IDS.POWER_STEERING)).toBe(false)
 		})
@@ -36,6 +46,7 @@ describe("experiments", () => {
 			const experiments: Record<ExperimentId, boolean> = {
 				powerSteering: true,
 				multiFileApplyDiff: false,
+				showEnhancePromptButton: true,
 			}
 			expect(Experiments.isEnabled(experiments, EXPERIMENT_IDS.POWER_STEERING)).toBe(true)
 		})
@@ -44,8 +55,27 @@ describe("experiments", () => {
 			const experiments: Record<ExperimentId, boolean> = {
 				powerSteering: false,
 				multiFileApplyDiff: false,
+				showEnhancePromptButton: true,
 			}
 			expect(Experiments.isEnabled(experiments, EXPERIMENT_IDS.POWER_STEERING)).toBe(false)
 		})
+
+		it("returns true when SHOW_ENHANCE_PROMPT_BUTTON is enabled", () => {
+			const experiments: Record<ExperimentId, boolean> = {
+				powerSteering: false,
+				multiFileApplyDiff: false,
+				showEnhancePromptButton: true,
+			}
+			expect(Experiments.isEnabled(experiments, EXPERIMENT_IDS.SHOW_ENHANCE_PROMPT_BUTTON)).toBe(true)
+		})
+
+		it("returns false when SHOW_ENHANCE_PROMPT_BUTTON is disabled", () => {
+			const experiments: Record<ExperimentId, boolean> = {
+				powerSteering: false,
+				multiFileApplyDiff: false,
+				showEnhancePromptButton: false,
+			}
+			expect(Experiments.isEnabled(experiments, EXPERIMENT_IDS.SHOW_ENHANCE_PROMPT_BUTTON)).toBe(false)
+		})
 	})
 })