working on better piping system
This commit is contained in:
47
tests/test_cmdpipe.py
Normal file
47
tests/test_cmdpipe.py
Normal file
@ -0,0 +1,47 @@
|
||||
from basetest import *
|
||||
from zfs_autobackup.CmdPipe import CmdPipe
|
||||
|
||||
|
||||
class TestCmdPipe(unittest2.TestCase):
|
||||
|
||||
def test_single(self):
|
||||
"""single process stdout and stderr"""
|
||||
p=CmdPipe(readonly=False, inp=None)
|
||||
err=[]
|
||||
out=[]
|
||||
p.add(["ls", "-d", "/", "/", "/nonexistent"], stderr_handler=lambda line: err.append(line))
|
||||
exits=p.execute(stdout_handler=lambda line: out.append(line))
|
||||
|
||||
self.assertEqual(err, ["ls: cannot access '/nonexistent': No such file or directory"])
|
||||
self.assertEqual(out, ["/","/"])
|
||||
self.assertEqual(exits, [2])
|
||||
|
||||
def test_input(self):
|
||||
"""test stdinput"""
|
||||
p=CmdPipe(readonly=False, inp="test")
|
||||
err=[]
|
||||
out=[]
|
||||
p.add(["echo", "test"], stderr_handler=lambda line: err.append(line))
|
||||
exits=p.execute(stdout_handler=lambda line: out.append(line))
|
||||
|
||||
self.assertEqual(err, [])
|
||||
self.assertEqual(out, ["test"])
|
||||
self.assertEqual(exits, [0])
|
||||
|
||||
def test_pipe(self):
|
||||
"""test piped"""
|
||||
p=CmdPipe(readonly=False)
|
||||
err1=[]
|
||||
err2=[]
|
||||
err3=[]
|
||||
out=[]
|
||||
p.add(["echo", "test"], stderr_handler=lambda line: err1.append(line))
|
||||
p.add(["tr", "e", "E"], stderr_handler=lambda line: err2.append(line))
|
||||
p.add(["tr", "t", "T"], stderr_handler=lambda line: err3.append(line))
|
||||
exits=p.execute(stdout_handler=lambda line: out.append(line))
|
||||
|
||||
self.assertEqual(err1, [])
|
||||
self.assertEqual(err2, [])
|
||||
self.assertEqual(err3, [])
|
||||
self.assertEqual(out, ["TEsT"])
|
||||
self.assertEqual(exits, [0,0,0])
|
||||
Reference in New Issue
Block a user