Added unit test for books getChildren() method (#6602)

* started unit tests for books

* added first unit test for getchildren()

* fixed dependencies and use tmpdir

* latest from master

extensions/notebook/src/book/bookTreeItem.ts

@@ -100,6 +100,14 @@ export class BookTreeItem extends vscode.TreeItem {
 		}
 	}
 
+	public get title(): string {
+		return this.book.title;
+	}
+
+	public get uri(): string {
+		return this._uri;
+	}
+
 	public get root(): string {
 		return this.book.root;
 	}
@@ -128,4 +136,4 @@ export class BookTreeItem extends vscode.TreeItem {
 			return undefined;
 		}
 	}
-}
+}

extensions/notebook/src/book/bookTreeView.ts

@@ -121,7 +121,7 @@ export class BookTreeViewProvider implements vscode.TreeDataProvider<BookTreeIte
 		return array.reduce((acc, val) => Array.isArray(val.sections) ? acc.concat(val).concat(this.flattenArray(val.sections)) : acc.concat(val), []);
 	}
 
-	private getBooks(): BookTreeItem[] {
+	public getBooks(): BookTreeItem[] {
 		let books: BookTreeItem[] = [];
 		for (let i in this._tableOfContentsPath) {
 			let root = path.dirname(path.dirname(this._tableOfContentsPath[i]));

extensions/notebook/src/test/book/book.test.ts

@@ -0,0 +1,86 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the Source EULA. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import * as vscode from 'vscode';
+import * as should from 'should';
+import * as TypeMoq from 'typemoq';
+import * as path from 'path';
+import * as fs from 'fs';
+import * as rimraf from 'rimraf';
+import * as os from 'os';
+import { BookTreeViewProvider } from '../../book/bookTreeView';
+import { BookTreeItem } from '../../book/bookTreeItem';
+
+describe('BookTreeViewProvider.getChildren', function (): void {
+	const rootFolderPath = path.join(os.tmpdir(), 'testBook');
+	const dataFolderPath = path.join(rootFolderPath, '_data');
+	const contentFolderPath = path.join(rootFolderPath, 'content');
+	const configFile = path.join(rootFolderPath, '_config.yml');
+	const tableOfContentsFile = path.join(dataFolderPath, 'toc.yml');
+	const notebookFile = path.join(contentFolderPath, 'notebook.ipynb');
+	const markdownFile = path.join(contentFolderPath, 'markdown.md');
+	const expectedNotebook = {
+		title: 'Notebook',
+		url: '/notebook'
+	};
+	const expectedMarkdown = {
+		title: 'Markdown',
+		url: '/markdown'
+	};
+	const expectedExternalLink = {
+		title: 'GitHub',
+		url: 'https://github.com/',
+		external: true
+	};
+	const expectedBook = {
+		sections: [expectedNotebook, expectedMarkdown, expectedExternalLink],
+		title: 'Test Book'
+	};
+
+	let mockExtensionContext: TypeMoq.IMock<vscode.ExtensionContext>;
+	let bookTreeViewProvider: BookTreeViewProvider;
+	let book: BookTreeItem;
+
+	this.beforeAll(async () => {
+		fs.mkdirSync(rootFolderPath);
+		fs.mkdirSync(dataFolderPath);
+		fs.mkdirSync(contentFolderPath);
+		fs.writeFileSync(configFile, 'title: Test Book');
+		fs.writeFileSync(tableOfContentsFile, '- title: Notebook\n  url: /notebook\n- title: Markdown\n  url: /markdown\n- title: GitHub\n  url: https://github.com/\n  external: true');
+		fs.writeFileSync(notebookFile, '');
+		fs.writeFileSync(markdownFile, '');
+		mockExtensionContext = TypeMoq.Mock.ofType<vscode.ExtensionContext>();
+		bookTreeViewProvider = new BookTreeViewProvider(rootFolderPath, mockExtensionContext.object);
+	});
+
+	it('should return all book nodes when element is undefined', async function (): Promise<void> {
+		const children = await bookTreeViewProvider.getChildren();
+		should(children).be.Array();
+		should(children.length).equal(1);
+		book = children[0];
+		should(book.title).equal(expectedBook.title);
+	});
+
+	it('should return all page nodes when element is a book', async function (): Promise<void> {
+		const children = await bookTreeViewProvider.getChildren(book);
+		should(children).be.Array();
+		should(children.length).equal(3);
+		const notebook = children[0];
+		const markdown = children[1];
+		const externalLink = children[2];
+		should(notebook.title).equal(expectedNotebook.title);
+		should(notebook.uri).equal(expectedNotebook.url);
+		should(markdown.title).equal(expectedMarkdown.title);
+		should(markdown.uri).equal(expectedMarkdown.url);
+		should(externalLink.title).equal(expectedExternalLink.title);
+		should(externalLink.uri).equal(expectedExternalLink.url);
+	});
+
+	after(async function () {
+		if (fs.existsSync(rootFolderPath)) {
+			rimraf.sync(rootFolderPath);
+		}
+	});
+});