pkg/vm_service/test/coverage_leaf_function_test.dart

// Copyright (c) 2019, the Dart project authors.  Please see the AUTHORS file
// for details. All rights reserved. Use of this source code is governed by a
// BSD-style license that can be found in the LICENSE file.

import 'package:test/test.dart';
import 'package:vm_service/vm_service.dart';

import 'common/service_test_common.dart';
import 'coverage_leaf_function_lib.dart' as testee_lib;

bool allRangesCompiled(coverage) {
  for (int i = 0; i < coverage['ranges'].length; i++) {
    if (!coverage['ranges'][i]['compiled']) {
      return false;
    }
  }
  return true;
}

IsolateTest coverageTest(
  Map<String, dynamic> expectedRange, {
  required bool reportLines,
}) {
  return (VmService service, IsolateRef isolateRef) async {
    final isolateId = isolateRef.id!;
    final isolate = await service.getIsolate(isolateId);
    final stack = await service.getStack(isolateId);

    // Make sure we are in the right place.
    expect(stack.frames!.length, greaterThanOrEqualTo(1));
    expect(stack.frames![0].function!.name, 'testFunction');

    final root = await service.getObject(
        isolateId,
        isolate.libraries!
            .firstWhere((l) => l.uri!.contains('coverage_leaf_function_lib'))
            .id!) as Library;
    final FuncRef funcRef =
        root.functions!.singleWhere((f) => f.name == 'leafFunction');
    final Func func = await service.getObject(isolateId, funcRef.id!) as Func;
    final location = func.location!;

    final report = await service.getSourceReport(
      isolateId,
      [SourceReportKind.kCoverage],
      scriptId: location.script!.id,
      tokenPos: location.tokenPos,
      endTokenPos: location.endTokenPos,
      forceCompile: true,
      reportLines: reportLines,
    );
    expect(report.ranges!.length, 1);
    expect(report.ranges![0].toJson(), expectedRange);
    expect(report.scripts!.length, 1);
    expect(
      report.scripts![0].uri,
      endsWith('coverage_leaf_function_lib.dart'),
    );
  };
}

void main([args = const <String>[]]) =>
    IsolateTestHarness('coverage_leaf_function_lib.dart', args)
        .hasStoppedAtBreakpoint()
        .addCustomTest(
          coverageTest(
            {
              'scriptIndex': 0,
              'startPos': 277,
              'endPos': 327,
              'compiled': true,
              'coverage': {
                'hits': [],
                'misses': [277],
              },
            },
            reportLines: false,
          ),
        )
        .addCustomTest(
          coverageTest(
            {
              'scriptIndex': 0,
              'startPos': 277,
              'endPos': 327,
              'compiled': true,
              'coverage': {
                'hits': [],
                'misses': [8],
              },
            },
            reportLines: true,
          ),
        )
        .resumeIsolate()
        .hasStoppedAtBreakpoint()
        .addCustomTest(
          coverageTest(
            {
              'scriptIndex': 0,
              'startPos': 277,
              'endPos': 327,
              'compiled': true,
              'coverage': {
                'hits': [277],
                'misses': [],
              },
            },
            reportLines: false,
          ),
        )
        .addCustomTest(
          coverageTest(
            {
              'scriptIndex': 0,
              'startPos': 277,
              'endPos': 327,
              'compiled': true,
              'coverage': {
                'hits': [8],
                'misses': [],
              },
            },
            reportLines: true,
          ),
        )
        .run(testeeMain: testee_lib.main);