Fix webview tests after telemetry removal

Author: mrubens

webview-ui/src/components/chat/ShareButton.tsx

@@ -35,6 +35,7 @@ export const ShareButton = ({ item, disabled = false }: ShareButtonProps) => {
 	// Use enhanced cloud upsell hook with auto-open on auth success
 	const {
 		isOpen: connectModalOpen,
+		openUpsell,
 		closeUpsell,
 		handleConnect,
 		isAuthenticated: cloudIsAuthenticated,
@@ -94,7 +95,11 @@ export const ShareButton = ({ item, disabled = false }: ShareButtonProps) => {
 		setShareDropdownOpen(false)
 	}
 
-	const handleShareButtonClick = () => {}
+	const handleShareButtonClick = () => {
+		if (!cloudIsAuthenticated) {
+			openUpsell()
+		}
+	}
 
 	// Determine share button state
 	const getShareButtonState = () => {

webview-ui/src/components/chat/__tests__/ChatView.keyboard-fix.spec.tsx

@@ -10,8 +10,16 @@ import { vscode } from "@src/utils/vscode"
 import ChatView, { ChatViewProps } from "../ChatView"
 
 // Mock vscode API
+vi.mock("@src/utils/vscode", () => ({
+	vscode: {
+		postMessage: vi.fn(),
+	},
+}))
 
 // Mock use-sound hook
+vi.mock("use-sound", () => ({
+	default: vi.fn(() => [vi.fn()]),
+}))
 
 // Mock components
 

webview-ui/src/components/chat/__tests__/ModeSelector.spec.tsx

@@ -6,6 +6,15 @@ import type { Mode } from "@roo/modes"
 
 import { ModeSelector } from "../ModeSelector"
 
+const mockSetHasOpenedModeSelector = vi.hoisted(() => vi.fn())
+
+vi.mock("@/context/ExtensionStateContext", () => ({
+	useExtensionState: () => ({
+		hasOpenedModeSelector: false,
+		setHasOpenedModeSelector: mockSetHasOpenedModeSelector,
+	}),
+}))
+
 // Create a variable to control what getAllModes returns.
 let mockModes: ModeConfig[] = []
 

webview-ui/src/components/chat/__tests__/ShareButton.spec.tsx

@@ -2,17 +2,30 @@ import { useTranslation } from "react-i18next"
 
 import { render, screen, fireEvent, waitFor } from "@/utils/test-utils"
 import { vscode } from "@/utils/vscode"
+import { useExtensionState } from "@/context/ExtensionStateContext"
 
 import { ShareButton } from "../ShareButton"
 
 // Mock the vscode utility
+vi.mock("@/utils/vscode", () => ({
+	vscode: {
+		postMessage: vi.fn(),
+	},
+}))
 
 // Mock react-i18next
+vi.mock("react-i18next", () => ({
+	useTranslation: vi.fn(),
+}))
 
 // Mock the extension state context
+vi.mock("@/context/ExtensionStateContext", () => ({
+	useExtensionState: vi.fn(),
+}))
 
 const mockUseTranslation = vi.mocked(useTranslation)
 const mockVscode = vi.mocked(vscode)
+const mockUseExtensionState = vi.mocked(useExtensionState)
 
 describe("ShareButton", () => {
 	const mockT = vi.fn((key: string) => key)
@@ -29,6 +42,15 @@ describe("ShareButton", () => {
 	beforeEach(() => {
 		vi.clearAllMocks()
 
+		mockUseExtensionState.mockReturnValue({
+			cloudIsAuthenticated: true,
+			sharingEnabled: true,
+			publicSharingEnabled: true,
+			cloudUserInfo: {
+				organizationName: "Test Organization",
+			},
+		} as any)
+
 		mockUseTranslation.mockReturnValue({
 			t: mockT,
 			i18n: {} as any,