mirror of
https://github.com/kevingruesser/bootstrap-vz.git
synced 2025-08-24 07:26:29 +00:00
40 lines
1.1 KiB
Python
40 lines
1.1 KiB
Python
|
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',
|
||
|
|
'No, three..', # The "four" is only recorded *after* the process is done (no newline)...
|
||
|
|
'four\rthree', # .. which is why "No, three" comes first
|
||
|
|
]
|
||
|
|
eq_(logged.getvalue().split("\n")[8:-1], expected_order)
|