Merge from vscode 112fa76c775ecb79ac2c9e9e5dba0d711d523543 (#6388)

test/coverage.js

@@ -0,0 +1,91 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the Source EULA. 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');
+
+// {{SQL CARBON EDIT}}
+var SQL_TEST_GLOB = '**/test/**/*.test.js|**/sqltest/**/*.test.jss';
+
+const REPO_PATH = toUpperDriveLetter(path.join(__dirname, '..'));
+
+exports.initialize = function (loaderConfig) {
+	const instrumenter = iLibInstrument.createInstrumenter();
+	loaderConfig.nodeInstrumenter = function (contents, source) {
+		if (minimatch(source, SQL_TEST_GLOB)) { // {{SQL CARBON EDIT}}
+			// tests don't get instrumented
+			return contents;
+		}
+		// Try to find a .map file
+		let map = null;
+		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__);
+	const transformed = mapStore.transformCoverage(coverageMap);
+
+	// Paths come out all broken
+	let newData = Object.create(null);
+	Object.keys(transformed.map.data).forEach((file) => {
+		const entry = transformed.map.data[file];
+		const fixedPath = fixPath(entry.path);
+		if (!fixedPath.includes('\\vs\\') && !fixedPath.includes('/vs/')) {
+			entry.data.path = fixedPath;
+			newData[fixedPath] = entry;
+		}
+	});
+	transformed.map.data = newData;
+
+	const tree = iLibReport.summarizers.flat(transformed.map);
+	const context = iLibReport.createContext({
+		dir: path.join(__dirname, `../.build/coverage${isSingle ? '-single' : ''}`)
+	});
+
+	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));
+}