Migrate flutter_tools to use package:coverage (#111681)

packages/flutter_tools/lib/src/test/coverage_collector.dart

@@ -2,11 +2,8 @@
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
-
-
 import 'package:coverage/coverage.dart' as coverage;
 import 'package:meta/meta.dart';
-import 'package:vm_service/vm_service.dart' as vm_service;
 
 import '../base/file_system.dart';
 import '../base/io.dart';
@@ -99,13 +96,11 @@ class CoverageCollector extends TestWatcher {
   ///
   /// The returned [Future] completes when the coverage is collected.
   Future<void> collectCoverageIsolate(Uri observatoryUri) async {
-    assert(observatoryUri != null);
     _logMessage('collecting coverage data from $observatoryUri...');
     final Map<String, dynamic> data = await collect(observatoryUri, libraryNames);
     if (data == null) {
       throw Exception('Failed to collect coverage.');
     }
-    assert(data != null);
 
     _logMessage('($observatoryUri): collected coverage data; merging...');
     _addHitmap(await coverage.HitMap.parseJson(
@@ -122,12 +117,12 @@ class CoverageCollector extends TestWatcher {
   /// has been run to completion so that all coverage data has been recorded.
   ///
   /// The returned [Future] completes when the coverage is collected.
-  Future<void> collectCoverage(TestDevice testDevice, {@visibleForTesting Future<FlutterVmService> Function(Uri?)? connector}) async {
-    assert(testDevice != null);
-
+  Future<void> collectCoverage(TestDevice testDevice, {
+    @visibleForTesting FlutterVmService? serviceOverride,
+  }) async {
     final Stopwatch? totalTestTimeRecorderStopwatch = testTimeRecorder?.start(TestTimePhases.CoverageTotal);
 
-    Map<String, dynamic>? data;
+    late Map<String, dynamic> data;
 
     final Stopwatch? collectTestTimeRecorderStopwatch = testTimeRecorder?.start(TestTimePhases.CoverageCollect);
 
@@ -141,7 +136,7 @@ class CoverageCollector extends TestWatcher {
     final Future<void> collectionComplete = testDevice.observatoryUri
       .then((Uri? observatoryUri) {
         _logMessage('collecting coverage data from $testDevice at $observatoryUri...');
-        return collect(observatoryUri, libraryNames, connector: connector ?? _defaultConnect)
+        return collect(observatoryUri!, libraryNames, serviceOverride: serviceOverride)
           .then<void>((Map<String, dynamic> result) {
             if (result == null) {
               throw Exception('Failed to collect coverage.');
@@ -152,13 +147,14 @@ class CoverageCollector extends TestWatcher {
       });
 
     await Future.any<void>(<Future<void>>[ processComplete, collectionComplete ]);
-    assert(data != null);
+
     testTimeRecorder?.stop(TestTimePhases.CoverageCollect, collectTestTimeRecorderStopwatch!);
 
     _logMessage('Merging coverage data...');
     final Stopwatch? parseTestTimeRecorderStopwatch = testTimeRecorder?.start(TestTimePhases.CoverageParseJson);
-    final Map<String, coverage.HitMap> hitmap = coverage.HitMap.parseJsonSync(
-        data!['coverage'] as List<Map<String, dynamic>>,
+
+   final Map<String, coverage.HitMap> hitmap = coverage.HitMap.parseJsonSync(
+        data['coverage'] as List<Map<String, dynamic>>,
         checkIgnoredLines: true,
         resolver: resolver ?? await CoverageCollector.getResolver(packageDirectory),
         ignoredLinesInFilesCache: _ignoredLinesInFilesCache);
@@ -253,158 +249,11 @@ class CoverageCollector extends TestWatcher {
   Future<void> handleTestTimedOut(TestDevice testDevice) async { }
 }
 
-Future<FlutterVmService> _defaultConnect(Uri? serviceUri) {
-  return connectToVmService(
-      serviceUri!, compression: CompressionOptions.compressionOff, logger: globals.logger,);
-}
-
-Future<Map<String, dynamic>> collect(Uri? serviceUri, Set<String>? libraryNames, {
+Future<Map<String, dynamic>> collect(Uri serviceUri, Set<String>? libraryNames, {
   bool waitPaused = false,
   String? debugName,
-  Future<FlutterVmService> Function(Uri?) connector = _defaultConnect,
   @visibleForTesting bool forceSequential = false,
-}) async {
-  final FlutterVmService vmService = await connector(serviceUri);
-  final Map<String, dynamic> result = await _getAllCoverage(vmService.service, libraryNames, forceSequential);
-  await vmService.dispose();
-  return result;
-}
-
-Future<Map<String, dynamic>> _getAllCoverage(
-  vm_service.VmService service,
-  Set<String>? libraryNames,
-  bool forceSequential,
-) async {
-  final vm_service.Version version = await service.getVersion();
-  final bool libraryFilters = (version.major == 3 && version.minor! >= 57) || version.major! > 3;
-  final vm_service.VM vm = await service.getVM();
-  final List<Map<String, dynamic>> coverage = <Map<String, dynamic>>[];
-  bool libraryPredicate(String? libraryName) {
-    if (libraryNames == null) {
-      return true;
-    }
-    final Uri uri = Uri.parse(libraryName!);
-    if (uri.scheme != 'package') {
-      return false;
-    }
-    final String scope = uri.path.split('/').first;
-    return libraryNames.contains(scope);
-  }
-  for (final vm_service.IsolateRef isolateRef in vm.isolates!) {
-    if (isolateRef.isSystemIsolate!) {
-      continue;
-    }
-    if (libraryFilters) {
-      final vm_service.SourceReport sourceReport = await service.getSourceReport(
-          isolateRef.id!,
-          <String>['Coverage'],
-          forceCompile: true,
-          reportLines: true,
-          libraryFilters: libraryNames == null ? null : List<String>.from(
-              libraryNames.map((String name) => 'package:$name/')),
-        );
-      _buildCoverageMap(
-          <String, vm_service.Script>{},
-          <vm_service.SourceReport>[sourceReport],
-          coverage,
-        );
-    } else {
-      vm_service.ScriptList scriptList;
-      try {
-        scriptList = await service.getScripts(isolateRef.id!);
-      } on vm_service.SentinelException {
-        continue;
-      }
-
-      final List<Future<void>> futures = <Future<void>>[];
-      final Map<String, vm_service.Script> scripts = <String, vm_service.Script>{};
-      final List<vm_service.SourceReport> sourceReports = <vm_service.SourceReport>[];
-
-      // For each ScriptRef loaded into the VM, load the corresponding Script
-      // and SourceReport object.
-      for (final vm_service.ScriptRef script in scriptList.scripts!) {
-        final String? libraryUri = script.uri;
-        if (!libraryPredicate(libraryUri)) {
-          continue;
-        }
-        final String? scriptId = script.id;
-        final Future<void> getSourceReport = service.getSourceReport(
-          isolateRef.id!,
-          <String>['Coverage'],
-          scriptId: scriptId,
-          forceCompile: true,
-          reportLines: true,
-        )
-        .then((vm_service.SourceReport report) {
-          sourceReports.add(report);
-        });
-        if (forceSequential) {
-          await null;
-        }
-        futures.add(getSourceReport);
-      }
-      await Future.wait(futures);
-      _buildCoverageMap(scripts, sourceReports, coverage);
-    }
-  }
-  return <String, dynamic>{'type': 'CodeCoverage', 'coverage': coverage};
-}
-
-// Build a hitmap of Uri -> Line -> Hit Count for each script object.
-void _buildCoverageMap(
-  Map<String, vm_service.Script> scripts,
-  List<vm_service.SourceReport> sourceReports,
-  List<Map<String, dynamic>> coverage,
-) {
-  final Map<String?, Map<int, int>> hitMaps = <String?, Map<int, int>>{};
-  for (final vm_service.SourceReport sourceReport  in sourceReports) {
-    for (final vm_service.SourceReportRange range in sourceReport.ranges!) {
-      final vm_service.SourceReportCoverage? coverage = range.coverage;
-      // Coverage reports may sometimes be null for a Script.
-      if (coverage == null) {
-        continue;
-      }
-      final vm_service.ScriptRef scriptRef = sourceReport.scripts![range.scriptIndex!];
-      final String? uri = scriptRef.uri;
-
-      hitMaps[uri] ??= <int, int>{};
-      final Map<int, int>? hitMap = hitMaps[uri];
-      final List<int>? hits = coverage.hits;
-      final List<int>? misses = coverage.misses;
-      if (hits != null) {
-        for (final int line in hits) {
-          final int current = hitMap![line] ?? 0;
-          hitMap[line] = current + 1;
-        }
-      }
-      if (misses != null) {
-        for (final int line in misses) {
-          hitMap![line] ??= 0;
-        }
-      }
-    }
-  }
-  hitMaps.forEach((String? uri, Map<int, int> hitMap) {
-    coverage.add(_toScriptCoverageJson(uri!, hitMap));
-  });
-}
-
-// Returns a JSON hit map backward-compatible with pre-1.16.0 SDKs.
-Map<String, dynamic> _toScriptCoverageJson(String scriptUri, Map<int, int> hitMap) {
-  final Map<String, dynamic> json = <String, dynamic>{};
-  final List<int> hits = <int>[];
-  hitMap.forEach((int line, int hitCount) {
-    hits.add(line);
-    hits.add(hitCount);
-  });
-  json['source'] = scriptUri;
-  json['script'] = <String, dynamic>{
-    'type': '@Script',
-    'fixedId': true,
-    'id': 'libraries/1/scripts/${Uri.encodeComponent(scriptUri)}',
-    'uri': scriptUri,
-    '_kind': 'library',
-  };
-  json['hits'] = hits;
-  return json;
+  @visibleForTesting FlutterVmService? serviceOverride,
+}) {
+  return coverage.collect(serviceUri, false, false, false, libraryNames, serviceOverrideForTesting: serviceOverride?.service);
 }