From 6abdf09ef7615ad9de9e14e68afc77998fb8ea3b Mon Sep 17 00:00:00 2001
From: cwandrews <cwandrews.oh@gmail.com>
Date: Sun, 1 Oct 2017 19:56:49 -0400
Subject: [PATCH] Moved tests/util/get_passed_env.py into
 /bonobo/examples/env_vars/get_passed_env.py.

---
 bonobo/examples/env_vars/__init__.py       |  0
 bonobo/examples/env_vars/get_passed_env.py | 22 ++++++++++++++++++++++
 tests/test_commands.py                     |  4 ++--
 3 files changed, 24 insertions(+), 2 deletions(-)
 create mode 100644 bonobo/examples/env_vars/__init__.py
 create mode 100644 bonobo/examples/env_vars/get_passed_env.py

diff --git a/bonobo/examples/env_vars/__init__.py b/bonobo/examples/env_vars/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/bonobo/examples/env_vars/get_passed_env.py b/bonobo/examples/env_vars/get_passed_env.py
new file mode 100644
index 0000000..d9c4ba6
--- /dev/null
+++ b/bonobo/examples/env_vars/get_passed_env.py
@@ -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)
diff --git a/tests/test_commands.py b/tests/test_commands.py
index 730bc0b..3f26e44 100644
--- a/tests/test_commands.py
+++ b/tests/test_commands.py
@@ -103,7 +103,7 @@ def test_version(runner, capsys):
 def test_run_with_env(runner, capsys):
     runner(
         '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'"
     )
     out, err = capsys.readouterr()
@@ -116,7 +116,7 @@ def test_run_with_env(runner, capsys):
 @all_runners
 def test_run_module_with_env(runner, capsys):
     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'"
     )
     out, err = capsys.readouterr()