def test_write(self):
p = CPopen(["dd", "of=/dev/null",
"bs=%d" % self.BUFSIZE],
stdin=subprocess.PIPE,
stdout=None,
stderr=subprocess.PIPE)
start = monotonic_time()
total = self.COUNT * self.BUFSIZE
sent = 0
with io.open("/dev/zero", "rb") as f:
while sent < total:
n = min(total - sent, self.BUFSIZE)
data = f.read(n)
if not data:
raise RuntimeError("/dev/zero closed?!")
p.stdin.write(data)
sent += len(data)
p.stdin.flush()
p.stdin.close()
for _, data in cmdutils.receive(p, 10):
pass
elapsed = monotonic_time() - start
sent_gb = sent / float(1024**3)
print("%.2fg in %.2f seconds (%.2fg/s)" %
(sent_gb, elapsed, sent_gb / elapsed),
end=" ")
self.assertEqual(p.returncode, 0)
def test_stderr(self):
p = CPopen(["sh", "-c", "echo error >/dev/stderr"],
stdin=None,
stdout=subprocess.PIPE,
stderr=subprocess.PIPE)
received = list(cmdutils.receive(p))
self.assertEqual(received, [(cmdutils.ERR, b"error\n")])
self.assertEqual(p.returncode, 0)
def test_stdout(self):
p = CPopen(["echo", "output"],
stdin=None,
stdout=subprocess.PIPE,
stderr=subprocess.PIPE)
received = list(cmdutils.receive(p))
self.assertEqual(received, [(cmdutils.OUT, b"output\n")])
self.assertEqual(p.returncode, 0)
def test_no_output_error(self):
p = CPopen(["false"],
stdin=None,
stdout=subprocess.PIPE,
stderr=subprocess.PIPE)
received = list(cmdutils.receive(p))
self.assertEqual(received, [])
self.assertEqual(p.returncode, 1)
def test_kill(self):
p = CPopen(["sleep", "1"],
stdin=None,
stdout=subprocess.PIPE,
stderr=subprocess.PIPE)
p.kill()
list(cmdutils.receive(p))
self.assertEqual(p.returncode, -signal.SIGKILL)
def test_terminate(self):
p = Popen(["sleep", "1"],
stdin=None,
stdout=subprocess.PIPE,
stderr=subprocess.PIPE)
p.terminate()
list(cmdutils.receive(p))
self.assertEqual(p.returncode, -signal.SIGTERM)
def test_no_fds(self):
p = CPopen(["sleep", "1"], stdin=None, stdout=None, stderr=None)
try:
with self.assertRaises(cmdutils.TimeoutExpired):
for _ in cmdutils.receive(p, 0.5):
pass
finally:
p.kill()
p.wait()
def test_both_stdout_stderr(self):
p = CPopen(["sh", "-c", "echo output; echo error >/dev/stderr;"],
stdin=None,
stdout=subprocess.PIPE,
stderr=subprocess.PIPE)
received = list(cmdutils.receive(p))
self.assertEqual(
sorted(received),
sorted([(cmdutils.OUT, b"output\n"), (cmdutils.ERR, b"error\n")]))
self.assertEqual(p.returncode, 0)
def test_timeout_with_data(self):
p = CPopen(["yes"],
stdin=None,
stdout=subprocess.PIPE,
stderr=subprocess.PIPE)
try:
with self.assertRaises(cmdutils.TimeoutExpired):
for _ in cmdutils.receive(p, 0.5):
pass
finally:
p.kill()
p.wait()
def test_fds_closed(self):
cmd = [
"python", "-c",
"import os, time; os.close(1); os.close(2); time.sleep(1)"
]
p = CPopen(cmd,
stdin=None,
stdout=subprocess.PIPE,
stderr=subprocess.PIPE)
try:
with self.assertRaises(cmdutils.TimeoutExpired):
for _ in cmdutils.receive(p, 0.5):
pass
finally:
p.kill()
p.wait()
def watch(self):
"""
Run a command, iterating on data received from underlying command
stdout.
Data read from stderr is collected and will be included in the
cmdutils.Error raised if the underlying command failed.
Raises:
`RuntimeError` if invoked more then once
`exception.ActionStopped` if the command was aborted
`cmdutils.Error` if the command failed
"""
self._start_process()
err = bytearray()
for src, data in cmdutils.receive(self._proc):
if src == cmdutils.OUT:
yield data
else:
err += data
self._finalize(b"", err)
def test_read(self):
p = CPopen([
"dd", "if=/dev/zero",
"bs=%d" % self.BUFSIZE,
"count=%d" % self.COUNT
],
stdin=None,
stdout=subprocess.PIPE,
stderr=subprocess.PIPE)
start = monotonic_time()
received = 0
for src, data in cmdutils.receive(p, bufsize=self.BUFSIZE):
if src == cmdutils.OUT:
received += len(data)
elapsed = monotonic_time() - start
received_gb = received / float(1024**3)
print("%.2fg in %.2f seconds (%.2fg/s)" %
(received_gb, elapsed, received_gb / elapsed),
end=" ")
self.assertEqual(received, self.COUNT * self.BUFSIZE)
self.assertEqual(p.returncode, 0)
