Test that both new and old style messages are handled the same way

release-engineering · Jan 27, 2025 · 2218859 · 2218859
1 parent a305ae3
commit 2218859
Showing 1 changed file with 17 additions and 3 deletions.
diff --git a/tests/test_main.py b/tests/test_main.py
@@ -339,15 +339,29 @@ def test_listen(self, mock_fedmsg, mock_handlers_issue, mock_handle_msg):
                 "dummy",
                 "d.d.d.github.issue.comment",
                 self.old_style_mock_message,
-            )
+            ),
+            (
+                "dummy",
+                "dummy",
+                "d.d.d.github.issue.comment",
+                self.new_style_mock_message,
+            ),
         ]
 
         # Call the function
         m.listen(self.mock_config)
 
         # Assert everything was called correctly
-        mock_handle_msg.assert_called_with(
-            self.mock_message_body, "github.issue.comment", self.mock_config
+        # It should be called twice, once for the old style message and once for the new.
+        mock_handle_msg.assert_has_calls(
+            [
+                mock.call(
+                    self.mock_message_body, "github.issue.comment", self.mock_config
+                ),
+                mock.call(
+                    self.mock_message_body, "github.issue.comment", self.mock_config
+                ),
+            ]
         )
 
     @mock.patch(PATH + "send_mail")