write tests

Author: QuentinGruber

src/extension.ts

@@ -1,9 +1,9 @@
 import * as vscode from 'vscode';
-const path = require('path');
+import * as path from 'path';
 import * as fs from 'fs';
 import * as _  from 'lodash';
 
-function GetTemplates(templatePath: fs.PathLike): Array<Array<string>> {
+export function GetTemplates(templatePath: fs.PathLike): Array<Array<string>> {
 	const files = fs.readdirSync(templatePath);
 	const templates: Array<string> = [];
 	files.filter(e => e.includes(".dockerignore") ? false : true).forEach(file => {
@@ -12,14 +12,28 @@ function GetTemplates(templatePath: fs.PathLike): Array<Array<string>> {
 	return [templates, files];
 }
 
-function CreateDockerFile(option: any , templatePath: String, workspaceFolder: String): void {
-	const data = fs.readFileSync(`${templatePath}/${option}.dockerfile`);
-	fs.writeFileSync(`${workspaceFolder}/Dockerfile`, data);
+export function CreateDockerFile(option: any, templatePath: String, workspaceFolder: String): Boolean {
+	try {
+		const data = fs.readFileSync(`${templatePath}/${option}.dockerfile`);
+	  fs.writeFileSync(`${workspaceFolder}/Dockerfile`, data);
+	} catch (error) {
+		return false
+	}
+	finally {
+		return true
+	}
 }
 
-function CreateIgnoreDockerFile(option: any , templatePath: String, workspaceFolder: String): void {
-	const data = fs.readFileSync(`${templatePath}/.dockerignore ${option}`);
-	fs.writeFileSync(`${workspaceFolder}/.dockerignore`, data);
+export function CreateIgnoreDockerFile(option: any, templatePath: String, workspaceFolder: String): Boolean {
+	try {
+		const data = fs.readFileSync(`${templatePath}/.dockerignore ${option}`);
+		fs.writeFileSync(`${workspaceFolder}/.dockerignore`, data);
+	} catch (error) {
+		return false
+	}
+	finally {
+		return true
+	}
 }
 
 export function activate(context: vscode.ExtensionContext) {
@@ -45,9 +59,9 @@ export function activate(context: vscode.ExtensionContext) {
 	}
 	else {
 		vscode.window.showInformationMessage("workspaceFolder not found")
-		}
+	}
 	})
 	context.subscriptions.push(disposable);
 }
 
-export function deactivate() {}
+export function deactivate() { }

src/test/suite/extension.test.ts

@@ -1,15 +1,29 @@
 import * as assert from 'assert';
-
+import * as path from 'path';
 // You can import and use all API from the 'vscode' module
 // as well as import your extension to test it
 import * as vscode from 'vscode';
-// import * as myExtension from '../../extension';
+import { CreateDockerFile, CreateIgnoreDockerFile, GetTemplates } from '../../extension';
 
 suite('Extension Test Suite', () => {
 	vscode.window.showInformationMessage('Start all tests.');
 
-	test('Sample test', () => {
-		assert.strictEqual(-1, [1, 2, 3].indexOf(5));
-		assert.strictEqual(-1, [1, 2, 3].indexOf(0));
+	test('CreateDockerFile', () => {
+		const templatePath = path.join(__dirname, '../templates');
+		const { workspace: { workspaceFolders } } = vscode;
+		const workspaceFolder = workspaceFolders ? workspaceFolders[0].uri.fsPath : path.join(__dirname, '../');
+		assert.strictEqual(CreateDockerFile("Nodejs app",templatePath,workspaceFolder), true);
+	});
+
+	test('CreateIgnoreDockerFile', () => {
+		const templatePath = path.join(__dirname, '../templates');
+		const { workspace: { workspaceFolders } } = vscode;
+		const workspaceFolder = workspaceFolders ? workspaceFolders[0].uri.fsPath : path.join(__dirname, '../');
+		assert.strictEqual(CreateIgnoreDockerFile("Nodejs app",templatePath,workspaceFolder), true);
+	});
+
+	test('GetTemplates', () => {
+		const templatePath = path.join(__dirname, '../../../templates');
+		assert.strictEqual(GetTemplates(templatePath).length > 0 ? true : false, true);
 	});
 });