diff --git a/lib/internal/test_runner/coverage.js b/lib/internal/test_runner/coverage.js index 7d78bd69886566..4f9bd8d20e3409 100644 --- a/lib/internal/test_runner/coverage.js +++ b/lib/internal/test_runner/coverage.js @@ -23,6 +23,7 @@ const { mkdtempSync, opendirSync, readFileSync, + rmSync, } = require('fs'); const { setupCoverageHooks } = require('internal/util'); const { tmpdir } = require('os'); @@ -272,28 +273,35 @@ class TestCoverage { cleanup() { // Restore the original value of process.env.NODE_V8_COVERAGE. Then, copy // all of the created coverage files to the original coverage directory. + internalBinding('profiler').endCoverage(); + if (this.originalCoverageDirectory === undefined) { delete process.env.NODE_V8_COVERAGE; - return; - } - - process.env.NODE_V8_COVERAGE = this.originalCoverageDirectory; - let dir; + } else { + process.env.NODE_V8_COVERAGE = this.originalCoverageDirectory; + let dir; - try { - mkdirSync(this.originalCoverageDirectory, { __proto__: null, recursive: true }); - dir = opendirSync(this.coverageDirectory); + try { + mkdirSync(this.originalCoverageDirectory, { __proto__: null, recursive: true }); + dir = opendirSync(this.coverageDirectory); - for (let entry; (entry = dir.readSync()) !== null;) { - const src = join(this.coverageDirectory, entry.name); - const dst = join(this.originalCoverageDirectory, entry.name); - copyFileSync(src, dst); - } - } finally { - if (dir) { - dir.closeSync(); + for (let entry; (entry = dir.readSync()) !== null;) { + const src = join(this.coverageDirectory, entry.name); + const dst = join(this.originalCoverageDirectory, entry.name); + copyFileSync(src, dst); + } + } finally { + if (dir) { + dir.closeSync(); + } } } + + try { + rmSync(this.coverageDirectory, { __proto__: null, recursive: true }); + } catch { + // Ignore cleanup errors. + } } getCoverageFromDirectory() { diff --git a/lib/internal/test_runner/harness.js b/lib/internal/test_runner/harness.js index 1bc6cddabd41a0..34ebbf50c658c1 100644 --- a/lib/internal/test_runner/harness.js +++ b/lib/internal/test_runner/harness.js @@ -159,12 +159,15 @@ function collectCoverage(rootTest, coverage) { try { summary = coverage.summary(); - coverage.cleanup(); } catch (err) { - const op = summary ? 'clean up' : 'report'; - const msg = `Warning: Could not ${op} code coverage. ${err}`; + rootTest.diagnostic(`Warning: Could not report code coverage. ${err}`); + process.exitCode = kGenericUserError; + } - rootTest.diagnostic(msg); + try { + coverage.cleanup(); + } catch (err) { + rootTest.diagnostic(`Warning: Could not clean up code coverage. ${err}`); process.exitCode = kGenericUserError; } diff --git a/src/inspector_profiler.cc b/src/inspector_profiler.cc index 8cd39e091242bf..c7554e424be327 100644 --- a/src/inspector_profiler.cc +++ b/src/inspector_profiler.cc @@ -556,6 +556,21 @@ static void StopCoverage(const FunctionCallbackInfo& args) { } } +static void EndCoverage(const FunctionCallbackInfo& args) { + Environment* env = Environment::GetCurrent(args); + V8CoverageConnection* connection = env->coverage_connection(); + + Debug(env, + DebugCategory::INSPECTOR_PROFILER, + "EndCoverage, connection %s nullptr\n", + connection == nullptr ? "==" : "!="); + + if (connection != nullptr) { + Debug(env, DebugCategory::INSPECTOR_PROFILER, "Ending coverage\n"); + connection->End(); + } +} + static void Initialize(Local target, Local unused, Local context, @@ -565,6 +580,7 @@ static void Initialize(Local target, context, target, "setSourceMapCacheGetter", SetSourceMapCacheGetter); SetMethod(context, target, "takeCoverage", TakeCoverage); SetMethod(context, target, "stopCoverage", StopCoverage); + SetMethod(context, target, "endCoverage", EndCoverage); } void RegisterExternalReferences(ExternalReferenceRegistry* registry) { @@ -572,6 +588,7 @@ void RegisterExternalReferences(ExternalReferenceRegistry* registry) { registry->Register(SetSourceMapCacheGetter); registry->Register(TakeCoverage); registry->Register(StopCoverage); + registry->Register(EndCoverage); } } // namespace profiler