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));
  });
}