swydd/tests/test_operators.py

59 lines
1.7 KiB
Python
Executable file

from swydd import SwyddPath, asset, get, geterr, pipe, proc, seq, sub
def test_sub():
assert sub("echo 'hello'")
def test_pipes():
assert sub(
pipe("cat ../src/swydd/__init__.py").pipe("grep '__version__'").pipe("wc -l")
)
assert sub(
proc("cat ../src/swydd/__init__.py").pipe("grep '__version__'").pipe("wc -l")
)
def test_seqs():
# -a is not an arg to cat so the subprocess should return false
assert not sub(seq("cat -a").then("echo hello"))
assert not sub(proc("cat -a").then("echo hello"))
def test_capture():
result = get("ls src not-src")
assert result == "src:\nswydd"
result = geterr("ls src not-src")
assert result == "ls: cannot access 'not-src': No such file or directory"
assert "hello part deux" == get(proc("echo 'hello'").then("echo hello part deux"))
commands = proc("cp").then("echo hello part deux")
assert "" == get(commands)
assert "cp: missing file operand\nTry 'cp --help' for more information." == (
geterr(commands)
)
def check_result_file(file: SwyddPath, text: str) -> bool:
if p := file._path:
return p.read_text() == text
return False
def test_write_to_path():
result_f = asset("products/result.txt")
result_txt = "text to file"
result_f.write(result_txt)
assert check_result_file(result_f, result_txt + "\n")
def test_copy_and_rename():
src_f = asset("fixtures/input.txt")
dst_f = asset("products/input.txt")
dst_f.copy(src_f)
assert check_result_file(dst_f, "data to copy to another file\n")
dst_f.rename("products/input2.txt")
assert check_result_file(
asset("products/input2.txt"), "data to copy to another file\n"
)