Test for caplog utils

Signed-off-by: Luka Govedič <lgovedic@redhat.com>

Author: ProExpertProg

tests/conftest.py

@@ -1070,7 +1070,7 @@ def caplog_vllm(temporary_enable_log_propagate, caplog):
     yield caplog
 
 
-@pytest.fixture(scope="session")
+@pytest.fixture()
 def caplog_mp_fork():
     """
     This fixture enables capturing logs from a forked MP subprocess.
@@ -1115,7 +1115,7 @@ def __init__(self):
         self.text = None
 
 
-@pytest.fixture(scope="session")
+@pytest.fixture()
 def caplog_mp_spawn(tmp_path, monkeypatch):
     """
     This fixture enables capturing logs from a forked MP subprocess.

tests/test_logger.py

@@ -515,5 +515,34 @@ def mp_function(**kwargs):
 
 
 def test_caplog_mp_fork(caplog_vllm, caplog_mp_fork):
-    pass
-    # TODO
+    with caplog_vllm.at_level(logging.DEBUG), caplog_mp_fork():
+        import multiprocessing
+
+        ctx = multiprocessing.get_context("fork")
+        p = ctx.Process(
+            target=mp_function,
+            name=f"SubProcess{1}",
+            kwargs={"a": "AAAA", "b": "BBBBB"},
+        )
+        p.start()
+        p.join()
+
+    assert "AAAA" in caplog_vllm.text
+    assert "BBBBB" in caplog_vllm.text
+
+
+def test_caplog_mp_spawn(caplog_mp_spawn):
+    with caplog_mp_spawn(logging.DEBUG) as log_holder:
+        import multiprocessing
+
+        ctx = multiprocessing.get_context("spawn")
+        p = ctx.Process(
+            target=mp_function,
+            name=f"SubProcess{1}",
+            kwargs={"a": "AAAA", "b": "BBBBB"},
+        )
+        p.start()
+        p.join()
+
+    assert "AAAA" in log_holder.text
+    assert "BBBBB" in log_holder.text