fix logstash test error

Signed-off-by: constanca <[email protected]>
elastic · May 29, 2024 · 88bc588 · 88bc588
1 parent e98c561
commit 88bc588
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/shippers/test_logstash.py b/tests/shippers/test_logstash.py
@@ -133,7 +133,7 @@ def test_send_failures(self) -> None:
             logstash_shipper.set_replay_handler(replay_handler)
             event = deepcopy(_dummy_event)
             assert logstash_shipper.send(event) == _EVENT_SENT
-            replay_handler.assert_called_once_with("logstash", {}, event)
+            replay_handler.assert_called_once_with(url, {}, event)
         with self.subTest("Exceeds max retries, replay handler not set"):
             for i in range(_MAX_RETRIES):
                 responses.put(url=url, status=429)
@@ -150,7 +150,7 @@ def test_send_failures(self) -> None:
             logstash_shipper.set_replay_handler(replay_handler)
             event = deepcopy(_dummy_event)
             assert logstash_shipper.send(event) == _EVENT_SENT
-            replay_handler.assert_called_once_with("logstash", {}, event)
+            replay_handler.assert_called_once_with(url, {}, event)
 
     @responses.activate
     def test_flush(self) -> None: