diff options
author | Lars Wirzenius <liw@liw.fi> | 2021-02-18 10:21:31 +0200 |
---|---|---|
committer | Lars Wirzenius <liw@liw.fi> | 2021-02-18 10:21:31 +0200 |
commit | 24f26b26d0ee56f0d61ce083ad88c02bfa5f72d2 (patch) | |
tree | 07748d63e17f45261c30bb7176c07e5ab60c462f /subplot/summain.py | |
parent | 2de0f1bd7d2d91f70b670d23a42049e2bf2f7e03 (diff) | |
download | summain-rs-24f26b26d0ee56f0d61ce083ad88c02bfa5f72d2.tar.gz |
feat: handle Unix domain sockets
Diffstat (limited to 'subplot/summain.py')
-rw-r--r-- | subplot/summain.py | 21 |
1 files changed, 21 insertions, 0 deletions
diff --git a/subplot/summain.py b/subplot/summain.py index 4efc8c4..e137ade 100644 --- a/subplot/summain.py +++ b/subplot/summain.py @@ -1,5 +1,7 @@ +import difflib import logging import os +import socket def install_summain(ctx): @@ -21,6 +23,15 @@ def create_symlink(ctx, linkname=None, target=None): os.symlink(target, linkname) +def create_socket(ctx, filename=None): + fd = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM) + fd.bind(filename) + + +def set_mode(ctx, filename=None, mode=None): + os.chmod(filename, int(mode, 8)) + + def set_atime(ctx, filename=None, timestamp=None): st = os.lstat(filename) os.utime(filename, (int(timestamp), int(st.st_mtime)), follow_symlinks=False) @@ -38,7 +49,17 @@ def output_matches_file(ctx, filename=None): actual = runcmd_get_stdout(ctx) expected = get_file(filename).decode("UTF-8") + diff = "".join( + line.rstrip("\n") + "\n" + for line in difflib.unified_diff( + expected.splitlines(), + actual.splitlines(), + fromfile="expected", + tofile="actual", + ) + ) logging.debug("output_matches:") logging.debug(f" actual:\n{actual}") logging.debug(f" expect:\n{expected}") + logging.debug(f" diff:\n{diff}") assert_eq(actual, expected) |