Updated tests in victorops_test.py to accommodate new fields passed…

… to alert payload. * Needed by jertel#329
ChristophShyper · Jul 20, 2021 · 81a1f8b · 81a1f8b
1 parent 4c74ab5
commit 81a1f8b
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/tests/alerters/victorops_test.py b/tests/alerters/victorops_test.py
@@ -36,7 +36,9 @@ def test_victorops(caplog):
         'message_type': rule['victorops_message_type'],
         'entity_display_name': rule['victorops_entity_display_name'],
         'monitoring_tool': 'ElastAlert',
-        'state_message': 'Test VictorOps Rule\n\n@timestamp: 2021-01-01T00:00:00\nsomefield: foobarbaz\n'
+        'state_message': 'Test VictorOps Rule\n\n@timestamp: 2021-01-01T00:00:00\nsomefield: foobarbaz\n',
+        '@timestamp': '2021-01-01T00:00:00',
+        'somefield': 'foobarbaz'
     }
 
     mock_post_request.assert_called_once_with(
@@ -76,7 +78,9 @@ def test_victorops_proxy():
         'message_type': rule['victorops_message_type'],
         'entity_display_name': rule['victorops_entity_display_name'],
         'monitoring_tool': 'ElastAlert',
-        'state_message': 'Test VictorOps Rule\n\n@timestamp: 2021-01-01T00:00:00\nsomefield: foobarbaz\n'
+        'state_message': 'Test VictorOps Rule\n\n@timestamp: 2021-01-01T00:00:00\nsomefield: foobarbaz\n',
+        '@timestamp': '2021-01-01T00:00:00',
+        'somefield': 'foobarbaz'
     }
 
     mock_post_request.assert_called_once_with(
@@ -141,7 +145,9 @@ def test_victorops_entity_id():
         'entity_display_name': rule['victorops_entity_display_name'],
         'monitoring_tool': 'ElastAlert',
         'state_message': 'Test VictorOps Rule\n\n@timestamp: 2021-01-01T00:00:00\nsomefield: foobarbaz\n',
-        'entity_id': '12345'
+        'entity_id': '12345',
+        '@timestamp': '2021-01-01T00:00:00',
+        'somefield': 'foobarbaz'
     }
 
     mock_post_request.assert_called_once_with(
@@ -186,7 +192,9 @@ def test_victorops_message_type(message_type, except_message_type):
         'message_type': except_message_type,
         'entity_display_name': rule['victorops_entity_display_name'],
         'monitoring_tool': 'ElastAlert',
-        'state_message': 'Test VictorOps Rule\n\n@timestamp: 2021-01-01T00:00:00\nsomefield: foobarbaz\n'
+        'state_message': 'Test VictorOps Rule\n\n@timestamp: 2021-01-01T00:00:00\nsomefield: foobarbaz\n',
+        '@timestamp': '2021-01-01T00:00:00',
+        'somefield': 'foobarbaz'
     }
 
     mock_post_request.assert_called_once_with(