Skip to content

Commit 2abbd87

Browse files
committed
introduced enums for alert status
1 parent c0338dc commit 2abbd87

2 files changed

Lines changed: 15 additions & 1 deletion

File tree

linode_api4/objects/monitor.py

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -115,6 +115,18 @@ class DashboardType(StrEnum):
115115
standard = "standard"
116116
custom = "custom"
117117

118+
class AlertStatus(StrEnum):
119+
"""
120+
Enum for supported alert status values.
121+
"""
122+
123+
AlertDefinitionStatusProvisioning = "provisioning"
124+
AlertDefinitionStatusEnabling = "enabling"
125+
AlertDefinitionStatusDisabling = "disabling"
126+
AlertDefinitionStatusEnabled = "enabled"
127+
AlertDefinitionStatusDisabled = "disabled"
128+
AlertDefinitionStatusFailed = "failed"
129+
118130

119131
@dataclass
120132
class Filter(JSONObject):
@@ -374,6 +386,7 @@ class AlertType(StrEnum):
374386
user = "user"
375387

376388

389+
377390
class AlertDefinition(DerivedBase):
378391
"""
379392
Represents an alert definition for a monitor service.

test/integration/models/monitor/test_monitor.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@
88
import pytest
99

1010
from linode_api4 import LinodeClient
11+
from linode_api4.objects.monitor import AlertStatus
1112
from linode_api4.objects import (
1213
AlertDefinition,
1314
ApiError,
@@ -227,7 +228,7 @@ def wait_for_alert_ready(alert_id, service_type: str):
227228
interval = initial_timeout
228229
alert = client.load(AlertDefinition, alert_id, service_type)
229230
while (
230-
getattr(alert, "status", None) == "provisioning"
231+
getattr(alert, "status", None) == AlertStatus.AlertDefinitionStatusEnabling
231232
and (time.time() - start) < timeout
232233
):
233234
time.sleep(interval)

0 commit comments

Comments
 (0)