@vybestack/llxprt-code 0.1.21 → 0.1.23

package/dist/src/ui/hooks/usePrivacySettings.test.js

@@ -0,0 +1,154 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, beforeEach, vi } from 'vitest';
+import { renderHook, waitFor } from '@testing-library/react';
+import { CodeAssistServer, LoggingContentGenerator, UserTierId, } from '@vybestack/llxprt-code-core';
+import { usePrivacySettings } from './usePrivacySettings.js';
+// Mock the dependencies
+vi.mock('@vybestack/llxprt-code-core', () => {
+    // Mock classes for instanceof checks
+    class MockCodeAssistServer {
+        projectId = 'test-project-id';
+        loadCodeAssist = vi.fn();
+        getCodeAssistGlobalUserSetting = vi.fn();
+        setCodeAssistGlobalUserSetting = vi.fn();
+        constructor(_client, _projectId, _httpOptions, _sessionId, _userTier) { }
+    }
+    class MockLoggingContentGenerator {
+        getWrapped = vi.fn();
+        constructor(_wrapped, _config) { }
+    }
+    return {
+        Config: vi.fn(),
+        CodeAssistServer: MockCodeAssistServer,
+        LoggingContentGenerator: MockLoggingContentGenerator,
+        GeminiClient: vi.fn(),
+        UserTierId: {
+            FREE: 'free-tier',
+            LEGACY: 'legacy-tier',
+            STANDARD: 'standard-tier',
+        },
+    };
+});
+describe('usePrivacySettings', () => {
+    let mockConfig;
+    let mockClient;
+    let mockCodeAssistServer;
+    let mockLoggingContentGenerator;
+    beforeEach(() => {
+        vi.clearAllMocks();
+        // Create mock CodeAssistServer instance
+        mockCodeAssistServer = new CodeAssistServer(null, 'test-project-id');
+        mockCodeAssistServer.loadCodeAssist.mockResolvedValue({
+            currentTier: { id: UserTierId.FREE },
+        });
+        mockCodeAssistServer.getCodeAssistGlobalUserSetting.mockResolvedValue({
+            freeTierDataCollectionOptin: true,
+        });
+        mockCodeAssistServer.setCodeAssistGlobalUserSetting.mockResolvedValue({
+            freeTierDataCollectionOptin: false,
+        });
+        // Create mock LoggingContentGenerator that wraps the CodeAssistServer
+        mockLoggingContentGenerator = new LoggingContentGenerator(mockCodeAssistServer, null);
+        mockLoggingContentGenerator.getWrapped.mockReturnValue(mockCodeAssistServer);
+        // Create mock GeminiClient
+        mockClient = {
+            getContentGenerator: vi.fn().mockReturnValue(mockLoggingContentGenerator),
+        };
+        // Create mock Config
+        mockConfig = {
+            getGeminiClient: vi.fn().mockReturnValue(mockClient),
+        };
+    });
+    it('should handle LoggingContentGenerator wrapper correctly and not throw "Oauth not being used" error', async () => {
+        const { result } = renderHook(() => usePrivacySettings(mockConfig));
+        // Initial state should be loading
+        expect(result.current.privacyState.isLoading).toBe(true);
+        expect(result.current.privacyState.error).toBeUndefined();
+        // Wait for the hook to complete
+        await waitFor(() => {
+            expect(result.current.privacyState.isLoading).toBe(false);
+        });
+        // Should not have the "Oauth not being used" error
+        expect(result.current.privacyState.error).toBeUndefined();
+        expect(result.current.privacyState.isFreeTier).toBe(true);
+        expect(result.current.privacyState.dataCollectionOptIn).toBe(true);
+        // Verify that getWrapped was called to unwrap the LoggingContentGenerator
+        expect(mockLoggingContentGenerator.getWrapped).toHaveBeenCalled();
+    });
+    it('should work with direct CodeAssistServer (no wrapper)', async () => {
+        // Test case where the content generator is directly a CodeAssistServer
+        const directServer = new CodeAssistServer(null, 'test-project-id');
+        directServer.loadCodeAssist.mockResolvedValue({
+            currentTier: { id: UserTierId.FREE },
+        });
+        directServer.getCodeAssistGlobalUserSetting.mockResolvedValue({
+            freeTierDataCollectionOptin: true,
+        });
+        mockClient.getContentGenerator = vi.fn().mockReturnValue(directServer);
+        const { result } = renderHook(() => usePrivacySettings(mockConfig));
+        await waitFor(() => {
+            expect(result.current.privacyState.isLoading).toBe(false);
+        });
+        expect(result.current.privacyState.error).toBeUndefined();
+        expect(result.current.privacyState.isFreeTier).toBe(true);
+        expect(result.current.privacyState.dataCollectionOptIn).toBe(true);
+    });
+    it('should handle paid tier users correctly', async () => {
+        // Mock paid tier response
+        mockCodeAssistServer.loadCodeAssist.mockResolvedValue({
+            currentTier: { id: UserTierId.STANDARD },
+        });
+        const { result } = renderHook(() => usePrivacySettings(mockConfig));
+        await waitFor(() => {
+            expect(result.current.privacyState.isLoading).toBe(false);
+        });
+        expect(result.current.privacyState.error).toBeUndefined();
+        expect(result.current.privacyState.isFreeTier).toBe(false);
+        expect(result.current.privacyState.dataCollectionOptIn).toBeUndefined();
+    });
+    it('should throw error when content generator is not a CodeAssistServer', async () => {
+        // Mock a non-CodeAssistServer content generator
+        const mockOtherGenerator = { someOtherMethod: vi.fn() };
+        mockLoggingContentGenerator.getWrapped.mockReturnValue(mockOtherGenerator);
+        const { result } = renderHook(() => usePrivacySettings(mockConfig));
+        await waitFor(() => {
+            expect(result.current.privacyState.isLoading).toBe(false);
+        });
+        expect(result.current.privacyState.error).toBe('Oauth not being used');
+    });
+    it('should throw error when CodeAssistServer has no projectId', async () => {
+        // Mock CodeAssistServer without projectId
+        const mockServerNoProject = {
+            ...mockCodeAssistServer,
+            projectId: undefined,
+        };
+        mockLoggingContentGenerator.getWrapped.mockReturnValue(mockServerNoProject);
+        const { result } = renderHook(() => usePrivacySettings(mockConfig));
+        await waitFor(() => {
+            expect(result.current.privacyState.isLoading).toBe(false);
+        });
+        expect(result.current.privacyState.error).toBe('Oauth not being used');
+    });
+    it('should update data collection opt-in setting', async () => {
+        const { result } = renderHook(() => usePrivacySettings(mockConfig));
+        // Wait for initial load
+        await waitFor(() => {
+            expect(result.current.privacyState.isLoading).toBe(false);
+        });
+        // Update the setting
+        await result.current.updateDataCollectionOptIn(false);
+        // Wait for update to complete
+        await waitFor(() => {
+            expect(result.current.privacyState.dataCollectionOptIn).toBe(false);
+        });
+        expect(mockCodeAssistServer.setCodeAssistGlobalUserSetting).toHaveBeenCalledWith({
+            cloudaicompanionProject: 'test-project-id',
+            freeTierDataCollectionOptin: false,
+        });
+    });
+});
+//# sourceMappingURL=usePrivacySettings.test.js.map

