File tree Expand file tree Collapse file tree
aikido_zen/storage/statistics Expand file tree Collapse file tree Original file line number Diff line number Diff 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" , {}))
Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments