More HttpClientResponse implements Stream<Uint8List> fixes (flutter…

…#35149) dart-lang/sdk#36900
johnsonmh · Jul 30, 2019 · 77e40db · 77e40db
1 parent 95a700c
commit 77e40db
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/packages/flutter_tools/test/base/net_test.dart b/packages/flutter_tools/test/base/net_test.dart
@@ -233,7 +233,7 @@ class MockHttpClientRequest implements io.HttpClientRequest {
   }
 }
 
-class MockHttpClientResponse extends Stream<List<int>> implements io.HttpClientResponse {
+class MockHttpClientResponse implements io.HttpClientResponse {
   MockHttpClientResponse(this.statusCode);
 
   @override
@@ -253,6 +253,11 @@ class MockHttpClientResponse extends Stream<List<int>> implements io.HttpClientR
       .listen(onData, onError: onError, onDone: onDone, cancelOnError: cancelOnError);
   }
 
+  @override
+  Future<dynamic> forEach(void Function(Uint8List element) action) {
+    return Future<void>.error(const io.SocketException('test'));
+  }
+
   @override
   dynamic noSuchMethod(Invocation invocation) {
     throw 'io.HttpClientResponse - $invocation';

diff --git a/packages/flutter_tools/test/commands/create_test.dart b/packages/flutter_tools/test/commands/create_test.dart
@@ -1286,7 +1286,7 @@ class MockHttpClientRequest implements HttpClientRequest {
   }
 }
 
-class MockHttpClientResponse extends Stream<List<int>> implements HttpClientResponse {
+class MockHttpClientResponse implements HttpClientResponse {
   MockHttpClientResponse(this.statusCode, {this.result});
 
   @override
@@ -1313,6 +1313,12 @@ class MockHttpClientResponse extends Stream<List<int>> implements HttpClientResp
       .listen(onData, onError: onError, onDone: onDone, cancelOnError: cancelOnError);
   }
 
+  @override
+  Future<dynamic> forEach(void Function(Uint8List element) action) {
+    action(Uint8List.fromList(result.codeUnits));
+    return Future<void>.value();
+  }
+
   @override
   dynamic noSuchMethod(Invocation invocation) {
     throw 'io.HttpClientResponse - $invocation';