import os from nose.tools import eq_ from bootstrapvz.common.tools import log_call subprocess_path = os.path.join(os.path.dirname(os.path.realpath(__file__)), 'subprocess.sh') def setup_logger(): import logging root = logging.getLogger() root.setLevel(logging.NOTSET) import StringIO output = StringIO.StringIO() string_handler = logging.StreamHandler(output) string_handler.setLevel(logging.DEBUG) root.addHandler(string_handler) return output def test_log_call(): logged = setup_logger() fixture = """ 2 0.1 one\\\\n 1 0.2 two\\\\n 1 0.5 four 2 0.6 \\\\rNo, three.. 1 0.8 \\\\rthree """ status, stdout, stderr = log_call([subprocess_path], stdin=fixture) eq_(status, 0) eq_(stderr, ['one', 'No, three..']) eq_(stdout, ['two', 'four\rthree']) expected_order = ['one', 'two', 'four\rthree', 'No, three..', ] eq_(logged.getvalue().split("\n")[8:-1], expected_order)