mirror of
https://github.com/ckaczor/azuredatastudio.git
synced 2026-01-19 01:25:36 -05:00
Add tests for statusmodeselect (#10235)
* add tests for statusmodeselect * add another test
This commit is contained in:
Anthony Dresser
@@ -0,0 +1,165 @@
|
||||
/*---------------------------------------------------------------------------------------------
|
||||
* Copyright (c) Microsoft Corporation. All rights reserved.
|
||||
* Licensed under the Source EULA. See License.txt in the project root for license information.
|
||||
*--------------------------------------------------------------------------------------------*/
|
||||
|
||||
import * as assert from 'assert';
|
||||
import * as sinon from 'sinon';
|
||||
import { setMode } from 'sql/workbench/browser/parts/editor/editorStatusModeSelect';
|
||||
import { IDisposable, dispose } from 'vs/base/common/lifecycle';
|
||||
import { QueryEditorLanguageAssociation } from 'sql/workbench/contrib/query/browser/queryInputFactory';
|
||||
import { NotebookEditorInputAssociation } from 'sql/workbench/contrib/notebook/browser/models/nodebookInputFactory';
|
||||
import { workbenchInstantiationService } from 'sql/workbench/test/workbenchTestServices';
|
||||
import { INotebookService } from 'sql/workbench/services/notebook/browser/notebookService';
|
||||
import { IEditorService } from 'vs/workbench/services/editor/common/editorService';
|
||||
import { IQueryEditorService } from 'sql/workbench/services/queryEditor/common/queryEditorService';
|
||||
import { Registry } from 'vs/platform/registry/common/platform';
|
||||
import { ILanguageAssociationRegistry, Extensions as LanguageAssociationExtensions } from 'sql/workbench/services/languageAssociation/common/languageAssociation';
|
||||
import { TestQueryEditorService } from 'sql/workbench/services/queryEditor/test/common/testQueryEditorService';
|
||||
import { TestEditorService } from 'vs/workbench/test/browser/workbenchTestServices';
|
||||
import { NotebookServiceStub } from 'sql/workbench/contrib/notebook/test/stubs';
|
||||
import { IInstantiationService } from 'vs/platform/instantiation/common/instantiation';
|
||||
import { IUntitledTextResourceEditorInput, EditorInput, IVisibleEditorPane } from 'vs/workbench/common/editor';
|
||||
import { UntitledTextEditorInput } from 'vs/workbench/services/untitled/common/untitledTextEditorInput';
|
||||
import { IUntitledTextEditorService } from 'vs/workbench/services/untitled/common/untitledTextEditorService';
|
||||
import { FileEditorInput } from 'vs/workbench/contrib/files/common/editors/fileEditorInput';
|
||||
import { URI } from 'vs/base/common/uri';
|
||||
import { FileQueryEditorInput } from 'sql/workbench/contrib/query/common/fileQueryEditorInput';
|
||||
import { QueryResultsInput } from 'sql/workbench/common/editor/query/queryResultsInput';
|
||||
import { ICodeEditor } from 'vs/editor/browser/editorBrowser';
|
||||
import { EditorType } from 'vs/editor/common/editorCommon';
|
||||
import { INotificationService } from 'vs/platform/notification/common/notification';
|
||||
import { TestInstantiationService } from 'vs/platform/instantiation/test/common/instantiationServiceMock';
|
||||
import { TestNotificationService } from 'vs/platform/notification/test/common/testNotificationService';
|
||||
|
||||
const languageAssociations = Registry.as<ILanguageAssociationRegistry>(LanguageAssociationExtensions.LanguageAssociations);
|
||||
|
||||
suite('set mode', () => {
|
||||
let disposables: IDisposable[] = [];
|
||||
|
||||
setup(() => {
|
||||
disposables.push(languageAssociations.registerLanguageAssociation(QueryEditorLanguageAssociation.languages, QueryEditorLanguageAssociation, QueryEditorLanguageAssociation.isDefault));
|
||||
disposables.push(languageAssociations.registerLanguageAssociation(NotebookEditorInputAssociation.languages, NotebookEditorInputAssociation));
|
||||
const instantiationService = workbenchInstantiationService();
|
||||
instantiationService.stub(INotebookService, new NotebookServiceStub());
|
||||
const editorService = new MockEditorService(instantiationService);
|
||||
instantiationService.stub(IEditorService, editorService);
|
||||
instantiationService.stub(IQueryEditorService, instantiationService.createInstance(TestQueryEditorService));
|
||||
instantiationService.invokeFunction(accessor => {
|
||||
languageAssociations.start(accessor);
|
||||
});
|
||||
});
|
||||
|
||||
teardown(() => {
|
||||
disposables = dispose(disposables);
|
||||
});
|
||||
|
||||
test('does leave editor alone and change mode when changed from plaintext to json', async () => {
|
||||
const instantiationService = workbenchInstantiationService();
|
||||
const editorService = new MockEditorService(instantiationService, 'plaintext');
|
||||
instantiationService.stub(IEditorService, editorService);
|
||||
const replaceEditorStub = sinon.stub(editorService, 'replaceEditors', () => Promise.resolve());
|
||||
const stub = sinon.stub();
|
||||
const modeSupport = { setMode: stub };
|
||||
const activeEditor = instantiationService.createInstance(FileEditorInput, URI.file('/test/file.txt'), undefined, 'plaintext');
|
||||
await instantiationService.invokeFunction(setMode, modeSupport, activeEditor, 'json');
|
||||
assert(stub.calledOnce);
|
||||
assert(stub.calledWithExactly('json'));
|
||||
assert(replaceEditorStub.notCalled);
|
||||
});
|
||||
|
||||
test('does replace editor and set mode correctly when changed from sql to notebooks', async () => {
|
||||
const instantiationService = workbenchInstantiationService();
|
||||
const editorService = new MockEditorService(instantiationService, 'sql');
|
||||
instantiationService.stub(IEditorService, editorService);
|
||||
const stub = sinon.stub();
|
||||
const modeSupport = { setMode: stub };
|
||||
const uri = URI.file('/test/file.sql');
|
||||
const textInput = instantiationService.createInstance(FileEditorInput, uri, undefined, 'sql');
|
||||
const activeEditor = instantiationService.createInstance(FileQueryEditorInput, '', textInput, instantiationService.createInstance(QueryResultsInput, uri.toString()));
|
||||
await instantiationService.invokeFunction(setMode, modeSupport, activeEditor, 'notebooks');
|
||||
assert(stub.calledOnce);
|
||||
assert(stub.calledWithExactly('notebooks'));
|
||||
});
|
||||
|
||||
test('does replace editor and set mode correctly when changed from sql to plaintext', async () => {
|
||||
const instantiationService = workbenchInstantiationService();
|
||||
const editorService = new MockEditorService(instantiationService, 'sql');
|
||||
instantiationService.stub(IEditorService, editorService);
|
||||
const stub = sinon.stub();
|
||||
const modeSupport = { setMode: stub };
|
||||
const uri = URI.file('/test/file.sql');
|
||||
const textInput = instantiationService.createInstance(FileEditorInput, uri, undefined, 'sql');
|
||||
const activeEditor = instantiationService.createInstance(FileQueryEditorInput, '', textInput, instantiationService.createInstance(QueryResultsInput, uri.toString()));
|
||||
await instantiationService.invokeFunction(setMode, modeSupport, activeEditor, 'plaintext');
|
||||
assert(stub.calledOnce);
|
||||
assert(stub.calledWithExactly('plaintext'));
|
||||
});
|
||||
|
||||
test('does replace editor and set mode correctly when changed from plaintext to sql', async () => {
|
||||
const instantiationService = workbenchInstantiationService();
|
||||
const editorService = new MockEditorService(instantiationService, 'plaintext');
|
||||
instantiationService.stub(IEditorService, editorService);
|
||||
const stub = sinon.stub();
|
||||
const modeSupport = { setMode: stub };
|
||||
const activeEditor = instantiationService.createInstance(FileEditorInput, URI.file('/test/file.txt'), undefined, 'plaintext');
|
||||
await instantiationService.invokeFunction(setMode, modeSupport, activeEditor, 'sql');
|
||||
assert(stub.calledOnce);
|
||||
assert(stub.calledWithExactly('sql'));
|
||||
});
|
||||
|
||||
test('does show error if mode change happens on a dirty file', async () => {
|
||||
const instantiationService = workbenchInstantiationService();
|
||||
const editorService = new MockEditorService(instantiationService, 'plaintext');
|
||||
const errorStub = sinon.stub();
|
||||
instantiationService.stub(IEditorService, editorService);
|
||||
instantiationService.stub(INotificationService, TestNotificationService);
|
||||
(instantiationService as TestInstantiationService).stub(INotificationService, 'error', errorStub);
|
||||
const stub = sinon.stub();
|
||||
const modeSupport = { setMode: stub };
|
||||
const activeEditor = instantiationService.createInstance(FileEditorInput, URI.file('/test/file.txt'), undefined, 'plaintext');
|
||||
sinon.stub(activeEditor, 'isDirty', () => true);
|
||||
await instantiationService.invokeFunction(setMode, modeSupport, activeEditor, 'sql');
|
||||
assert(stub.notCalled);
|
||||
assert(errorStub.calledOnce);
|
||||
});
|
||||
});
|
||||
|
||||
|
||||
class MockEditorService extends TestEditorService {
|
||||
|
||||
constructor(private readonly instantiationService: IInstantiationService, private readonly mode?: string) {
|
||||
super();
|
||||
}
|
||||
|
||||
activeEditorPane: IVisibleEditorPane = <any>{
|
||||
group: {}
|
||||
};
|
||||
|
||||
activeTextEditorControl: ICodeEditor = <any>{
|
||||
getModel: () => {
|
||||
return <any>{
|
||||
getLanguageIdentifier: () => {
|
||||
return { language: this.mode };
|
||||
}
|
||||
};
|
||||
},
|
||||
getEditorType: () => EditorType.ICodeEditor
|
||||
};
|
||||
|
||||
openEditor(_editor: any, _options?: any, _group?: any): Promise<any> {
|
||||
return Promise.resolve(_editor);
|
||||
}
|
||||
|
||||
createEditorInput(_input: IUntitledTextResourceEditorInput): EditorInput {
|
||||
const accessor = this.instantiationService.createInstance(ServiceAccessor);
|
||||
const service = accessor.untitledTextEditorService;
|
||||
return this.instantiationService.createInstance(UntitledTextEditorInput, service.create());
|
||||
}
|
||||
}
|
||||
|
||||
class ServiceAccessor {
|
||||
constructor(
|
||||
@IUntitledTextEditorService public readonly untitledTextEditorService: IUntitledTextEditorService
|
||||
) { }
|
||||
}
|
||||
Reference in New Issue
Block a user