Better methods for running commands

tests/test_decman_core_command.py

@@ -0,0 +1,70 @@
+import json
+import sys
+
+import pytest
+
+import decman.core.command as command
+
+
+def test_run_simple():
+    code, out = command.run([sys.executable, "-c", "print('ok')"])
+    assert code == 0
+    assert out.strip() == "ok"
+
+
+def test_run_exec_failure():
+    code, out = command.run(["/does/not/exist"])
+    assert code != 0
+    assert "not" in out.lower()
+
+
+def test_run_env_overrides_and_mimic_login_visible_in_child(monkeypatch):
+    class FakePw:
+        pw_dir = "/fake/home"
+        pw_name = "fakeuser"
+        pw_uid = 1000
+        pw_gid = 1000
+        pw_shell = "/bin/fakesh"
+
+    # Mock passwd lookup
+    monkeypatch.setattr(
+        "decman.core.command.pwd.getpwnam",
+        lambda user: FakePw(),
+    )
+
+    code, out = command.run(
+        [
+            sys.executable,
+            "-c",
+            (
+                "import os, json; "
+                "print(json.dumps({"
+                "'FOO': os.environ['FOO'], "
+                "'HOME': os.environ['HOME'], "
+                "'USER': os.environ['USER'], "
+                "'LOGNAME': os.environ['LOGNAME'], "
+                "'SHELL': os.environ['SHELL']"
+                "}))"
+            ),
+        ],
+        user="fakeuser",
+        mimic_login=True,
+        env_overrides={"FOO": "BAR"},
+    )
+
+    assert code == 0
+
+    data = json.loads(out.strip())
+    assert data["FOO"] == "BAR"
+    assert data["HOME"] == "/fake/home"
+    assert data["USER"] == "fakeuser"
+    assert data["LOGNAME"] == "fakeuser"
+    assert data["SHELL"] == "/bin/fakesh"
+
+
+@pytest.mark.skipif(not sys.stdin.isatty(), reason="requires TTY")
+def test_pty_run_simple():
+    code, out = command.pty_run([sys.executable, "-c", "print('ok')"])
+    assert code == 0
+    assert "ok" in out
+    assert "\r\n" not in out