test: update tests

gianlucapagliara · Feb 12, 2025 · 4acf4f7 · 4acf4f7
1 parent 7e766db
commit 4acf4f7
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tests/test_agency.py b/tests/test_agency.py
@@ -55,7 +55,7 @@ def __init__(self, agent: Agent) -> None:
         super().__init__(agent)
         self.published_events: list[tuple[Any, Any]] = []
 
-    def trigger_event(self, event_publication: Any, event_data: Any) -> None:
+    def publish(self, event_publication: Any, event_data: Any) -> None:
         self.published_events.append((event_publication, event_data))
 
     sm_transition_event = StateAgentPublishing.sm_transition_event

diff --git a/tests/test_agent.py b/tests/test_agent.py
@@ -86,7 +86,7 @@ def __init__(self, agent: Agent) -> None:
         super().__init__(agent)
         self.published_events: list[tuple[Any, Any]] = []
 
-    def trigger_event(self, event_publication: Any, event_data: Any) -> None:
+    def publish(self, event_publication: Any, event_data: Any) -> None:
         self.published_events.append((event_publication, event_data))
 
     # Override the event publication to use our mock event class
@@ -242,7 +242,7 @@ def test_agent_logger() -> None:
     """Test agent logger initialization and access."""
     logger = MockAgent.logger()
     assert logger is not None
-    assert logger.name == "agent"
+    assert logger.name == "MockAgent"
     # Test that subsequent calls return the same logger
     assert MockAgent.logger() is logger