| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2014, the Dart project authors. Please see the AUTHORS file | |
| 2 // for details. All rights reserved. Use of this source code is governed by a | |
| 3 // BSD-style license that can be found in the LICENSE file. | |
| 4 | |
| 5 library json_rpc_2.test.client.utils; | |
| 6 | |
| 7 import 'dart:async'; | |
| 8 import 'dart:convert'; | |
| 9 | |
| 10 import 'package:json_rpc_2/json_rpc_2.dart' as json_rpc; | |
| 11 import 'package:json_rpc_2/error_code.dart' as error_code; | |
| 12 import 'package:unittest/unittest.dart'; | |
| 13 | |
| 14 /// A controller used to test a [json_rpc.Client]. | |
| 15 class ClientController { | |
| 16 /// The controller for the client's response stream. | |
| 17 final _responseController = new StreamController<String>(); | |
| 18 | |
| 19 /// The controller for the client's request sink. | |
| 20 final _requestController = new StreamController<String>(); | |
| 21 | |
| 22 /// The client. | |
| 23 json_rpc.Client get client => _client; | |
| 24 json_rpc.Client _client; | |
| 25 | |
| 26 ClientController() { | |
| 27 _client = new json_rpc.Client( | |
| 28 _responseController.stream, _requestController.sink); | |
| 29 _client.listen(); | |
| 30 } | |
| 31 | |
| 32 /// Expects that the client will send a request. | |
| 33 /// | |
| 34 /// The request is passed to [callback], which can return a response. If it | |
| 35 /// returns a String, that's sent as the response directly. If it returns | |
| 36 /// null, no response is sent. Otherwise, the return value is encoded and sent | |
| 37 /// as the response. | |
| 38 void expectRequest(callback(request)) { | |
| 39 expect(_requestController.stream.first.then((request) { | |
| 40 return callback(JSON.decode(request)); | |
| 41 }).then((response) { | |
| 42 if (response == null) return; | |
| 43 if (response is! String) response = JSON.encode(response); | |
| 44 _responseController.add(response); | |
| 45 }), completes); | |
| 46 } | |
| 47 | |
| 48 /// Sends [response], a decoded response, to [client]. | |
| 49 Future sendResponse(response) => sendJsonResponse(JSON.encode(response)); | |
| 50 | |
| 51 /// Sends [response], a JSON-encoded response, to [client]. | |
| 52 Future sendJsonResponse(String request) => _responseController.add(request); | |
| 53 } | |
| 54 | |
| 55 /// Returns a [Future] that completes after pumping the event queue [times] | |
| 56 /// times. By default, this should pump the event queue enough times to allow | |
| 57 /// any code to run, as long as it's not waiting on some external event. | |
| 58 Future pumpEventQueue([int times = 20]) { | |
| 59 if (times == 0) return new Future.value(); | |
| 60 // We use a delayed future to allow microtask events to finish. The | |
| 61 // Future.value or Future() constructors use scheduleMicrotask themselves and | |
| 62 // would therefore not wait for microtask callbacks that are scheduled after | |
| 63 // invoking this method. | |
| 64 return new Future.delayed(Duration.ZERO, () => pumpEventQueue(times - 1)); | |
| 65 } | |
| OLD | NEW |