Function loadBanner

src/interactive-window/shiftEnterBanner.unit.test.ts:108–168

(
    config: typemoq.IMock<IConfigurationService>,
    stateEnabled: boolean,
    configCalled: boolean,
    questionResponse: string
)

// Create a test banner with the given settings
function loadBanner(
    config: typemoq.IMock<IConfigurationService>,
    stateEnabled: boolean,
    configCalled: boolean,
    questionResponse: string
): InteractiveShiftEnterBanner {
    // Config persist state
    const persistService: typemoq.IMock<IPersistentStateFactory> = typemoq.Mock.ofType<IPersistentStateFactory>();
    const enabledState: typemoq.IMock<IPersistentState<boolean>> = typemoq.Mock.ofType<IPersistentState<boolean>>();
    enabledState.setup((a) => a.value).returns(() => stateEnabled);
    enabledState.setup((a) => a.updateValue(typemoq.It.isAny())).returns(() => Promise.resolve());
    persistService
        .setup((a) =>
            a.createGlobalPersistentState(
                typemoq.It.isValue(InteractiveShiftEnterStateKeys.ShowBanner),
                typemoq.It.isValue(true)
            )
        )
        .returns(() => {
            return enabledState.object;
        });
    persistService
        .setup((a) =>
            a.createGlobalPersistentState(
                typemoq.It.isValue(InteractiveShiftEnterStateKeys.ShowBanner),
                typemoq.It.isValue(false)
            )
        )
        .returns(() => {
            return enabledState.object;
        });

    // Config settings
    const dataScienceSettings = typemoq.Mock.ofType<IWatchableJupyterSettings>();
    dataScienceSettings.setup((d) => d.sendSelectionToInteractiveWindow).returns(() => false);
    config.setup((c) => c.getSettings(typemoq.It.isAny())).returns(() => dataScienceSettings.object);

    // Config AppShell - mock showInformationMessage with exact expected arguments
    when(
        mockedVSCodeNamespaces.window.showInformationMessage(
            'Would you like shift-enter to send code to the new Interactive Window experience?',
            'Yes',
            'No'
        )
    ).thenReturn(Promise.resolve(questionResponse) as any);

    // Config settings
    config
        .setup((c) =>
            c.updateSetting(
                typemoq.It.isValue('interactiveWindow.textEditor.executeSelection'),
                typemoq.It.isAny(),
                typemoq.It.isAny(),
                typemoq.It.isAny()
            )
        )
        .returns(() => Promise.resolve())
        .verifiable(configCalled ? typemoq.Times.once() : typemoq.Times.never());