Index: pkg/gcloud/test/common.dart |
diff --git a/pkg/gcloud/test/common.dart b/pkg/gcloud/test/common.dart |
new file mode 100644 |
index 0000000000000000000000000000000000000000..acac02aad4b514183f17eee1a1cde2f3d591313d |
--- /dev/null |
+++ b/pkg/gcloud/test/common.dart |
@@ -0,0 +1,226 @@ |
+// Copyright (c) 2014, 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 'dart:convert'; |
+ |
+import 'package:crypto/crypto.dart' as crypto; |
+import 'package:http/http.dart' as http; |
+import 'package:http/testing.dart' as http_testing; |
+import 'package:http_parser/http_parser.dart' as http_parser; |
+import 'package:mime/mime.dart' as mime; |
+import 'package:unittest/unittest.dart'; |
+ |
+const CONTENT_TYPE_JSON_UTF8 = 'application/json; charset=utf-8'; |
+ |
+const RESPONSE_HEADERS = const { |
+ 'content-type': CONTENT_TYPE_JSON_UTF8 |
+}; |
+ |
+class MockClient extends http.BaseClient { |
+ final String rootPath; |
+ final Uri rootUri; |
+ |
+ Map<String, Map<Pattern, Function>> mocks = {}; |
+ http_testing.MockClient client; |
+ |
+ MockClient(String rootPath) : |
+ rootPath = rootPath, |
+ rootUri = Uri.parse('https://www.googleapis.com${rootPath}') { |
+ client = new http_testing.MockClient(handler); |
+ } |
+ |
+ void register(String method, Pattern path, |
+ http_testing.MockClientHandler handler) { |
+ var map = mocks.putIfAbsent(method, () => new Map()); |
+ if (path is RegExp) { |
+ map[new RegExp('$rootPath${path.pattern}')] = handler; |
+ } else { |
+ map['$rootPath$path'] = handler; |
+ } |
+ } |
+ |
+ void registerUpload(String method, Pattern path, |
+ http_testing.MockClientHandler handler) { |
+ var map = mocks.putIfAbsent(method, () => new Map()); |
+ map['/upload$rootPath$path'] = handler; |
+ } |
+ |
+ void registerResumableUpload(String method, Pattern path, |
+ http_testing.MockClientHandler handler) { |
+ var map = mocks.putIfAbsent(method, () => new Map()); |
+ map['/resumable/upload$rootPath$path'] = handler; |
+ } |
+ |
+ void clear() { |
+ mocks = {}; |
+ } |
+ |
+ Future<http.Response> handler(http.Request request) { |
+ expect(request.url.host, 'www.googleapis.com'); |
+ var path = request.url.path; |
+ if (mocks[request.method] == null) { |
+ throw 'No mock handler for method ${request.method} found. ' |
+ 'Request URL was: ${request.url}'; |
+ } |
+ var mockHandler; |
+ mocks[request.method].forEach((pattern, handler) { |
+ if (pattern.matchAsPrefix(path) != null) { |
+ mockHandler = handler; |
+ } |
+ }); |
+ if (mockHandler == null) { |
+ throw 'No mock handler for method ${request.method} and path ' |
+ '[$path] found. Request URL was: ${request.url}'; |
+ } |
+ return mockHandler(request); |
+ } |
+ |
+ Future<http.StreamedResponse> send(http.BaseRequest request) { |
+ return client.send(request); |
+ } |
+ |
+ Future<http.Response> respond(response) { |
+ return new Future.value( |
+ new http.Response( |
+ JSON.encode(response.toJson()), |
+ 200, |
+ headers: RESPONSE_HEADERS)); |
+ } |
+ |
+ Future<http.Response> respondEmpty() { |
+ return new Future.value( |
+ new http.Response('', 200, headers: RESPONSE_HEADERS)); |
+ } |
+ |
+ Future<http.Response> respondInitiateResumableUpload(project) { |
+ Map headers = new Map.from(RESPONSE_HEADERS); |
+ headers['location'] = |
+ 'https://www.googleapis.com/resumable/upload$rootPath' |
+ 'b/$project/o?uploadType=resumable&alt=json&' |
+ 'upload_id=AEnB2UqucpaWy7d5cr5iVQzmbQcQlLDIKiClrm0SAX3rJ7UN' |
+ 'Mu5bEoC9b4teJcJUKpqceCUeqKzuoP_jz2ps_dV0P0nT8OTuZQ'; |
+ return new Future.value( |
+ new http.Response('', 200, headers: headers)); |
+ } |
+ |
+ Future<http.Response> respondContinueResumableUpload() { |
+ return new Future.value( |
+ new http.Response('', 308, headers: RESPONSE_HEADERS)); |
+ } |
+ |
+ Future<http.Response> respondBytes(List<int> bytes) { |
+ return new Future.value( |
+ new http.Response.bytes(bytes, 200, headers: RESPONSE_HEADERS)); |
+ } |
+ |
+ Future<http.Response> respondError(statusCode) { |
+ var error = { |
+ 'error': { |
+ 'code': statusCode, |
+ 'message': 'error' |
+ } |
+ }; |
+ return new Future.value( |
+ new http.Response( |
+ JSON.encode(error), statusCode, headers: RESPONSE_HEADERS)); |
+ } |
+ |
+ Future processNormalMediaUpload(http.Request request) { |
+ var completer = new Completer(); |
+ |
+ var contentType = new http_parser.MediaType.parse( |
+ request.headers['content-type']); |
+ expect(contentType.mimeType, 'multipart/related'); |
+ var boundary = contentType.parameters['boundary']; |
+ |
+ var partCount = 0; |
+ var json; |
+ new Stream.fromIterable([request.bodyBytes, [13, 10]]) |
+ .transform(new mime.MimeMultipartTransformer(boundary)) |
+ .listen( |
+ ((mime.MimeMultipart mimeMultipart) { |
+ var contentType = mimeMultipart.headers['content-type']; |
+ partCount++; |
+ if (partCount == 1) { |
+ // First part in the object JSON. |
+ expect(contentType, 'application/json; charset=utf-8'); |
+ mimeMultipart |
+ .transform(UTF8.decoder) |
+ .fold('', (p, e) => '$p$e') |
+ .then((j) => json = j); |
+ } else if (partCount == 2) { |
+ // Second part is the base64 encoded bytes. |
+ mimeMultipart |
+ .transform(ASCII.decoder) |
+ .fold('', (p, e) => '$p$e') |
+ .then(crypto.CryptoUtils.base64StringToBytes) |
+ .then((bytes) { |
+ completer.complete( |
+ new NormalMediaUpload(json, bytes, contentType)); |
+ }); |
+ } else { |
+ // Exactly two parts expected. |
+ throw 'Unexpected part count'; |
+ } |
+ })); |
+ |
+ return completer.future; |
+ } |
+} |
+ |
+class NormalMediaUpload { |
+ final String json; |
+ final List<int> bytes; |
+ final String contentType; |
+ NormalMediaUpload(this.json, this.bytes, this.contentType); |
+} |
+ |
+// Implementation of http.Client which traces all requests and responses. |
+// Mainly useful for local testing. |
+class TraceClient extends http.BaseClient { |
+ final http.Client client; |
+ |
+ TraceClient(this.client); |
+ |
+ Future<http.StreamedResponse> send(http.BaseRequest request) { |
+ print(request); |
+ return request.finalize().toBytes().then((body) { |
+ print('--- START REQUEST ---'); |
+ print(UTF8.decode(body)); |
+ print('--- END REQUEST ---'); |
+ var r = new RequestImpl(request.method, request.url, body); |
+ r.headers.addAll(request.headers); |
+ return client.send(r).then((http.StreamedResponse rr) { |
+ return rr.stream.toBytes().then((body) { |
+ print('--- START RESPONSE ---'); |
+ print(UTF8.decode(body)); |
+ print('--- END RESPONSE ---'); |
+ return new http.StreamedResponse( |
+ new http.ByteStream.fromBytes(body), |
+ rr.statusCode, |
+ headers: rr.headers); |
+ |
+ }); |
+ }); |
+ }); |
+ } |
+ |
+ void close() { |
+ client.close(); |
+ } |
+} |
+ |
+// http.BaseRequest implementationn used by the TraceClient. |
+class RequestImpl extends http.BaseRequest { |
+ final List<int> _body; |
+ |
+ RequestImpl(String method, Uri url, this._body) |
+ : super(method, url); |
+ |
+ http.ByteStream finalize() { |
+ super.finalize(); |
+ return new http.ByteStream.fromBytes(_body); |
+ } |
+} |