63 lines
1.8 KiB
Dart
63 lines
1.8 KiB
Dart
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 = Angel();
|
|
|
|
app.chain([throttleRequests(1, Duration(hours: 1))]).get(
|
|
'/once-per-hour', (req, res) => 'OK');
|
|
|
|
app.chain([throttleRequests(3, Duration(minutes: 1))]).get(
|
|
'/thrice-per-minute', (req, res) => 'OK');
|
|
|
|
client = await connectTo(app);
|
|
});
|
|
|
|
tearDown(() => client.close());
|
|
|
|
test('once per hour', () async {
|
|
// First request within the hour is fine
|
|
var response = await client.get('/once-per-hour', headers: {
|
|
'accept': 'application/json',
|
|
});
|
|
print(response.body);
|
|
expect(response, hasBody('"OK"'));
|
|
|
|
// Second request within an hour? No no no!
|
|
response = await client.get('/once-per-hour', headers: {
|
|
'accept': 'application/json',
|
|
});
|
|
print(response.body);
|
|
expect(response, isAngelHttpException(statusCode: 429));
|
|
});
|
|
|
|
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, hasBody('"OK"'));
|
|
|
|
// Second request within the minute is fine
|
|
response = await client.get('/thrice-per-minute');
|
|
print(response.body);
|
|
expect(response, hasBody('"OK"'));
|
|
|
|
// Third request within the minute is fine
|
|
response = await client.get('/thrice-per-minute');
|
|
print(response.body);
|
|
expect(response, hasBody('"OK"'));
|
|
|
|
// Fourth request within a minute? No no no!
|
|
response = await client.get('/thrice-per-minute', headers: {
|
|
'accept': 'application/json',
|
|
});
|
|
print(response.body);
|
|
expect(response, isAngelHttpException(statusCode: 429));
|
|
});
|
|
}
|