Skip to content

Commit dd77188

Browse files
committed
rate_limited -> rateLimited
1 parent ed82846 commit dd77188

2 files changed

Lines changed: 7 additions & 7 deletions

File tree

aikido_zen/storage/statistics/__init__.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ def get_record(self):
4343
"endedAt": current_time,
4444
"requests": {
4545
"total": self.total_hits,
46-
"rate_limited": self.rate_limited_hits,
46+
"rateLimited": self.rate_limited_hits,
4747
"aborted": 0, # statistic currently not in use
4848
"attacksDetected": {
4949
"total": self.attacks_detected,
@@ -56,7 +56,7 @@ def get_record(self):
5656
def import_from_record(self, record):
5757
attacks_detected = record.get("requests", {}).get("attacksDetected", {})
5858
self.total_hits += record.get("requests", {}).get("total", 0)
59-
self.rate_limited_hits += record.get("requests", {}).get("rate_limited", 0)
59+
self.rate_limited_hits += record.get("requests", {}).get("rateLimited", 0)
6060
self.attacks_detected += attacks_detected.get("total", 0)
6161
self.attacks_blocked += attacks_detected.get("blocked", 0)
6262
self.operations.update(record.get("operations", {}))

aikido_zen/storage/statistics/init_test.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,7 @@ def test_get_record(monkeypatch):
7979
assert record["startedAt"] == stats.started_at
8080
assert record["endedAt"] == mock_time
8181
assert record["requests"]["total"] == 10
82-
assert record["requests"]["rate_limited"] == 2
82+
assert record["requests"]["rateLimited"] == 2
8383
assert record["requests"]["aborted"] == 0
8484
assert record["requests"]["attacksDetected"]["total"] == 5
8585
assert record["requests"]["attacksDetected"]["blocked"] == 3
@@ -100,7 +100,7 @@ def test_import_from_record():
100100
record = {
101101
"requests": {
102102
"total": 10,
103-
"rate_limited": 5,
103+
"rateLimited": 5,
104104
"attacksDetected": {
105105
"total": 5,
106106
"blocked": 3,
@@ -157,7 +157,7 @@ def test_multiple_imports(stats):
157157
record1 = {
158158
"requests": {
159159
"total": 10,
160-
"rate_limited": 20,
160+
"rateLimited": 20,
161161
"attacksDetected": {
162162
"total": 5,
163163
"blocked": 3,
@@ -174,7 +174,7 @@ def test_multiple_imports(stats):
174174
record2 = {
175175
"requests": {
176176
"total": 20,
177-
"rate_limited": 5,
177+
"rateLimited": 5,
178178
"attacksDetected": {
179179
"total": 10,
180180
"blocked": 7,
@@ -276,7 +276,7 @@ def test_rate_limit_in_get_record():
276276
stats.on_rate_limit()
277277

278278
record = stats.get_record()
279-
assert record["requests"]["rate_limited"] == 3
279+
assert record["requests"]["rateLimited"] == 3
280280
assert record["requests"]["total"] == 10
281281

282282

0 commit comments

Comments
 (0)