pkgs/test_api/lib/src/suite_channel_manager.dart

// Copyright (c) 2018, 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:stream_channel/stream_channel.dart';

/// A class that connects incoming and outgoing channels with the same names.
class SuiteChannelManager {
  /// Connections from the test runner that have yet to connect to corresponding
  /// calls to [connectOut].
  final _incomingConnections = <String, StreamChannel<Object?>>{};

  /// Connections from calls to [connectOut] that have yet to connect to
  /// corresponding connections from the test runner.
  final _outgoingConnections = <String, StreamChannelCompleter<Object?>>{};

  /// The channel names that have already been used.
  final _names = <String>{};

  /// Creates a connection to the test runnner's channel with the given [name].
  StreamChannel<Object?> connectOut(String name) {
    if (_incomingConnections.containsKey(name)) {
      return (_incomingConnections[name])!;
    } else if (_names.contains(name)) {
      throw StateError('Duplicate suiteChannel() connection "$name".');
    } else {
      _names.add(name);
      var completer = StreamChannelCompleter<Object?>();
      _outgoingConnections[name] = completer;
      return completer.channel;
    }
  }

  /// Connects [channel] to this worker's channel with the given [name].
  void connectIn(String name, StreamChannel<Object?> channel) {
    if (_outgoingConnections.containsKey(name)) {
      _outgoingConnections.remove(name)!.setChannel(channel);
    } else if (_incomingConnections.containsKey(name)) {
      throw StateError('Duplicate RunnerSuite.channel() connection "$name".');
    } else {
      _incomingConnections[name] = channel;
    }
  }
}