def setUp(self):
self.out = StoreOutput()
self.err = StoreOutput()
self.finished = False
def on_finished():
self.finished = True
self.timeout = False
def on_timeout():
self.timeout = True
self.reader = ProcessReader(stdout_callback=self.out,
stderr_callback=self.err,
finished_callback=on_finished,
timeout_callback=on_timeout)
def setUp(self):
self.out = StoreOutput()
self.err = StoreOutput()
self.finished = False
def on_finished():
self.finished = True
self.timeout = False
def on_timeout():
self.timeout = True
self.reader = ProcessReader(stdout_callback=self.out,
stderr_callback=self.err,
finished_callback=on_finished,
timeout_callback=on_timeout)
class TestProcessReader(unittest.TestCase):
def setUp(self):
self.out = StoreOutput()
self.err = StoreOutput()
self.finished = False
def on_finished():
self.finished = True
self.timeout = False
def on_timeout():
self.timeout = True
self.reader = ProcessReader(stdout_callback=self.out,
stderr_callback=self.err,
finished_callback=on_finished,
timeout_callback=on_timeout)
def test_stdout_callback(self):
proc = run_python('print 1; print 2')
self.reader.start(proc)
self.reader.thread.join()
self.assertEqual(self.out.output, ['1', '2'])
self.assertEqual(self.err.output, [])
def test_stderr_callback(self):
proc = run_python('import sys; sys.stderr.write("hello world\\n")')
self.reader.start(proc)
self.reader.thread.join()
self.assertEqual(self.out.output, [])
self.assertEqual(self.err.output, ['hello world'])
def test_stdout_and_stderr_callbacks(self):
proc = run_python('import sys; sys.stderr.write("hello world\\n"); print 1; print 2')
self.reader.start(proc)
self.reader.thread.join()
self.assertEqual(self.out.output, ['1', '2'])
self.assertEqual(self.err.output, ['hello world'])
def test_finished_callback(self):
self.assertFalse(self.finished)
proc = run_python('')
self.reader.start(proc)
self.reader.thread.join()
self.assertTrue(self.finished)
def test_timeout(self):
self.reader.timeout = 0.05
self.assertFalse(self.timeout)
proc = run_python('import time; time.sleep(0.1)')
self.reader.start(proc)
self.reader.thread.join()
self.assertTrue(self.timeout)
self.assertFalse(self.finished)
def test_output_timeout(self):
self.reader.output_timeout = 0.05
self.assertFalse(self.timeout)
proc = run_python('import time; time.sleep(0.1)')
self.reader.start(proc)
self.reader.thread.join()
self.assertTrue(self.timeout)
self.assertFalse(self.finished)
def test_read_without_eol(self):
proc = run_python('import sys; sys.stdout.write("1")')
self.reader.start(proc)
self.reader.thread.join()
self.assertEqual(self.out.output, ['1'])
def test_read_with_strange_eol(self):
proc = run_python('import sys; sys.stdout.write("1\\r\\r\\r\\n")')
self.reader.start(proc)
self.reader.thread.join()
self.assertEqual(self.out.output, ['1'])
def test_mixed_stdout_stderr(self):
proc = run_python('import sys; sys.stderr.write("hello world\\n"); print 1; print 2',
stderr=subprocess.STDOUT)
self.reader.start(proc)
self.reader.thread.join()
self.assertEqual(sorted(self.out.output), sorted(['1', '2', 'hello world']))
self.assertEqual(self.err.output, [])
class TestProcessReader(unittest.TestCase):
def setUp(self):
self.out = StoreOutput()
self.err = StoreOutput()
self.finished = False
def on_finished():
self.finished = True
self.timeout = False
def on_timeout():
self.timeout = True
self.reader = ProcessReader(stdout_callback=self.out,
stderr_callback=self.err,
finished_callback=on_finished,
timeout_callback=on_timeout)
def test_stdout_callback(self):
proc = run_python('print 1; print 2')
self.reader.start(proc)
self.reader.thread.join()
self.assertEqual(self.out.output, ['1', '2'])
self.assertEqual(self.err.output, [])
def test_stderr_callback(self):
proc = run_python('import sys; sys.stderr.write("hello world\\n")')
self.reader.start(proc)
self.reader.thread.join()
self.assertEqual(self.out.output, [])
self.assertEqual(self.err.output, ['hello world'])
def test_stdout_and_stderr_callbacks(self):
proc = run_python(
'import sys; sys.stderr.write("hello world\\n"); print 1; print 2')
self.reader.start(proc)
self.reader.thread.join()
self.assertEqual(self.out.output, ['1', '2'])
self.assertEqual(self.err.output, ['hello world'])
def test_finished_callback(self):
self.assertFalse(self.finished)
proc = run_python('')
self.reader.start(proc)
self.reader.thread.join()
self.assertTrue(self.finished)
def test_timeout(self):
self.reader.timeout = 0.05
self.assertFalse(self.timeout)
proc = run_python('import time; time.sleep(0.1)')
self.reader.start(proc)
self.reader.thread.join()
self.assertTrue(self.timeout)
self.assertFalse(self.finished)
def test_output_timeout(self):
self.reader.output_timeout = 0.05
self.assertFalse(self.timeout)
proc = run_python('import time; time.sleep(0.1)')
self.reader.start(proc)
self.reader.thread.join()
self.assertTrue(self.timeout)
self.assertFalse(self.finished)
def test_read_without_eol(self):
proc = run_python('import sys; sys.stdout.write("1")')
self.reader.start(proc)
self.reader.thread.join()
self.assertEqual(self.out.output, ['1'])
def test_read_with_strange_eol(self):
proc = run_python('import sys; sys.stdout.write("1\\r\\r\\r\\n")')
self.reader.start(proc)
self.reader.thread.join()
self.assertEqual(self.out.output, ['1'])
def test_mixed_stdout_stderr(self):
proc = run_python(
'import sys; sys.stderr.write("hello world\\n"); print 1; print 2',
stderr=subprocess.STDOUT)
self.reader.start(proc)
self.reader.thread.join()
self.assertEqual(sorted(self.out.output),
sorted(['1', '2', 'hello world']))
self.assertEqual(self.err.output, [])
class TestProcessReader(unittest.TestCase):
def setUp(self):
self.out = StoreOutput()
self.err = StoreOutput()
self.finished = False
def on_finished():
self.finished = True
self.timeout = False
def on_timeout():
self.timeout = True
self.reader = ProcessReader(
stdout_callback=self.out,
stderr_callback=self.err,
finished_callback=on_finished,
timeout_callback=on_timeout,
)
def test_stdout_callback(self):
proc = run_python("print(1); print(2)")
self.reader.start(proc)
self.reader.thread.join()
self.assertEqual([x.decode("utf8") for x in self.out.output],
["1", "2"])
self.assertEqual(self.err.output, [])
def test_stderr_callback(self):
proc = run_python('import sys; sys.stderr.write("hello world\\n")')
self.reader.start(proc)
self.reader.thread.join()
self.assertEqual(self.out.output, [])
self.assertEqual([x.decode("utf8") for x in self.err.output],
["hello world"])
def test_stdout_and_stderr_callbacks(self):
proc = run_python(
'import sys; sys.stderr.write("hello world\\n"); print(1); print(2)'
)
self.reader.start(proc)
self.reader.thread.join()
self.assertEqual([x.decode("utf8") for x in self.out.output],
["1", "2"])
self.assertEqual([x.decode("utf8") for x in self.err.output],
["hello world"])
def test_finished_callback(self):
self.assertFalse(self.finished)
proc = run_python("")
self.reader.start(proc)
self.reader.thread.join()
self.assertTrue(self.finished)
def test_timeout(self):
self.reader.timeout = 0.05
self.assertFalse(self.timeout)
proc = run_python("import time; time.sleep(0.1)")
self.reader.start(proc)
self.reader.thread.join()
self.assertTrue(self.timeout)
self.assertFalse(self.finished)
def test_output_timeout(self):
self.reader.output_timeout = 0.05
self.assertFalse(self.timeout)
proc = run_python("import time; time.sleep(0.1)")
self.reader.start(proc)
self.reader.thread.join()
self.assertTrue(self.timeout)
self.assertFalse(self.finished)
def test_read_without_eol(self):
proc = run_python('import sys; sys.stdout.write("1")')
self.reader.start(proc)
self.reader.thread.join()
self.assertEqual([x.decode("utf8") for x in self.out.output], ["1"])
def test_read_with_strange_eol(self):
proc = run_python('import sys; sys.stdout.write("1\\r\\r\\r\\n")')
self.reader.start(proc)
self.reader.thread.join()
self.assertEqual([x.decode("utf8") for x in self.out.output], ["1"])
def test_mixed_stdout_stderr(self):
proc = run_python(
'import sys; sys.stderr.write("hello world\\n"); print(1); print(2)',
stderr=subprocess.STDOUT,
)
self.reader.start(proc)
self.reader.thread.join()
self.assertEqual(
sorted([x.decode("utf8") for x in self.out.output]),
sorted(["1", "2", "hello world"]),
)
self.assertEqual(self.err.output, [])