Fix test_datadog_notify_calls_requests_post

getredash · Sep 24, 2023 · 1ecc4ef · 1ecc4ef
1 parent 1acd7bb
commit 1ecc4ef
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/tests/handlers/test_destinations.py b/tests/handlers/test_destinations.py
@@ -251,13 +251,11 @@ def test_webex_notify_calls_requests_post():
 
 
 def test_datadog_notify_calls_requests_post():
-    alert = mock.Mock(spec_set=["id", "name", "options", "render_template"])
+    alert = mock.Mock(spec_set=["id", "name", "custom_subject", "custom_body", "render_template"])
     alert.id = 1
     alert.name = "Test Alert"
-    alert.options = {
-        "custom_subject": "Test custom subject",
-        "custom_body": "Test custom body",
-    }
+    alert.custom_subject = "Test custom subject"
+    alert.custom_body = "Test custom body"
     alert.render_template = mock.Mock(return_value={"Rendered": "template"})
     query = mock.Mock()
     query.id = 1