Skip to content

Commit 4e26ceb

Browse files
committed
Update SYNC_DATA test cases to include rateLimited field
1 parent dd77188 commit 4e26ceb

1 file changed

Lines changed: 7 additions & 0 deletions

File tree

aikido_zen/background_process/commands/sync_data_test.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -54,6 +54,7 @@ def test_process_sync_data_initialization(setup_connection_manager):
5454
"endedAt": 1,
5555
"requests": {
5656
"total": 10,
57+
"rateLimited": 0,
5758
"aborted": 0,
5859
"attacksDetected": {
5960
"total": 5,
@@ -94,6 +95,7 @@ def test_process_sync_data_initialization(setup_connection_manager):
9495
"aborted": 0,
9596
"attacksDetected": {"blocked": 0, "total": 5},
9697
"total": 10,
98+
"rateLimited": 0,
9799
}
98100

99101
# Check that the return value is correct
@@ -135,6 +137,7 @@ def test_process_sync_data_with_last_updated_at_below_zero(setup_connection_mana
135137
"endedAt": 1,
136138
"requests": {
137139
"total": 10,
140+
"rateLimited": 0,
138141
"aborted": 0,
139142
"attacksDetected": {
140143
"total": 5,
@@ -167,6 +170,7 @@ def test_process_sync_data_with_last_updated_at_below_zero(setup_connection_mana
167170
"aborted": 0,
168171
"attacksDetected": {"blocked": 0, "total": 5},
169172
"total": 10,
173+
"rateLimited": 0,
170174
}
171175
assert connection_manager.middleware_installed == True
172176
assert len(connection_manager.hostnames.as_array()) == 0
@@ -199,6 +203,7 @@ def test_process_sync_data_existing_route_and_hostnames(setup_connection_manager
199203
"endedAt": 1,
200204
"requests": {
201205
"total": 5,
206+
"rateLimited": 0,
202207
"aborted": 0,
203208
"attacksDetected": {
204209
"total": 5,
@@ -227,6 +232,7 @@ def test_process_sync_data_existing_route_and_hostnames(setup_connection_manager
227232
"endedAt": 1,
228233
"requests": {
229234
"total": 15,
235+
"rateLimited": 0,
230236
"aborted": 0,
231237
"attacksDetected": {
232238
"total": 5,
@@ -251,6 +257,7 @@ def test_process_sync_data_existing_route_and_hostnames(setup_connection_manager
251257
"aborted": 0,
252258
"attacksDetected": {"blocked": 0, "total": 10},
253259
"total": 20,
260+
"rateLimited": 0,
254261
}
255262
assert connection_manager.middleware_installed == False
256263
assert connection_manager.hostnames.as_array() == [

0 commit comments

Comments
 (0)