Skip to content

Commit 9f2be31

Browse files
authored
Fix some test instantiation service errors in notebooks (microsoft#166069)
For microsoft#164297
1 parent ef41557 commit 9f2be31

File tree

3 files changed

+20
-11
lines changed

3 files changed

+20
-11
lines changed

src/vs/workbench/contrib/notebook/test/browser/notebookEditorModel.test.ts

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -44,9 +44,9 @@ suite('NotebookFileWorkingCopyModel', function () {
4444
const notebook = instantiationService.createInstance(NotebookTextModel,
4545
'notebook',
4646
URI.file('test'),
47-
[{ cellKind: CellKind.Code, language: 'foo', source: 'foo', outputs: [{ outputId: 'id', outputs: [{ mime: Mimes.text, value: 'Hello Out' }] }] }],
47+
[{ cellKind: CellKind.Code, language: 'foo', mime: 'foo', source: 'foo', outputs: [{ outputId: 'id', outputs: [{ mime: Mimes.text, data: VSBuffer.fromString('Hello Out') }] }] }],
4848
{},
49-
{ transientCellMetadata: {}, transientDocumentMetadata: {}, transientOutputs: false }
49+
{ transientCellMetadata: {}, transientDocumentMetadata: {}, cellContentMetadata: {}, transientOutputs: false }
5050
);
5151

5252
{ // transient output
@@ -92,9 +92,9 @@ suite('NotebookFileWorkingCopyModel', function () {
9292
const notebook = instantiationService.createInstance(NotebookTextModel,
9393
'notebook',
9494
URI.file('test'),
95-
[{ cellKind: CellKind.Code, language: 'foo', source: 'foo', outputs: [] }],
95+
[{ cellKind: CellKind.Code, language: 'foo', mime: 'foo', source: 'foo', outputs: [] }],
9696
{ foo: 123, bar: 456 },
97-
{ transientCellMetadata: {}, transientDocumentMetadata: {}, transientOutputs: false }
97+
{ transientCellMetadata: {}, transientDocumentMetadata: {}, cellContentMetadata: {}, transientOutputs: false }
9898
);
9999

100100
{ // transient
@@ -140,9 +140,9 @@ suite('NotebookFileWorkingCopyModel', function () {
140140
const notebook = instantiationService.createInstance(NotebookTextModel,
141141
'notebook',
142142
URI.file('test'),
143-
[{ cellKind: CellKind.Code, language: 'foo', source: 'foo', outputs: [], metadata: { foo: 123, bar: 456 } }],
143+
[{ cellKind: CellKind.Code, language: 'foo', mime: 'foo', source: 'foo', outputs: [], metadata: { foo: 123, bar: 456 } }],
144144
{},
145-
{ transientCellMetadata: {}, transientDocumentMetadata: {}, transientOutputs: false }
145+
{ transientCellMetadata: {}, transientDocumentMetadata: {}, cellContentMetadata: {}, transientOutputs: false, }
146146
);
147147

148148
{ // transient

src/vs/workbench/contrib/notebook/test/browser/notebookExecutionService.test.ts

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@ import { PLAINTEXT_LANGUAGE_ID } from 'vs/editor/common/languages/modesRegistry'
1414
import { IMenu, IMenuService } from 'vs/platform/actions/common/actions';
1515
import { ExtensionIdentifier } from 'vs/platform/extensions/common/extensions';
1616
import { TestInstantiationService } from 'vs/platform/instantiation/test/common/instantiationServiceMock';
17+
import { MockContextKeyService } from 'vs/platform/keybinding/test/common/mockKeybindingService';
1718
import { insertCellAtIndex } from 'vs/workbench/contrib/notebook/browser/controller/cellOperations';
1819
import { NotebookExecutionService } from 'vs/workbench/contrib/notebook/browser/services/notebookExecutionServiceImpl';
1920
import { NotebookKernelService } from 'vs/workbench/contrib/notebook/browser/services/notebookKernelServiceImpl';
@@ -108,7 +109,7 @@ suite('NotebookExecutionService', () => {
108109
kernel.executeNotebookCellsRequest = executeSpy;
109110

110111
const cell = insertCellAtIndex(viewModel, 0, 'var c = 3', 'javascript', CellKind.Code, {}, [], true, true);
111-
await executionService.executeNotebookCells(viewModel.notebookDocument, [cell]);
112+
await executionService.executeNotebookCells(viewModel.notebookDocument, [cell], new MockContextKeyService());
112113
assert.strictEqual(executeSpy.calledOnce, true);
113114
});
114115
});
@@ -139,7 +140,7 @@ suite('NotebookExecutionService', () => {
139140
kernelService.onDidChangeSelectedNotebooks(e => event = e);
140141

141142
const cell = insertCellAtIndex(viewModel, 0, 'var c = 3', 'javascript', CellKind.Code, {}, [], true, true);
142-
await executionService.executeNotebookCells(viewModel.notebookDocument, [cell]);
143+
await executionService.executeNotebookCells(viewModel.notebookDocument, [cell], new MockContextKeyService());
143144

144145
assert.strictEqual(didExecute, true);
145146
assert.ok(event !== undefined);
@@ -169,7 +170,7 @@ suite('NotebookExecutionService', () => {
169170
const exeStateService = instantiationService.get(INotebookExecutionStateService);
170171

171172
const cell = insertCellAtIndex(viewModel, 0, 'var c = 3', 'javascript', CellKind.Code, {}, [], true, true);
172-
await executionService.executeNotebookCells(viewModel.notebookDocument, [cell]);
173+
await executionService.executeNotebookCells(viewModel.notebookDocument, [cell], new MockContextKeyService());
173174

174175
assert.strictEqual(didExecute, true);
175176
assert.strictEqual(exeStateService.getCellExecution(cell.uri), undefined);

src/vs/workbench/contrib/notebook/test/browser/notebookKernelService.test.ts

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@ import { DisposableStore } from 'vs/base/common/lifecycle';
1717
import { NotebookTextModel } from 'vs/workbench/contrib/notebook/common/model/notebookTextModel';
1818
import { PLAINTEXT_LANGUAGE_ID } from 'vs/editor/common/languages/modesRegistry';
1919
import { IMenu, IMenuService } from 'vs/platform/actions/common/actions';
20+
import { TransientOptions } from 'vs/workbench/contrib/notebook/common/notebookCommon';
2021

2122
suite('NotebookKernelService', () => {
2223

@@ -150,18 +151,25 @@ suite('NotebookKernelService', () => {
150151
kernelService.selectKernelForNotebook(jupyterKernel, jupyter);
151152
kernelService.selectKernelForNotebook(dotnetKernel, dotnet);
152153

154+
const transientOptions: TransientOptions = {
155+
transientOutputs: false,
156+
transientCellMetadata: {},
157+
transientDocumentMetadata: {},
158+
cellContentMetadata: {},
159+
};
160+
153161
{
154162
// open as jupyter -> bind event
155163
const p1 = Event.toPromise(kernelService.onDidChangeSelectedNotebooks);
156-
const d1 = instantiationService.createInstance(NotebookTextModel, jupyter.viewType, jupyter.uri, [], {}, {});
164+
const d1 = instantiationService.createInstance(NotebookTextModel, jupyter.viewType, jupyter.uri, [], {}, transientOptions);
157165
onDidAddNotebookDocument.fire(d1);
158166
const event = await p1;
159167
assert.strictEqual(event.newKernel, jupyterKernel.id);
160168
}
161169
{
162170
// RE-open as dotnet -> bind event
163171
const p2 = Event.toPromise(kernelService.onDidChangeSelectedNotebooks);
164-
const d2 = instantiationService.createInstance(NotebookTextModel, dotnet.viewType, dotnet.uri, [], {}, {});
172+
const d2 = instantiationService.createInstance(NotebookTextModel, dotnet.viewType, dotnet.uri, [], {}, transientOptions);
165173
onDidAddNotebookDocument.fire(d2);
166174
const event2 = await p2;
167175
assert.strictEqual(event2.newKernel, dotnetKernel.id);

0 commit comments

Comments
 (0)