Commentator2.0
dbff52bfbc
with_env now overrides the environment, similar to with_stdin an additional function update_env was created to mirror the prior functionality of with_env, updating the env This was changed as previously it was impossible to delete variables from the env replaced the code of .ok() with a call to .expect, to remove the code duplication Change-Id: I83933893c7f2ccfdc7bd4933b7592b475c435e76
88 lines
2.4 KiB
Python
88 lines
2.4 KiB
Python
import logging
|
|
import os
|
|
import stat
|
|
from pathlib import Path
|
|
from subprocess import CalledProcessError
|
|
|
|
import pytest
|
|
from _pytest.logging import LogCaptureFixture
|
|
from functional2.testlib.commands import Command
|
|
from functional2.testlib.fixtures.file_helper import File
|
|
|
|
|
|
def test_command_valid_runs():
|
|
cmd = Command(["echo", "water"]).with_env(**os.environ.copy())
|
|
cmd.run().ok()
|
|
|
|
|
|
def test_command_captures_stdout():
|
|
cmd = Command(["echo", "fire"]).with_env(**os.environ.copy())
|
|
res = cmd.run().ok()
|
|
assert res.stdout_s == "fire\n"
|
|
|
|
|
|
def test_command_plain_strips():
|
|
cmd = Command(["echo", " earth "]).with_env(**os.environ.copy())
|
|
res = cmd.run().ok()
|
|
assert res.stdout_plain == "earth"
|
|
|
|
|
|
def test_command_stdin_passed_correctly():
|
|
inp = b"air"
|
|
cmd = Command(["cat", "/dev/stdin"]).with_stdin(inp).with_env(**os.environ.copy())
|
|
res = cmd.run().ok()
|
|
assert res.stdout == inp
|
|
|
|
|
|
def test_command_expect_failure():
|
|
cmd = Command(["grep", "xxx"]).with_env(**os.environ.copy()).with_stdin(b"")
|
|
cmd.run().expect(1)
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"files",
|
|
[
|
|
{
|
|
"script.sh": File(
|
|
"#!/bin/sh\necho forb\nexit 1", mode=stat.S_IRWXU | stat.S_IRWXG | stat.S_IRWXO
|
|
)
|
|
}
|
|
],
|
|
indirect=True,
|
|
)
|
|
def test_command_ok_fails_on_bad_exit_code(files: Path, caplog: LogCaptureFixture):
|
|
cmd = Command(["./script.sh"], cwd=files).with_env(**os.environ.copy())
|
|
res = cmd.run()
|
|
|
|
with pytest.raises(CalledProcessError), caplog.at_level(logging.ERROR):
|
|
res.ok()
|
|
msgs = caplog.messages
|
|
assert len(msgs) == 2
|
|
out_msg, err_msg = msgs
|
|
assert out_msg == "stdout: forb\n"
|
|
assert err_msg == "stderr: "
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"files",
|
|
[
|
|
{
|
|
"script.sh": File(
|
|
"#!/bin/sh\necho drgn fops\nexit 2", mode=stat.S_IRWXU | stat.S_IRWXG | stat.S_IRWXO
|
|
)
|
|
}
|
|
],
|
|
indirect=True,
|
|
)
|
|
def test_command_exec_fails_on_other_bad_exit_code(files: Path, caplog: LogCaptureFixture):
|
|
cmd = Command(["./script.sh"], cwd=files).with_env(**os.environ.copy())
|
|
res = cmd.run()
|
|
|
|
with pytest.raises(CalledProcessError), caplog.at_level(logging.ERROR):
|
|
res.expect(1)
|
|
msgs = caplog.messages
|
|
assert len(msgs) == 2
|
|
out_msg, err_msg = msgs
|
|
assert out_msg == "stdout: drgn fops\n"
|
|
assert err_msg == "stderr: "
|