Roo-Code/src/shared/__tests__/experiments-preventFocusDisruption.spec.ts

import { EXPERIMENT_IDS, experimentConfigsMap, experimentDefault, experiments } from "../experiments"

describe("PREVENT_FOCUS_DISRUPTION experiment", () => {
	it("should include PREVENT_FOCUS_DISRUPTION in EXPERIMENT_IDS", () => {
		expect(EXPERIMENT_IDS.PREVENT_FOCUS_DISRUPTION).toBe("preventFocusDisruption")
	})

	it("should have PREVENT_FOCUS_DISRUPTION in experimentConfigsMap", () => {
		expect(experimentConfigsMap.PREVENT_FOCUS_DISRUPTION).toBeDefined()
		expect(experimentConfigsMap.PREVENT_FOCUS_DISRUPTION.enabled).toBe(false)
	})

	it("should have PREVENT_FOCUS_DISRUPTION in experimentDefault", () => {
		expect(experimentDefault.preventFocusDisruption).toBe(false)
	})

	it("should correctly check if PREVENT_FOCUS_DISRUPTION is enabled", () => {
		// Test when experiment is disabled (default)
		const disabledConfig = { preventFocusDisruption: false }
		expect(experiments.isEnabled(disabledConfig, EXPERIMENT_IDS.PREVENT_FOCUS_DISRUPTION)).toBe(false)

		// Test when experiment is enabled
		const enabledConfig = { preventFocusDisruption: true }
		expect(experiments.isEnabled(enabledConfig, EXPERIMENT_IDS.PREVENT_FOCUS_DISRUPTION)).toBe(true)

		// Test when experiment is not in config (should use default)
		const emptyConfig = {}
		expect(experiments.isEnabled(emptyConfig, EXPERIMENT_IDS.PREVENT_FOCUS_DISRUPTION)).toBe(false)
	})
})