pkgs/_macro_server/test/macro_server_test.dart

// Copyright (c) 2024, 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 'dart:async';

import 'package:_macro_client/macro_client.dart';
import 'package:_macro_server/macro_server.dart';
import 'package:_test_macros/declare_x_macro.dart';
import 'package:macro_service/macro_service.dart';
import 'package:test/test.dart';

void main() {
  for (final protocol in [
    Protocol(encoding: 'json'),
    Protocol(encoding: 'binary')
  ]) {
    group('MacroServer using ${protocol.encoding}', () {
      test('serves a macro service', () async {
        final service = TestHostService();
        final server =
            await MacroServer.serve(protocol: protocol, service: service);

        await MacroClient.run(
            // TODO(davidmorgan): this should be negotiated, not set here.
            protocol: protocol,
            endpoint: server.endpoint,
            macros: [DeclareXImplementation()]);

        // Check that the macro sent its description to the host on startup.
        expect((await service.macroStartedRequests.first).macroDescription,
            DeclareXImplementation().description);
      });
    });
  }
}

class TestHostService implements HostService {
  final StreamController<MacroStartedRequest> _macroStartedRequestsController =
      StreamController();
  Stream<MacroStartedRequest> get macroStartedRequests =>
      _macroStartedRequestsController.stream;

  @override
  Future<Response> handle(MacroRequest request) async {
    if (request.type == MacroRequestType.macroStartedRequest) {
      _macroStartedRequestsController.add(request.asMacroStartedRequest);
      return Response.macroStartedResponse(MacroStartedResponse(),
          requestId: request.id);
    }
    return Response.errorResponse(ErrorResponse(error: 'unimplemented'),
        requestId: request.id);
  }
}