From ea09fdc12cf6fab5971405b40a20f0a37f730873 Mon Sep 17 00:00:00 2001
From: AjTheSpidey <aj.gupta0911@gmail.com>
Date: Thu, 21 May 2026 19:35:26 +0800
Subject: [PATCH] test: expose source selector hooks

---
 src/components/launch/LaunchWindow.tsx   | 1 +
 src/components/launch/SourceSelector.tsx | 3 +++
 2 files changed, 4 insertions(+)

diff --git a/src/components/launch/LaunchWindow.tsx b/src/components/launch/LaunchWindow.tsx
index 570ec2809..66d8d92e5 100644
--- a/src/components/launch/LaunchWindow.tsx
+++ b/src/components/launch/LaunchWindow.tsx
@@ -612,6 +612,7 @@ export function LaunchWindow() {
 
 				{/* Source selector */}
 				<button
+					data-testid="launch-source-selector-button"
 					className={`${hudGroupClasses} h-8 px-2.5 ${styles.electronNoDrag}`}
 					onClick={openSourceSelector}
 					disabled={recording}
diff --git a/src/components/launch/SourceSelector.tsx b/src/components/launch/SourceSelector.tsx
index f3ae98908..65daa9aef 100644
--- a/src/components/launch/SourceSelector.tsx
+++ b/src/components/launch/SourceSelector.tsx
@@ -74,9 +74,12 @@ export function SourceSelector() {
 
 	const renderSourceCard = (source: DesktopSource) => {
 		const isSelected = selectedSource?.id === source.id;
+		const sourceKind = source.id.startsWith("screen:") ? "screen" : "window";
 		return (
 			<div
 				key={source.id}
+				data-testid="source-selector-card"
+				data-source-kind={sourceKind}
 				className={`${styles.sourceCard} ${isSelected ? styles.selected : ""} p-1.5`}
 				onClick={() => handleSourceSelect(source)}
 			>