|
6 | 6 | import { describe, expect, it } from 'vitest'; |
7 | 7 | import { SessionIndexingPreference } from '../sessionIndexingPreference'; |
8 | 8 |
|
9 | | -function createMockConfigService(storageLevel: string = 'none') { |
| 9 | +function createMockConfigService(opts: { |
| 10 | + localIndexEnabled?: boolean; |
| 11 | + cloudSyncEnabled?: boolean; |
| 12 | + excludeRepositories?: string[]; |
| 13 | +} = {}) { |
| 14 | + const configs: Record<string, unknown> = {}; |
| 15 | + // Map by fullyQualifiedId |
| 16 | + configs['github.copilot.chat.sessionSearch.localIndex.enabled'] = opts.localIndexEnabled ?? false; |
| 17 | + configs['github.copilot.chat.advanced.sessionSearch.cloudSync.enabled'] = opts.cloudSyncEnabled ?? false; |
| 18 | + configs['github.copilot.chat.advanced.sessionSearch.cloudSync.excludeRepositories'] = opts.excludeRepositories ?? []; |
| 19 | + |
10 | 20 | return { |
11 | | - getConfig: (key: unknown) => { |
12 | | - if (typeof key === 'object' && key !== null && 'key' in key) { |
13 | | - return storageLevel; |
14 | | - } |
15 | | - return storageLevel; |
16 | | - }, |
| 21 | + getConfig: (key: { fullyQualifiedId: string }) => configs[key.fullyQualifiedId], |
17 | 22 | } as unknown as import('../../../../platform/configuration/common/configurationService').IConfigurationService; |
18 | 23 | } |
19 | 24 |
|
20 | 25 | describe('SessionIndexingPreference', () => { |
21 | | - it('returns undefined when storage level is none', () => { |
22 | | - const config = createMockConfigService('none'); |
23 | | - const pref = new SessionIndexingPreference(config); |
24 | | - expect(pref.getStorageLevel()).toBeUndefined(); |
25 | | - }); |
26 | | - |
27 | | - it('returns local when configured', () => { |
28 | | - const config = createMockConfigService('local'); |
29 | | - const pref = new SessionIndexingPreference(config); |
| 26 | + it('getStorageLevel returns local when no cloud sync', () => { |
| 27 | + const pref = new SessionIndexingPreference(createMockConfigService({ localIndexEnabled: true })); |
30 | 28 | expect(pref.getStorageLevel()).toBe('local'); |
31 | 29 | }); |
32 | 30 |
|
33 | | - it('returns user when configured', () => { |
34 | | - const config = createMockConfigService('user'); |
35 | | - const pref = new SessionIndexingPreference(config); |
| 31 | + it('getStorageLevel returns user when cloud sync enabled', () => { |
| 32 | + const pref = new SessionIndexingPreference(createMockConfigService({ |
| 33 | + localIndexEnabled: true, |
| 34 | + cloudSyncEnabled: true, |
| 35 | + })); |
36 | 36 | expect(pref.getStorageLevel()).toBe('user'); |
37 | 37 | }); |
38 | 38 |
|
39 | | - it('returns repo_and_user when configured', () => { |
40 | | - const config = createMockConfigService('repo_and_user'); |
41 | | - const pref = new SessionIndexingPreference(config); |
42 | | - expect(pref.getStorageLevel()).toBe('repo_and_user'); |
43 | | - }); |
44 | | - |
45 | | - it('needsPrompt returns true when none', () => { |
46 | | - const config = createMockConfigService('none'); |
47 | | - const pref = new SessionIndexingPreference(config); |
48 | | - expect(pref.needsPrompt()).toBe(true); |
| 39 | + it('getStorageLevel returns local for excluded repo', () => { |
| 40 | + const pref = new SessionIndexingPreference(createMockConfigService({ |
| 41 | + localIndexEnabled: true, |
| 42 | + cloudSyncEnabled: true, |
| 43 | + excludeRepositories: ['my-org/private-repo'], |
| 44 | + })); |
| 45 | + expect(pref.getStorageLevel('my-org/private-repo')).toBe('local'); |
49 | 46 | }); |
50 | 47 |
|
51 | | - it('needsPrompt returns false when configured', () => { |
52 | | - const config = createMockConfigService('user'); |
53 | | - const pref = new SessionIndexingPreference(config); |
54 | | - expect(pref.needsPrompt()).toBe(false); |
| 48 | + it('getStorageLevel returns user for non-excluded repo', () => { |
| 49 | + const pref = new SessionIndexingPreference(createMockConfigService({ |
| 50 | + localIndexEnabled: true, |
| 51 | + cloudSyncEnabled: true, |
| 52 | + excludeRepositories: ['my-org/private-repo'], |
| 53 | + })); |
| 54 | + expect(pref.getStorageLevel('microsoft/vscode')).toBe('user'); |
55 | 55 | }); |
56 | 56 |
|
57 | | - it('hasCloudConsent returns true for user', () => { |
58 | | - const config = createMockConfigService('user'); |
59 | | - const pref = new SessionIndexingPreference(config); |
60 | | - expect(pref.hasCloudConsent()).toBe(true); |
| 57 | + it('hasCloudConsent returns false when cloud sync disabled', () => { |
| 58 | + const pref = new SessionIndexingPreference(createMockConfigService({ cloudSyncEnabled: false })); |
| 59 | + expect(pref.hasCloudConsent()).toBe(false); |
61 | 60 | }); |
62 | 61 |
|
63 | | - it('hasCloudConsent returns true for repo_and_user', () => { |
64 | | - const config = createMockConfigService('repo_and_user'); |
65 | | - const pref = new SessionIndexingPreference(config); |
| 62 | + it('hasCloudConsent returns true when cloud sync enabled', () => { |
| 63 | + const pref = new SessionIndexingPreference(createMockConfigService({ cloudSyncEnabled: true })); |
66 | 64 | expect(pref.hasCloudConsent()).toBe(true); |
67 | 65 | }); |
68 | 66 |
|
69 | | - it('hasCloudConsent returns false for local', () => { |
70 | | - const config = createMockConfigService('local'); |
71 | | - const pref = new SessionIndexingPreference(config); |
72 | | - expect(pref.hasCloudConsent()).toBe(false); |
| 67 | + it('hasCloudConsent returns false for excluded repo', () => { |
| 68 | + const pref = new SessionIndexingPreference(createMockConfigService({ |
| 69 | + cloudSyncEnabled: true, |
| 70 | + excludeRepositories: ['my-org/*'], |
| 71 | + })); |
| 72 | + expect(pref.hasCloudConsent('my-org/secret-repo')).toBe(false); |
73 | 73 | }); |
74 | 74 |
|
75 | | - it('hasCloudConsent returns false for none', () => { |
76 | | - const config = createMockConfigService('none'); |
77 | | - const pref = new SessionIndexingPreference(config); |
78 | | - expect(pref.hasCloudConsent()).toBe(false); |
| 75 | + it('hasCloudConsent supports glob patterns', () => { |
| 76 | + const pref = new SessionIndexingPreference(createMockConfigService({ |
| 77 | + cloudSyncEnabled: true, |
| 78 | + excludeRepositories: ['private-org/*'], |
| 79 | + })); |
| 80 | + expect(pref.hasCloudConsent('private-org/repo-a')).toBe(false); |
| 81 | + expect(pref.hasCloudConsent('private-org/repo-b')).toBe(false); |
| 82 | + expect(pref.hasCloudConsent('public-org/repo-a')).toBe(true); |
79 | 83 | }); |
80 | 84 | }); |
0 commit comments