fix: add setup_module to prevent test ordering issues with Message patch

The teardown_module from one test file was restoring the real Message
before another test file's tests ran. Adding setup_module ensures the
stub is re-applied before each module's tests execute.

Also fix test assertion to check contents instead of text since the
middleware now uses Message(contents=).

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>

Author: 2 people

src/processor/src/tests/unit/libs/agent_framework/test_groupchat_orchestrator_internals.py

@@ -40,6 +40,11 @@ def __init__(self, *, role, text=None, contents=None, author_name=None):
 )
 
 
+def setup_module(module=None):
+    """Re-apply the Message patch in case another module's teardown restored it."""
+    groupchat_module.Message = Message
+
+
 def teardown_module(module=None):
     """Restore the original Message class to avoid leaking into other tests."""
     if _original_message is not None:

src/processor/src/tests/unit/libs/agent_framework/test_input_observer_middleware.py

@@ -26,6 +26,11 @@ def __init__(self, *, role, text=None, contents=None, author_name=None):
 from libs.agent_framework.middlewares import InputObserverMiddleware  # noqa: E402
 
 
+def setup_module(module=None):
+    """Re-apply the Message patch in case another module's teardown restored it."""
+    middlewares_module.Message = Message
+
+
 def teardown_module(module=None):
     """Restore the original Message class to avoid leaking into other tests."""
     if _original_message is not None:

src/processor/src/tests/unit/libs/agent_framework/test_middlewares_extras.py

@@ -33,6 +33,11 @@ def __init__(self, *, role, text=None, contents=None, author_name=None):
 )
 
 
+def setup_module(module=None):
+    """Re-apply the Message patch in case another module's teardown restored it."""
+    middlewares_module.Message = Message
+
+
 def teardown_module(module=None):
     """Restore the original Message class to avoid leaking into other tests."""
     if _original_message is not None:
@@ -130,4 +135,4 @@ def test_no_replacement_keeps_text(self):
         ctx.messages = [msg]
         mw = InputObserverMiddleware(replacement=None)
         _run(mw.process(ctx, AsyncMock()))
-        assert ctx.messages[0].text == "keep me"
+        assert ctx.messages[0].contents == "keep me"