From b7736a24c7f9545432d3398ca20df45adf3fd736 Mon Sep 17 00:00:00 2001
From: rebornix <penn.lv@gmail.com>
Date: Fri, 1 May 2020 17:21:30 -0700
Subject: [PATCH] more testing for unsaved document

---
 .../src/notebook.test.ts                      | 83 +++++++++++++++++++
 1 file changed, 83 insertions(+)

diff --git a/extensions/vscode-notebook-tests/src/notebook.test.ts b/extensions/vscode-notebook-tests/src/notebook.test.ts
index f08c9e2d4ae..e4eba9aefb3 100644
--- a/extensions/vscode-notebook-tests/src/notebook.test.ts
+++ b/extensions/vscode-notebook-tests/src/notebook.test.ts
@@ -189,3 +189,86 @@ suite('notebook workflow', () => {
 		await vscode.commands.executeCommand('workbench.action.closeActiveEditor');
 	});
 });
+
+suite('notebook dirty state', () => {
+	test('notebook open', async function () {
+		const resource = vscode.Uri.parse(join(vscode.workspace.rootPath || '', './first.vsctestnb'));
+		await vscode.commands.executeCommand('vscode.openWith', resource, 'notebookCoreTest');
+
+		await waitFor(500);
+		assert.equal(vscode.notebook.activeNotebookEditor !== undefined, true, 'notebook first');
+		assert.equal(vscode.notebook.activeNotebookEditor!.selection?.source, 'test');
+		assert.equal(vscode.notebook.activeNotebookEditor!.selection?.language, 'typescript');
+
+		await vscode.commands.executeCommand('notebook.cell.insertCodeCellBelow');
+		assert.equal(vscode.notebook.activeNotebookEditor!.selection?.source, '');
+
+		await vscode.commands.executeCommand('notebook.cell.insertCodeCellAbove');
+		const activeCell = vscode.notebook.activeNotebookEditor!.selection;
+		assert.notEqual(vscode.notebook.activeNotebookEditor!.selection, undefined);
+		assert.equal(activeCell!.source, '');
+		assert.equal(vscode.notebook.activeNotebookEditor!.document.cells.length, 3);
+		assert.equal(vscode.notebook.activeNotebookEditor!.document.cells.indexOf(activeCell!), 1);
+
+
+		await vscode.commands.executeCommand('default:type', { text: 'var abc = 0;' });
+		// await vscode.commands.executeCommand('workbench.action.files.save');
+		await vscode.commands.executeCommand('workbench.action.closeActiveEditor');
+
+		await vscode.commands.executeCommand('vscode.openWith', resource, 'notebookCoreTest');
+
+		await waitFor(500);
+		assert.equal(vscode.notebook.activeNotebookEditor !== undefined, true);
+		assert.equal(vscode.notebook.activeNotebookEditor?.selection !== undefined, true);
+		assert.deepEqual(vscode.notebook.activeNotebookEditor?.document.cells[1], vscode.notebook.activeNotebookEditor?.selection);
+		assert.equal(vscode.notebook.activeNotebookEditor?.selection?.source, 'var abc = 0;');
+
+		await vscode.commands.executeCommand('workbench.action.files.save');
+		await vscode.commands.executeCommand('workbench.action.closeActiveEditor');
+	});
+});
+
+suite('notebook undo redo', () => {
+	test('notebook open', async function () {
+		const resource = vscode.Uri.parse(join(vscode.workspace.rootPath || '', './first.vsctestnb'));
+		await vscode.commands.executeCommand('vscode.openWith', resource, 'notebookCoreTest');
+
+		await waitFor(500);
+		assert.equal(vscode.notebook.activeNotebookEditor !== undefined, true, 'notebook first');
+		assert.equal(vscode.notebook.activeNotebookEditor!.selection?.source, 'test');
+		assert.equal(vscode.notebook.activeNotebookEditor!.selection?.language, 'typescript');
+
+		await vscode.commands.executeCommand('notebook.cell.insertCodeCellBelow');
+		assert.equal(vscode.notebook.activeNotebookEditor!.selection?.source, '');
+
+		await vscode.commands.executeCommand('notebook.cell.insertCodeCellAbove');
+		const activeCell = vscode.notebook.activeNotebookEditor!.selection;
+		assert.notEqual(vscode.notebook.activeNotebookEditor!.selection, undefined);
+		assert.equal(activeCell!.source, '');
+		assert.equal(vscode.notebook.activeNotebookEditor!.document.cells.length, 3);
+		assert.equal(vscode.notebook.activeNotebookEditor!.document.cells.indexOf(activeCell!), 1);
+
+
+		// modify the second cell, delete it
+		await vscode.commands.executeCommand('default:type', { text: 'var abc = 0;' });
+		await vscode.commands.executeCommand('notebook.cell.delete');
+		assert.equal(vscode.notebook.activeNotebookEditor!.document.cells.length, 2);
+		assert.equal(vscode.notebook.activeNotebookEditor!.document.cells.indexOf(vscode.notebook.activeNotebookEditor!.selection!), 1);
+
+
+		// undo should bring back the deleted cell, and revert to previous content and selection
+		await vscode.commands.executeCommand('notebook.undo');
+		assert.equal(vscode.notebook.activeNotebookEditor!.document.cells.length, 3);
+		assert.equal(vscode.notebook.activeNotebookEditor!.document.cells.indexOf(vscode.notebook.activeNotebookEditor!.selection!), 1);
+		assert.equal(vscode.notebook.activeNotebookEditor?.selection?.source, 'var abc = 0;');
+
+		// redo
+		// await vscode.commands.executeCommand('notebook.redo');
+		// assert.equal(vscode.notebook.activeNotebookEditor!.document.cells.length, 2);
+		// assert.equal(vscode.notebook.activeNotebookEditor!.document.cells.indexOf(vscode.notebook.activeNotebookEditor!.selection!), 1);
+		// assert.equal(vscode.notebook.activeNotebookEditor?.selection?.source, 'test');
+
+		await vscode.commands.executeCommand('workbench.action.files.save');
+		await vscode.commands.executeCommand('workbench.action.closeActiveEditor');
+	});
+});