import { EMPTY, of, Subject } from 'rxjs' import sinon from 'sinon' import { describe, expect, test } from 'vitest' import { getGraphQLClient as getGraphQLClientBase, type SuccessGraphQLResult } from '@sourcegraph/http-client' import { cache } from '../../backend/apolloCache' import type { PlatformContext } from '../../platform/context' import type { SettingsCascade } from '../../settings/settings' import { noOpTelemetryRecorder } from '../../telemetry' import type { FlatExtensionHostAPI } from '../contract' import { pretendRemote } from '../util' import { initMainThreadAPI } from './mainthread-api' import type { SettingsEdit } from './services/settings' describe('MainThreadAPI', () => { // TODO(tj): commands, notifications const getGraphQLClient = () => getGraphQLClientBase({ cache }) describe('graphQL', () => { test('PlatformContext#requestGraphQL is called with the correct arguments', async () => { const requestGraphQL = sinon.spy(_options => of({ data: null, errors: [] })) const platformContext: Pick< PlatformContext, | 'updateSettings' | 'settings' | 'getGraphQLClient' | 'requestGraphQL' | 'clientApplication' | 'telemetryRecorder' > = { settings: EMPTY, getGraphQLClient, updateSettings: () => Promise.resolve(), requestGraphQL, clientApplication: 'other', telemetryRecorder: noOpTelemetryRecorder, } const { api } = initMainThreadAPI(pretendRemote({}), platformContext) const gqlRequestOptions = { request: 'search', variables: { query: 'test', }, mightContainPrivateInfo: true, } await api.requestGraphQL(gqlRequestOptions.request, gqlRequestOptions.variables) sinon.assert.calledOnceWithExactly(requestGraphQL, gqlRequestOptions) }) test('extension host receives the value returned by PlatformContext#requestGraphQL', async () => { const graphQLResult: SuccessGraphQLResult = { data: { search: { results: { results: [] } } }, errors: undefined, } const requestGraphQL = sinon.spy(_options => of(graphQLResult)) const platformContext: Pick< PlatformContext, | 'updateSettings' | 'settings' | 'getGraphQLClient' | 'requestGraphQL' | 'clientApplication' | 'telemetryRecorder' > = { settings: EMPTY, getGraphQLClient, updateSettings: () => Promise.resolve(), requestGraphQL, clientApplication: 'other', telemetryRecorder: noOpTelemetryRecorder, } const { api } = initMainThreadAPI(pretendRemote({}), platformContext) const result = await api.requestGraphQL('search', {}) expect(result).toStrictEqual(graphQLResult) }) }) describe('configuration', () => { test('changeConfiguration goes to platform with the last settings subject', async () => { let calledWith: Parameters | undefined const updateSettings: PlatformContext['updateSettings'] = (...args) => { calledWith = args return Promise.resolve() } const platformContext: Pick< PlatformContext, | 'updateSettings' | 'settings' | 'requestGraphQL' | 'getGraphQLClient' | 'clientApplication' | 'telemetryRecorder' > = { settings: of({ subjects: [ { settings: null, lastID: null, subject: { id: 'id1', __typename: 'DefaultSettings', viewerCanAdminister: true, settingsURL: null, latestSettings: null, }, }, { settings: null, lastID: null, subject: { id: 'id2', __typename: 'DefaultSettings', viewerCanAdminister: true, settingsURL: null, latestSettings: null, }, }, ], final: { a: 'value' }, }), updateSettings, getGraphQLClient, requestGraphQL: () => EMPTY, clientApplication: 'other', telemetryRecorder: noOpTelemetryRecorder, } const { api } = initMainThreadAPI( pretendRemote({ syncSettingsData: () => {} }), platformContext ) const edit: SettingsEdit = { path: ['a'], value: 'newVal' } await api.applySettingsEdit(edit) expect(calledWith).toEqual(['id2', edit] as Parameters) }) test('changes of settings from platform propagated to the ext host', () => { const values: SettingsCascade<{ a: string }>[] = [ { subjects: [], // this is valid actually even though it shouldn't final: { a: 'one' }, }, { subjects: null as any, // invalid and should be ignored final: { a: 'invalid two' }, }, { subjects: [], // one more valid case final: { a: 'three' }, }, ] const platformContext: Pick< PlatformContext, | 'updateSettings' | 'settings' | 'getGraphQLClient' | 'requestGraphQL' | 'clientApplication' | 'telemetryRecorder' > = { getGraphQLClient, settings: of(...values), updateSettings: () => Promise.resolve(), requestGraphQL: () => EMPTY, clientApplication: 'other', telemetryRecorder: noOpTelemetryRecorder, } const passedToExtensionHost: SettingsCascade