paulin/bonobo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Moved env vars tests to class.

Browse Source
This commit is contained in:
cwandrews
2017-10-15 15:49:14 -04:00
parent 84e197b209
commit dc34ab4a8b
2 changed files with 51 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
bonobo/examples/environment/env_vars/get_passed_env.py
View File

@ -4,9 +4,9 @@ import bonobo
def extract(): def extract():
env_test_user = os.getenv('ENV_TEST_USER') env_test_user = os.getenv('ENV_TEST_USER', 'user')
env_test_number = os.getenv('ENV_TEST_NUMBER') env_test_number = os.getenv('ENV_TEST_NUMBER', 'number')
env_test_string = os.getenv('ENV_TEST_STRING') env_test_string = os.getenv('ENV_TEST_STRING', 'string')
return env_test_user, env_test_number, env_test_string return env_test_user, env_test_number, env_test_string

73
tests/test_commands.py
View File

@ -229,30 +229,53 @@ def test_run_file_with_default_env_file_and_env_file_and_env_vars(runner, capsys
@all_runners @all_runners
def test_run_file_with_env_vars(runner, capsys): class TestEnvVars(object):
runner( def test_run_file_with_env_var(self, runner, capsys):
'run', '--quiet', runner(
get_examples_path('environment/env_vars/get_passed_env.py'), 'run', '--quiet',
'--env', 'ENV_TEST_NUMBER=123', '--env', 'ENV_TEST_USER=cwandrews', get_examples_path('environment/env_vars/get_passed_env.py'),
'--env', "ENV_TEST_STRING='my_test_string'" '--env', 'ENV_TEST_NUMBER=123'
) )
out, err = capsys.readouterr() out, err = capsys.readouterr()
out = out.split('\n') out = out.split('\n')
assert out[0] == 'cwandrews' assert out[0] != 'test_user'
assert out[1] == '123' assert out[1] == '123'
assert out[2] == 'my_test_string' assert out[2] == 'string'
def test_run_file_with_env_vars(self, runner, capsys):
runner(
'run', '--quiet',
get_examples_path('environment/env_vars/get_passed_env.py'),
'--env', 'ENV_TEST_NUMBER=123', '--env', 'ENV_TEST_USER=cwandrews',
'--env', "ENV_TEST_STRING='my_test_string'"
)
out, err = capsys.readouterr()
out = out.split('\n')
assert out[0] == 'cwandrews'
assert out[1] == '123'
assert out[2] == 'my_test_string'
@all_runners def test_run_module_with_env_var(self, runner, capsys):
def test_run_module_with_env_vars(runner, capsys): runner(
runner( 'run', '--quiet', '-m',
'run', '--quiet', '-m', 'bonobo.examples.environment.env_vars.get_passed_env',
'bonobo.examples.environment.env_vars.get_passed_env', '--env', 'ENV_TEST_NUMBER=123'
'--env', 'ENV_TEST_NUMBER=123', '--env', 'ENV_TEST_USER=cwandrews', )
'--env', "ENV_TEST_STRING='my_test_string'" out, err = capsys.readouterr()
) out = out.split('\n')
out, err = capsys.readouterr() assert out[0] == 'cwandrews'
out = out.split('\n') assert out[1] == '123'
assert out[0] == 'cwandrews' assert out[2] == 'my_test_string'
assert out[1] == '123'
assert out[2] == 'my_test_string' def test_run_module_with_env_vars(self, runner, capsys):
runner(
'run', '--quiet', '-m',
'bonobo.examples.environment.env_vars.get_passed_env',
'--env', 'ENV_TEST_NUMBER=123', '--env', 'ENV_TEST_USER=cwandrews',
'--env', "ENV_TEST_STRING='my_test_string'"
)
out, err = capsys.readouterr()
out = out.split('\n')
assert out[0] == 'cwandrews'
assert out[1] == '123'
assert out[2] == 'my_test_string'