@@ -84,7 +84,7 @@ def test_alert_in_active_maintenance_window(
84
84
mock_session , active_maintenance_window_rule , alert_dto
85
85
):
86
86
# Simulate the query to return the active maintenance_window
87
- mock_session .query .return_value .filter .return_value .filter .return_value .filter .return_value .all .return_value = [
87
+ mock_session .query .return_value .filter .return_value .filter .return_value .filter .return_value .filter . return_value . all .return_value = [
88
88
active_maintenance_window_rule
89
89
]
90
90
@@ -100,7 +100,7 @@ def test_alert_in_active_maintenance_window_with_suppress(
100
100
mock_session , active_maintenance_window_rule_with_suppression_on , alert_dto
101
101
):
102
102
# Simulate the query to return the active maintenance_window
103
- mock_session .query .return_value .filter .return_value .filter .return_value .filter .return_value .all .return_value = [
103
+ mock_session .query .return_value .filter .return_value .filter .return_value .filter .return_value .filter . return_value . all .return_value = [
104
104
active_maintenance_window_rule_with_suppression_on
105
105
]
106
106
@@ -117,7 +117,7 @@ def test_alert_not_in_expired_maintenance_window(
117
117
mock_session , expired_maintenance_window_rule , alert_dto
118
118
):
119
119
# Simulate the query to return the expired maintenance_window
120
- mock_session .query .return_value .filter .return_value .filter .return_value .filter .return_value .all .return_value = [
120
+ mock_session .query .return_value .filter .return_value .filter .return_value .filter .return_value .filter . return_value . all .return_value = [
121
121
expired_maintenance_window_rule
122
122
]
123
123
@@ -132,7 +132,7 @@ def test_alert_not_in_expired_maintenance_window(
132
132
133
133
def test_alert_in_no_maintenance_window (mock_session , alert_dto ):
134
134
# Simulate the query to return no maintenance_windows
135
- mock_session .query .return_value .filter .return_value .filter .return_value .filter .return_value .all .return_value = (
135
+ mock_session .query .return_value .filter .return_value .filter .return_value .filter .return_value .filter . return_value . all .return_value = (
136
136
[]
137
137
)
138
138
@@ -149,7 +149,7 @@ def test_alert_in_maintenance_window_with_non_matching_cel(
149
149
):
150
150
# Modify the cel_query so that the alert won't match
151
151
active_maintenance_window_rule .cel_query = 'source == "other-source"'
152
- mock_session .query .return_value .filter .return_value .filter .return_value .filter .return_value .all .return_value = [
152
+ mock_session .query .return_value .filter .return_value .filter .return_value .filter .return_value .filter . return_value . all .return_value = [
153
153
active_maintenance_window_rule
154
154
]
155
155
@@ -167,7 +167,7 @@ def test_alert_ignored_due_to_resolved_status(
167
167
# Set the alert status to RESOLVED
168
168
alert_dto .status = "resolved"
169
169
170
- mock_session .query .return_value .filter .return_value .filter .return_value .filter .return_value .all .return_value = [
170
+ mock_session .query .return_value .filter .return_value .filter .return_value .filter .return_value .filter . return_value . all .return_value = [
171
171
active_maintenance_window_rule
172
172
]
173
173
@@ -186,7 +186,7 @@ def test_alert_ignored_due_to_acknowledged_status(
186
186
# Set the alert status to ACKNOWLEDGED
187
187
alert_dto .status = "acknowledged"
188
188
189
- mock_session .query .return_value .filter .return_value .filter .return_value .filter .return_value .all .return_value = [
189
+ mock_session .query .return_value .filter .return_value .filter .return_value .filter .return_value .filter . return_value . all .return_value = [
190
190
active_maintenance_window_rule
191
191
]
192
192
@@ -202,7 +202,7 @@ def test_alert_ignored_due_to_acknowledged_status(
202
202
def test_alert_with_missing_cel_field (mock_session , active_maintenance_window_rule , alert_dto ):
203
203
# Modify the cel_query to reference a non-existent field
204
204
active_maintenance_window_rule .cel_query = 'alertname == "test-alert"'
205
- mock_session .query .return_value .filter .return_value .filter .return_value .filter .return_value .all .return_value = [
205
+ mock_session .query .return_value .filter .return_value .filter .return_value .filter .return_value .filter . return_value . all .return_value = [
206
206
active_maintenance_window_rule
207
207
]
208
208
@@ -220,7 +220,7 @@ def test_alert_not_ignored_due_to_custom_status(
220
220
):
221
221
# Set the alert status to RESOLVED
222
222
223
- mock_session .query .return_value .filter .return_value .filter .return_value .filter .return_value .all .return_value = [
223
+ mock_session .query .return_value .filter .return_value .filter .return_value .filter .return_value .filter . return_value . all .return_value = [
224
224
active_maintenance_window_rule_custom_ignore
225
225
]
226
226
0 commit comments