package/dist/src/ui/hooks/usePrivacySettings.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"usePrivacySettings.test.js","sourceRoot":"","sources":["../../../../src/ui/hooks/usePrivacySettings.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,MAAM,wBAAwB,CAAC;AAC7D,OAAO,EAEL,gBAAgB,EAChB,uBAAuB,EACvB,UAAU,GAGX,MAAM,6BAA6B,CAAC;AAErC,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAE7D,wBAAwB;AACxB,EAAE,CAAC,IAAI,CAAC,6BAA6B,EAAE,GAAG,EAAE;IAC1C,qCAAqC;IACrC,MAAM,oBAAoB;QACxB,SAAS,GAAG,iBAAiB,CAAC;QAC9B,cAAc,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACzB,8BAA8B,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACzC,8BAA8B,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAEzC,YACE,OAAsB,EACtB,UAAmB,EACnB,YAAsC,EACtC,UAAmB,EACnB,SAAsB,IACrB,CAAC;KACL;IAED,MAAM,2BAA2B;QAC/B,UAAU,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAErB,YACE,QAA2B,EAC3B,OAAiC,IAChC,CAAC;KACL;IAED,OAAO;QACL,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;QACf,gBAAgB,EAAE,oBAAoB;QACtC,uBAAuB,EAAE,2BAA2B;QACpD,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE;QACrB,UAAU,EAAE;YACV,IAAI,EAAE,WAAW;YACjB,MAAM,EAAE,aAAa;YACrB,QAAQ,EAAE,eAAe;SAC1B;KACF,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAClC,IAAI,UAAkB,CAAC;IACvB,IAAI,UAAwB,CAAC;IAC7B,IAAI,oBAAsC,CAAC;IAC3C,IAAI,2BAAoD,CAAC;IAEzD,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QAEnB,wCAAwC;QACxC,oBAAoB,GAAG,IAAI,gBAAgB,CACzC,IAA+B,EAC/B,iBAAiB,CACa,CAAC;QAE/B,oBAAoB,CAAC,cACtB,CAAC,iBAAiB,CAAC;YAClB,WAAW,EAAE,EAAE,EAAE,EAAE,UAAU,CAAC,IAAI,EAAE;SACrC,CAAC,CAAC;QAED,oBAAoB,CAAC,8BAGtB,CAAC,iBAAiB,CAAC;YAClB,2BAA2B,EAAE,IAAI;SAClC,CAAC,CAAC;QAED,oBAAoB,CAAC,8BAGtB,CAAC,iBAAiB,CAAC;YAClB,2BAA2B,EAAE,KAAK;SACnC,CAAC,CAAC;QAEH,sEAAsE;QACtE,2BAA2B,GAAG,IAAI,uBAAuB,CACvD,oBAAoB,EACpB,IAAyB,CACY,CAAC;QAEtC,2BAA2B,CAAC,UAC7B,CAAC,eAAe,CAAC,oBAAoB,CAAC,CAAC;QAExC,2BAA2B;QAC3B,UAAU,GAAG;YACX,mBAAmB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,2BAA2B,CAAC;SAC/C,CAAC;QAE7B,qBAAqB;QACrB,UAAU,GAAG;YACX,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,UAAU,CAAC;SAChC,CAAC;IACzB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oGAAoG,EAAE,KAAK,IAAI,EAAE;QAClH,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,kBAAkB,CAAC,UAAU,CAAC,CAAC,CAAC;QAEpE,kCAAkC;QAClC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,aAAa,EAAE,CAAC;QAE1D,gCAAgC;QAChC,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,mDAAmD;QACnD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,aAAa,EAAE,CAAC;QAC1D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC1D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,mBAAmB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAEnE,0EAA0E;QAC1E,MAAM,CAAC,2BAA2B,CAAC,UAAU,CAAC,CAAC,gBAAgB,EAAE,CAAC;IACpE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;QACrE,uEAAuE;QACvE,MAAM,YAAY,GAAG,IAAI,gBAAgB,CACvC,IAA+B,EAC/B,iBAAiB,CACa,CAAC;QAChC,YAAY,CAAC,cAA2C,CAAC,iBAAiB,CACzE;YACE,WAAW,EAAE,EAAE,EAAE,EAAE,UAAU,CAAC,IAAI,EAAE;SACrC,CACF,CAAC;QAEA,YAAY,CAAC,8BACd,CAAC,iBAAiB,CAAC;YAClB,2BAA2B,EAAE,IAAI;SAClC,CAAC,CAAC;QAEH,UAAU,CAAC,mBAAmB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,YAAY,CAAC,CAAC;QAEvE,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,kBAAkB,CAAC,UAAU,CAAC,CAAC,CAAC;QAEpE,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,aAAa,EAAE,CAAC;QAC1D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC1D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,mBAAmB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACrE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;QACvD,0BAA0B;QAExB,oBAAoB,CAAC,cACtB,CAAC,iBAAiB,CAAC;YAClB,WAAW,EAAE,EAAE,EAAE,EAAE,UAAU,CAAC,QAAQ,EAAE;SACzC,CAAC,CAAC;QAEH,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,kBAAkB,CAAC,UAAU,CAAC,CAAC,CAAC;QAEpE,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,aAAa,EAAE,CAAC;QAC1D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC3D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,mBAAmB,CAAC,CAAC,aAAa,EAAE,CAAC;IAC1E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;QACnF,gDAAgD;QAChD,MAAM,kBAAkB,GAAG,EAAE,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC;QAEtD,2BAA2B,CAAC,UAC7B,CAAC,eAAe,CAAC,kBAAkB,CAAC,CAAC;QAEtC,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,kBAAkB,CAAC,UAAU,CAAC,CAAC,CAAC;QAEpE,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;IACzE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;QACzE,0CAA0C;QAC1C,MAAM,mBAAmB,GAAG;YAC1B,GAAG,oBAAoB;YACvB,SAAS,EAAE,SAAS;SACrB,CAAC;QAEA,2BAA2B,CAAC,UAC7B,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC;QAEvC,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,kBAAkB,CAAC,UAAU,CAAC,CAAC,CAAC;QAEpE,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;IACzE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;QAC5D,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,kBAAkB,CAAC,UAAU,CAAC,CAAC,CAAC;QAEpE,wBAAwB;QACxB,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,qBAAqB;QACrB,MAAM,MAAM,CAAC,OAAO,CAAC,yBAAyB,CAAC,KAAK,CAAC,CAAC;QAEtD,8BAA8B;QAC9B,MAAM,OAAO,CAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,mBAAmB,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACtE,CAAC,CAAC,CAAC;QAEH,MAAM,CACJ,oBAAoB,CAAC,8BAA8B,CACpD,CAAC,oBAAoB,CAAC;YACrB,uBAAuB,EAAE,iBAAiB;YAC1C,2BAA2B,EAAE,KAAK;SACnC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/ui/hooks/useToolScheduler.test.d.ts

@@ -0,0 +1,6 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export {};