Add Server on top of client support.

Author: gspencergoog

packages/a2a_dart/test/client/a2a_client_compliance_test.dart

@@ -1,4 +1,5 @@
 import 'package:a2a_dart/a2a_dart.dart';
+import 'package:a2a_dart/src/core/push_notification.dart';
 import 'package:test/test.dart';
 
 import '../fakes.dart';

packages/a2a_dart/test/core/data_models_test.dart

@@ -3,6 +3,7 @@
 // found in the LICENSE file.
 
 import 'package:a2a_dart/a2a_dart.dart';
+import 'package:a2a_dart/src/core/push_notification.dart';
 import 'package:test/test.dart';
 
 void main() {

packages/a2a_dart/test/fakes.dart

@@ -2,71 +2,161 @@
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
+import 'dart:async';
 import 'dart:convert';
 
 import 'package:a2a_dart/a2a_dart.dart';
+import 'package:a2a_dart/src/core/push_notification.dart';
+import 'package:http/http.dart';
 
-import 'package:http/http.dart' as http;
+class FakeTransport extends Transport {
+  Map<String, Object?> response;
+  final Stream<Map<String, Object?>>? stream;
+  @override
+  final Map<String, String> authHeaders;
 
-class FakeHttpClient implements http.Client {
-  final Map<String, Object?> response;
-  final int statusCode;
+  FakeTransport({
+    required this.response,
+    this.stream,
+    this.authHeaders = const {},
+  });
 
-  FakeHttpClient(this.response, {this.statusCode = 200});
+  @override
+  Future<Map<String, Object?>> send(
+    Map<String, Object?> request, {
+    String path = '/rpc',
+  }) async {
+    return response;
+  }
+
+  @override
+  Stream<Map<String, Object?>> sendStream(Map<String, Object?> request) {
+    if (stream == null) {
+      throw UnimplementedError();
+    }
+    return stream!;
+  }
 
   @override
-  Future<http.Response> get(Uri url, {Map<String, String>? headers}) async {
-    return http.Response(jsonEncode(response), statusCode);
+  void close() {}
+
+  @override
+  Future<Map<String, Object?>> get(
+    String path, {
+    Map<String, String>? headers,
+  }) async {
+    return response;
   }
+}
+
+class FakeHttpClient implements Client {
+  final Map<String, Object?> response;
+  final int statusCode;
+
+  FakeHttpClient(this.response, {this.statusCode = 200});
 
   @override
-  Future<http.Response> post(
+  Future<Response> post(
     Uri url, {
     Map<String, String>? headers,
     Object? body,
     Encoding? encoding,
   }) async {
-    return http.Response(jsonEncode(response), statusCode);
+    return Response(
+      jsonEncode(response),
+      statusCode,
+      headers: {'content-type': 'application/json'},
+    );
   }
 
+  @override
+  Future<Response> get(Uri url, {Map<String, String>? headers}) async {
+    return Response(
+      jsonEncode(response),
+      statusCode,
+      headers: {'content-type': 'application/json'},
+    );
+  }
+
+  @override
+  void close() {}
+
   @override
   dynamic noSuchMethod(Invocation invocation) => super.noSuchMethod(invocation);
 }
 
-class FakeTransport implements Transport {
+class FakeTaskManager implements TaskManager {
+  Task? taskToReturn;
+  final Map<String, Map<String, PushNotificationConfig>> _pushConfigs = {};
+
+  FakeTaskManager({this.taskToReturn});
+
+  void ensureTaskExists(String taskId) {
+    if (taskToReturn?.id != taskId) {
+      throw const A2AException.taskNotFound(message: 'Task not found');
+    }
+  }
+
   @override
-  Map<String, String> authHeaders;
+  Future<Task> createTask([Message? message]) async {
+    return taskToReturn!;
+  }
 
-  final Map<String, Object?> response;
-  final Stream<Map<String, Object?>> stream;
+  @override
+  Future<void> addEvent(String taskId, Event event) async {}
 
-  FakeTransport({
-    required this.response,
-    Stream<Map<String, Object?>>? stream,
-    this.authHeaders = const {},
-  }) : stream = stream ?? Stream.value(response);
+  @override
+  Future<void> updateTask(Task task) async {}
 
   @override
-  Future<Map<String, Object?>> get(
-    String path, {
-    Map<String, String> headers = const {},
-  }) async {
-    return response;
+  Future<Task?> getTask(String id) async {
+    return taskToReturn;
   }
 
   @override
-  Future<Map<String, Object?>> send(
-    Map<String, Object?> request, {
-    String path = '/rpc',
-  }) async {
-    return response;
+  Future<Task?> cancelTask(String id) async {
+    return taskToReturn;
   }
 
   @override
-  Stream<Map<String, Object?>> sendStream(Map<String, Object?> request) {
-    return stream;
+  Stream<Event> resubscribeToTask(String id) {
+    return Stream.fromIterable([]);
   }
 
   @override
-  void close() {}
+  Future<ListTasksResult> listTasks(ListTasksParams params) async {
+    return ListTasksResult(
+      tasks: [if (taskToReturn != null) taskToReturn!],
+      totalSize: taskToReturn == null ? 0 : 1,
+      pageSize: 1,
+      nextPageToken: '',
+    );
+  }
+
+  @override
+  Future<void> setPushNotificationConfig(
+    String taskId,
+    PushNotificationConfig config,
+  ) async {
+    (_pushConfigs[taskId] ??= {})[config.id!] = config;
+  }
+
+  @override
+  Future<PushNotificationConfig?> getPushNotificationConfig(
+    String taskId,
+    String configId,
+  ) async => _pushConfigs[taskId]?[configId];
+
+  @override
+  Future<List<PushNotificationConfig>> listPushNotificationConfigs(
+    String taskId,
+  ) async => _pushConfigs[taskId]?.values.toList() ?? [];
+
+  @override
+  Future<void> deletePushNotificationConfig(
+    String taskId,
+    String configId,
+  ) async {
+    _pushConfigs[taskId]?.remove(configId);
+  }
 }

packages/a2a_dart/test/server/push_config_handlers_test.dart

@@ -42,7 +42,7 @@ void main() {
       getHandler = GetPushConfigHandler(fakeTaskManager);
       listHandler = ListPushConfigsHandler(fakeTaskManager);
       deleteHandler = DeletePushConfigHandler(fakeTaskManager);
-      fakeTaskManager.ensureTaskExists(testTask);
+      fakeTaskManager.ensureTaskExists(testTask.id);
     });
 
     group('SetPushConfigHandler', () {

packages/a2a_dart/test/server/resubscribe_handler_test.dart

@@ -2,8 +2,6 @@
 // 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:a2a_dart/a2a_dart.dart';
 import 'package:test/test.dart';
 
@@ -22,9 +20,8 @@ void main() {
       );
       taskManager = FakeTaskManager(
         taskToReturn: task,
-        stream: Stream.value({}),
       );
-      await (taskManager as FakeTaskManager).ensureTaskExists(task);
+      (taskManager as FakeTaskManager).ensureTaskExists(task.id);
       handler = ResubscribeHandler(taskManager);
 
       final result = await handler.handle({'id': task.id});