Merge commit 'be3e8236086165e5e45a5a10783823874b3f3ebd' as 'lib/vscode'

lib/vscode/test/unit/coverage.js

@@ -0,0 +1,87 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+const minimatch = require('minimatch');
+const fs = require('fs');
+const path = require('path');
+const iLibInstrument = require('istanbul-lib-instrument');
+const iLibCoverage = require('istanbul-lib-coverage');
+const iLibSourceMaps = require('istanbul-lib-source-maps');
+const iLibReport = require('istanbul-lib-report');
+const iReports = require('istanbul-reports');
+
+const REPO_PATH = toUpperDriveLetter(path.join(__dirname, '../../'));
+
+exports.initialize = function (loaderConfig) {
+	const instrumenter = iLibInstrument.createInstrumenter();
+	loaderConfig.nodeInstrumenter = function (contents, source) {
+		if (minimatch(source, '**/test/**')) {
+			// tests don't get instrumented
+			return contents;
+		}
+		// Try to find a .map file
+		let map = undefined;
+		try {
+			map = JSON.parse(fs.readFileSync(`${source}.map`).toString());
+		} catch (err) {
+			// missing source map...
+		}
+		return instrumenter.instrumentSync(contents, source, map);
+	};
+};
+
+exports.createReport = function (isSingle) {
+	const mapStore = iLibSourceMaps.createSourceMapStore();
+	const coverageMap = iLibCoverage.createCoverageMap(global.__coverage__);
+	return mapStore.transformCoverage(coverageMap).then((transformed) => {
+		// Paths come out all broken
+		let newData = Object.create(null);
+		Object.keys(transformed.data).forEach((file) => {
+			const entry = transformed.data[file];
+			const fixedPath = fixPath(entry.path);
+			entry.data.path = fixedPath;
+			newData[fixedPath] = entry;
+		});
+		transformed.data = newData;
+
+		const context = iLibReport.createContext({
+			dir: path.join(REPO_PATH, `.build/coverage${isSingle ? '-single' : ''}`),
+			coverageMap: transformed
+		});
+		const tree = context.getTree('flat');
+
+		let reports = [];
+		if (isSingle) {
+			reports.push(iReports.create('lcovonly'));
+		} else {
+			reports.push(iReports.create('json'));
+			reports.push(iReports.create('lcov'));
+			reports.push(iReports.create('html'));
+		}
+		reports.forEach(report => tree.visit(report, context));
+	});
+};
+
+function toUpperDriveLetter(str) {
+	if (/^[a-z]:/.test(str)) {
+		return str.charAt(0).toUpperCase() + str.substr(1);
+	}
+	return str;
+}
+
+function toLowerDriveLetter(str) {
+	if (/^[A-Z]:/.test(str)) {
+		return str.charAt(0).toLowerCase() + str.substr(1);
+	}
+	return str;
+}
+
+function fixPath(brokenPath) {
+	const startIndex = brokenPath.lastIndexOf(REPO_PATH);
+	if (startIndex === -1) {
+		return toLowerDriveLetter(brokenPath);
+	}
+	return toLowerDriveLetter(brokenPath.substr(startIndex));
+}