111 lines
3 KiB
Dart
111 lines
3 KiB
Dart
import 'dart:async';
|
|
import 'dart:convert';
|
|
import 'dart:io' show BytesBuilder;
|
|
import 'dart:io';
|
|
|
|
import 'package:angel_container/mirrors.dart';
|
|
import 'package:angel_framework/angel_framework.dart';
|
|
import 'package:angel_framework/http.dart';
|
|
import 'package:mock_request/mock_request.dart';
|
|
import 'package:test/test.dart';
|
|
|
|
Future<List<int>> getBody(MockHttpResponse rs) async {
|
|
var list = await rs.toList();
|
|
var bb = new BytesBuilder();
|
|
list.forEach(bb.add);
|
|
return bb.takeBytes();
|
|
}
|
|
|
|
main() {
|
|
Angel app;
|
|
|
|
setUp(() {
|
|
app = new Angel(reflector: MirrorsReflector());
|
|
app.encoders.addAll(
|
|
{
|
|
'deflate': zlib.encoder,
|
|
'gzip': gzip.encoder,
|
|
},
|
|
);
|
|
|
|
app.get('/hello', (req, res) {
|
|
res
|
|
..useBuffer()
|
|
..write('Hello, world!');
|
|
});
|
|
});
|
|
|
|
tearDown(() => app.close());
|
|
|
|
encodingTests(() => app);
|
|
}
|
|
|
|
void encodingTests(Angel getApp()) {
|
|
group('encoding', () {
|
|
Angel app;
|
|
AngelHttp http;
|
|
|
|
setUp(() {
|
|
app = getApp();
|
|
http = new AngelHttp(app);
|
|
});
|
|
|
|
test('sends plaintext if no accept-encoding', () async {
|
|
var rq = new MockHttpRequest('GET', Uri.parse('/hello'))..close();
|
|
var rs = rq.response;
|
|
await http.handleRequest(rq);
|
|
|
|
var body = await rs.transform(utf8.decoder).join();
|
|
expect(body, 'Hello, world!');
|
|
});
|
|
|
|
test('encodes if wildcard', () async {
|
|
var rq = new MockHttpRequest('GET', Uri.parse('/hello'))
|
|
..headers.set('accept-encoding', '*')
|
|
..close();
|
|
var rs = rq.response;
|
|
await http.handleRequest(rq);
|
|
|
|
var body = await getBody(rs);
|
|
print(rs.headers);
|
|
expect(rs.headers.value('content-encoding'), 'deflate');
|
|
expect(body, zlib.encode(utf8.encode('Hello, world!')));
|
|
});
|
|
|
|
test('encodes if wildcard + multiple', () async {
|
|
var rq = new MockHttpRequest('GET', Uri.parse('/hello'))
|
|
..headers.set('accept-encoding', ['foo', 'bar', '*'])
|
|
..close();
|
|
var rs = rq.response;
|
|
await http.handleRequest(rq);
|
|
|
|
var body = await getBody(rs);
|
|
expect(rs.headers.value('content-encoding'), 'deflate');
|
|
expect(body, zlib.encode(utf8.encode('Hello, world!')));
|
|
});
|
|
|
|
test('encodes if explicit', () async {
|
|
var rq = new MockHttpRequest('GET', Uri.parse('/hello'))
|
|
..headers.set('accept-encoding', 'gzip');
|
|
await rq.close();
|
|
var rs = rq.response;
|
|
await http.handleRequest(rq);
|
|
|
|
var body = await getBody(rs);
|
|
expect(rs.headers.value('content-encoding'), 'gzip');
|
|
expect(body, gzip.encode(utf8.encode('Hello, world!')));
|
|
});
|
|
|
|
test('only uses one encoder', () async {
|
|
var rq = new MockHttpRequest('GET', Uri.parse('/hello'))
|
|
..headers.set('accept-encoding', ['gzip', 'deflate']);
|
|
await rq.close();
|
|
var rs = rq.response;
|
|
await http.handleRequest(rq);
|
|
|
|
var body = await getBody(rs);
|
|
expect(rs.headers.value('content-encoding'), 'gzip');
|
|
expect(body, gzip.encode(utf8.encode('Hello, world!')));
|
|
});
|
|
});
|
|
}
|