MediaMetz/gemini-cli
1
0
Fork 0
mirror of https://github.com/google-gemini/gemini-cli.git synced 2026-04-20 02:00:40 -07:00
Code Issues Packages Projects Releases Wiki Activity

test(cli): refactor tests for async render utilities (#23252)

Browse Source
This commit is contained in:
Tommaso Sciortino
2026-03-20 20:08:29 +00:00
committed by GitHub
parent 86a3a913b5
commit 6c78eb7a39
198 changed files with 3592 additions and 4802 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
packages/cli/src/ui/hooks/usePrivacySettings.test.tsx
View File

@@ -33,13 +33,13 @@ describe('usePrivacySettings', () => {
vi.clearAllMocks();
});
const renderPrivacySettingsHook = () => {
const renderPrivacySettingsHook = async () => {
let hookResult: ReturnType<typeof usePrivacySettings>;
function TestComponent() {
hookResult = usePrivacySettings(mockConfig);
return null;
}
render(<TestComponent />);
await render(<TestComponent />);
return {
result: {
get current() {
@@ -52,7 +52,7 @@ describe('usePrivacySettings', () => {
it('should throw error when content generator is not a CodeAssistServer', async () => {
vi.mocked(getCodeAssistServer).mockReturnValue(undefined);
const { result } = renderPrivacySettingsHook();
const { result } = await act(async () => renderPrivacySettingsHook());
await waitFor(() => {
expect(result.current.privacyState.isLoading).toBe(false);
@@ -68,7 +68,7 @@ describe('usePrivacySettings', () => {
userTier: UserTierId.STANDARD,
} as unknown as CodeAssistServer);
const { result } = renderPrivacySettingsHook();
const { result } = await act(async () => renderPrivacySettingsHook());
await waitFor(() => {
expect(result.current.privacyState.isLoading).toBe(false);
@@ -84,7 +84,7 @@ describe('usePrivacySettings', () => {
userTier: UserTierId.FREE,
} as unknown as CodeAssistServer);
const { result } = renderPrivacySettingsHook();
const { result } = await act(async () => renderPrivacySettingsHook());
await waitFor(() => {
expect(result.current.privacyState.isLoading).toBe(false);
@@ -96,11 +96,15 @@ describe('usePrivacySettings', () => {
});
it('should update data collection opt-in setting', async () => {
let deferredGet: { resolve: (val: unknown) => void };
const mockCodeAssistServer = {
projectId: 'test-project-id',
getCodeAssistGlobalUserSetting: vi.fn().mockResolvedValue({
freeTierDataCollectionOptin: true,
}),
getCodeAssistGlobalUserSetting: vi.fn().mockImplementation(
() =>
new Promise((resolve) => {
deferredGet = { resolve };
}),
),
setCodeAssistGlobalUserSetting: vi.fn().mockResolvedValue({
freeTierDataCollectionOptin: false,
}),
@@ -108,9 +112,19 @@ describe('usePrivacySettings', () => {
} as unknown as CodeAssistServer;
vi.mocked(getCodeAssistServer).mockReturnValue(mockCodeAssistServer);
const { result } = renderPrivacySettingsHook();
const { result } = await act(async () => renderPrivacySettingsHook());
// Wait for initial load
// Initially loading
expect(result.current.privacyState.isLoading).toBe(true);
// Finish initial load
await act(async () => {
deferredGet.resolve({
freeTierDataCollectionOptin: true,
});
});
// Wait for initial load to process
await waitFor(() => {
expect(result.current.privacyState.isLoading).toBe(false);
});