Moved tests/util/get_passed_env.py into /bonobo/examples/env_vars/get_passed_env.py.
This commit is contained in:
0
bonobo/examples/env_vars/__init__.py
Normal file
0
bonobo/examples/env_vars/__init__.py
Normal file
22
bonobo/examples/env_vars/get_passed_env.py
Normal file
22
bonobo/examples/env_vars/get_passed_env.py
Normal file
@ -0,0 +1,22 @@
|
|||||||
|
import os
|
||||||
|
|
||||||
|
from bonobo import Graph
|
||||||
|
|
||||||
|
|
||||||
|
def extract():
|
||||||
|
env_test_user = os.getenv('ENV_TEST_USER')
|
||||||
|
env_test_number = os.getenv('ENV_TEST_NUMBER')
|
||||||
|
env_test_string = os.getenv('ENV_TEST_STRING')
|
||||||
|
return env_test_user, env_test_number, env_test_string
|
||||||
|
|
||||||
|
|
||||||
|
def load(s: str):
|
||||||
|
print(s)
|
||||||
|
|
||||||
|
|
||||||
|
graph = Graph(extract, load)
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
from bonobo import run
|
||||||
|
|
||||||
|
run(graph)
|
||||||
@ -103,7 +103,7 @@ def test_version(runner, capsys):
|
|||||||
def test_run_with_env(runner, capsys):
|
def test_run_with_env(runner, capsys):
|
||||||
runner(
|
runner(
|
||||||
'run', '--quiet',
|
'run', '--quiet',
|
||||||
str(pathlib.Path(os.path.dirname(__file__), 'util', 'get_passed_env.py')), '--env', 'ENV_TEST_NUMBER=123',
|
get_examples_path('env_vars/get_passed_env.py'), '--env', 'ENV_TEST_NUMBER=123',
|
||||||
'--env', 'ENV_TEST_USER=cwandrews', '--env', "ENV_TEST_STRING='my_test_string'"
|
'--env', 'ENV_TEST_USER=cwandrews', '--env', "ENV_TEST_STRING='my_test_string'"
|
||||||
)
|
)
|
||||||
out, err = capsys.readouterr()
|
out, err = capsys.readouterr()
|
||||||
@ -116,7 +116,7 @@ def test_run_with_env(runner, capsys):
|
|||||||
@all_runners
|
@all_runners
|
||||||
def test_run_module_with_env(runner, capsys):
|
def test_run_module_with_env(runner, capsys):
|
||||||
runner(
|
runner(
|
||||||
'run', '--quiet', '-m', 'tests.util.get_passed_env', '--env', 'ENV_TEST_NUMBER=123', '--env',
|
'run', '--quiet', '-m', 'bonobo.examples.env_vars.get_passed_env', '--env', 'ENV_TEST_NUMBER=123', '--env',
|
||||||
'ENV_TEST_USER=cwandrews', '--env', "ENV_TEST_STRING='my_test_string'"
|
'ENV_TEST_USER=cwandrews', '--env', "ENV_TEST_STRING='my_test_string'"
|
||||||
)
|
)
|
||||||
out, err = capsys.readouterr()
|
out, err = capsys.readouterr()
|
||||||
|
|||||||
Reference in New Issue
Block a user