Update tests/test_utils_logcfg.py

tests/test_utils_logcfg.py

@@ -1,45 +1,87 @@
 import logging
-import sys
-import pytest
 import logzero
+import pytest
+import tempfile
+import subprocess
+import os
+import sys
+import textwrap
 from logzero import logger as log
-from slic.utils.logcfg import add_log_Level, logcfg
-from slic.utils.logign import ignore_log_msg
+from slic.utils.logcfg import add_log_Level, logcfg, setup_import_logging
 
-@pytest.mark.parametrize("levelname,msg_to_ignore,msg_to_keep", [
-    ("WARNING", "This should be ignored", "This should appear"),
-    ("ENLARGE", "ENLARGE this", "Keep this ENLARGE"),
+# Set for all the tests
+@pytest.fixture(scope="module", autouse=True)
+def setup_custom_levels():
+    add_log_Level(log, "LONG", logging.INFO + 5, func_name="long")
+    add_log_Level(log, "ENLARGE", logging.INFO + 6, func_name="enlarge")
+    logcfg("LONG")
+    setup_import_logging()
+
+# Test simple custom log
+
+@pytest.mark.parametrize("levelname, logfunc, message", [
+    ("LONG", lambda msg: log.long(msg), "This is a LONG message"),
+    ("ENLARGE", lambda msg: log.enlarge(msg), "Please ENLARGE this!"),
 ])
-def test_ignore_log_msg_behavior(levelname, msg_to_ignore, msg_to_keep, capsys):
-    # Remove all handlers
+
+def test_custom_log_outputs(levelname, logfunc, message, capsys):
+    # Remove all handlers to avoid duplicate logs
     for h in log.handlers[:]:
         log.removeHandler(h)
 
-    # Set handler manually like your working test
+    # Create a new handler that logs to stderr
     handler = logging.StreamHandler(sys.stderr)
     formatter = logzero.LogFormatter()
     handler.setFormatter(formatter)
     log.addHandler(handler)
 
-    # Add custom level and set config
-    add_log_Level(log, "ENLARGE", logging.INFO + 6, func_name="enlarge")
-    logcfg("DEBUG")
+    logcfg(levelname)
 
-    # Clear any prior capsys capture
-    capsys.readouterr()
-
-    with ignore_log_msg(log, lvl=levelname, msg=msg_to_ignore):
-        if levelname == "WARNING":
-            log.warning(msg_to_ignore)
-            log.warning(msg_to_keep)
-        elif levelname == "ENLARGE":
-            log.enlarge(msg_to_ignore)
-            log.enlarge(msg_to_keep)
-
-    # Read captured stderr
+    # Call and capture
+    logfunc(message)
     captured = capsys.readouterr().err
+
     print("Captured stderr:", repr(captured))
 
-    # Assert the message to ignore is gone, the other one is present
-    assert msg_to_ignore not in captured, f"'{msg_to_ignore}' should have been filtered"
-    assert msg_to_keep in captured, f"'{msg_to_keep}' should have appeared"
+    assert message in captured, f"Expected message '{message}' not found in log"
+    assert levelname in captured, f"Expected level '{levelname}' not present in log"
+
+
+# Test subprocess: top-level import logs only once
+
+def test_import_logging_once_per_module():
+    code = textwrap.dedent("""
+        import sys
+        from slic.utils.logcfg import setup_import_logging, add_log_Level, logcfg
+        from logzero import logger as log
+        import logging
+
+        add_log_Level(log, "TRACE", logging.DEBUG - 1, func_name="trace")
+        logcfg("TRACE")
+        setup_import_logging()
+
+        import json
+        import math
+        import io
+        import random
+    """)
+
+    with tempfile.NamedTemporaryFile("w", suffix=".py", delete=False) as tmp:
+        tmp.write(code)
+        tmp_path = tmp.name
+
+    env = os.environ.copy()
+    root_path = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
+    env["PYTHONPATH"] = root_path + os.pathsep + env.get("PYTHONPATH", "")
+
+    result = subprocess.run([sys.executable, tmp_path], capture_output=True, text=True, env=env)
+    os.remove(tmp_path)
+
+    assert result.returncode == 0, f"Script failed:\n{result.stderr}"
+
+    stderr = result.stderr
+    lines = stderr.splitlines()
+
+    for mod in ["json", "math", "io", "random"]:
+        count = sum(1 for line in lines if f"importing: {mod}" in line)
+        assert count == 1, f"Expected 1 import log for '{mod}', found {count}"