packages/devtools_app/test/service/timeline_streams_test.dart

// Copyright 2021 The Flutter Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file or at https://developers.google.com/open-source/licenses/bsd.

@TestOn('vm')
library;

import 'package:devtools_app/devtools_app.dart';
import 'package:devtools_shared/devtools_test_utils.dart';
import 'package:flutter_test/flutter_test.dart';

import '../test_infra/flutter_test_driver.dart' show FlutterRunConfiguration;
import '../test_infra/flutter_test_environment.dart';

void main() {
  final env = FlutterTestEnvironment(
    const FlutterRunConfiguration(withDebugger: true),
  );

  group('TimelineStreamManager', () {
    tearDownAll(() async {
      await env.tearDownEnvironment(force: true);
    });

    test('timeline streams initialized on vm service opened', () async {
      await env.setupEnvironment();

      // Await a short delay to make sure the timelineStreamManager is done
      // initializing.
      await delay();

      expect(serviceConnection.serviceManager.service, equals(env.service));
      expect(serviceConnection.timelineStreamManager, isNotNull);
      expect(serviceConnection.timelineStreamManager.basicStreams, isNotEmpty);
      expect(
        serviceConnection.timelineStreamManager.advancedStreams,
        isNotEmpty,
      );

      await env.tearDownEnvironment();
    }, timeout: const Timeout.factor(4));

    test('notifies on stream change', () async {
      await env.setupEnvironment();

      final initialStreams =
          serviceConnection.timelineStreamManager.recordedStreams;
      expect(
        initialStreams.map((stream) => stream.name).toList(),
        equals(['Dart', 'Embedder', 'GC', 'Microtask']),
      );

      await serviceConnection.serviceManager.service!.setVMTimelineFlags([
        TimelineStreamManager.apiTimelineStream,
        TimelineStreamManager.compilerTimelineStream,
        TimelineStreamManager.isolateTimelineStream,
      ]);
      final newStreams =
          serviceConnection.timelineStreamManager.recordedStreams;
      expect(
        newStreams.map((stream) => stream.name).toList(),
        equals(['API', 'Compiler', 'Isolate']),
      );

      await env.tearDownEnvironment();
    }, timeout: const Timeout.factor(4));
  });
}