platform/test/throttle_test.dart

62 lines
1.6 KiB
Dart
Raw Normal View History

2017-01-14 00:45:35 +00:00
import 'package:angel_framework/angel_framework.dart';
import 'package:angel_security/angel_security.dart';
import 'package:angel_test/angel_test.dart';
import 'package:test/test.dart';
main() {
Angel app;
TestClient client;
setUp(() async {
app = new Angel();
app
.chain(throttleRequests(1, new Duration(hours: 1)))
.get('/once-per-hour', 'OK');
2017-01-28 20:29:20 +00:00
app
.chain(throttleRequests(3, new Duration(minutes: 1)))
.get('/thrice-per-minute', 'OK');
2017-01-14 00:45:35 +00:00
client = await connectTo(app);
});
tearDown(() => client.close());
2017-01-28 20:29:20 +00:00
test('once per hour', () async {
2017-01-14 00:45:35 +00:00
// First request within the hour is fine
var response = await client.get('/once-per-hour');
print(response.body);
expect(response.body, contains('OK'));
// Second request within an hour? No no no!
response = await client.get('/once-per-hour');
print(response.body);
expect(response, hasStatus(429));
});
2017-01-28 20:29:20 +00:00
test('thrice per minute', () async {
// First request within the minute is fine
var response = await client.get('/thrice-per-minute');
print(response.body);
expect(response.body, contains('OK'));
// Second request within the minute is fine
response = await client.get('/thrice-per-minute');
print(response.body);
expect(response.body, contains('OK'));
// Third request within the minute is fine
response = await client.get('/thrice-per-minute');
print(response.body);
expect(response.body, contains('OK'));
// Fourth request within a minute? No no no!
response = await client.get('/thrice-per-minute');
print(response.body);
expect(response, hasStatus(429));
});
2017-01-14 00:45:35 +00:00
}