def test_get_environment_variables():
cmd = ['echo', 'FOO\nNAME=value\n\nSOMETHING\nNAME2=value with spaces']
coroutine = get_environment_variables(cmd, shell=False)
env = run_until_complete(coroutine)
assert len(env.keys()) == 2
assert 'NAME' in env.keys()
assert env['NAME'] == 'value\nSOMETHING'
assert 'NAME2' in env.keys()
assert env['NAME2'] == 'value with spaces'
# test with environment strings which isn't decodable
async def check_output(cmd, **kwargs):
return b'DECODE_ERROR=\x81\nNAME=value'
with patch('colcon_core.shell.check_output', side_effect=check_output):
with patch('colcon_core.shell.logger.warning') as warn:
coroutine = get_environment_variables(['not-used'], shell=False)
env = run_until_complete(coroutine)
assert len(env.keys()) == 1
assert 'NAME' in env.keys()
assert env['NAME'] == 'value'
# the raised decode error is catched and results in a warning message
assert warn.call_count == 1
assert len(warn.call_args[0]) == 1
assert warn.call_args[0][0].startswith(
"Failed to decode line from the environment using the encoding '")
assert 'DECODE_ERROR=' in warn.call_args[0][0]
def test_get_environment_variables():
cmd = [
'echo', 'NAME=value\n\nSOMETHING\nNAME2=value with spaces']
coroutine = get_environment_variables(cmd, shell=False)
env = run_until_complete(coroutine)
assert len(env.keys()) == 2
assert 'NAME' in env.keys()
assert env['NAME'] == 'value'
assert 'NAME2' in env.keys()
assert env['NAME2'] == 'value with spaces'