GithubBackup/httprunner
1
0
Fork 0
mirror of https://github.com/httprunner/httprunner.git synced 2026-05-12 02:21:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

relocate functions related to .env

Browse Source
This commit is contained in:
debugtalk
2018-11-15 19:37:50 +08:00
parent 58fa2a4bf4
commit 0e54c4aec9
2 changed files with 6 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
httprunner/loader.py
View File

@@ -163,10 +163,11 @@ def load_dot_env_file(dot_env_path):
"""
if not os.path.isfile(dot_env_path):
raise exceptions.FileNotFound(".env file path is not exist.")
return {}
logger.log_info("Loading environment variables from {}".format(dot_env_path))
env_variables_mapping = {}
with io.open(dot_env_path, 'r', encoding='utf-8') as fp:
for line in fp:
# maxsplit=1
@@ -889,11 +890,9 @@ def load_project_tests(test_path, dot_env_path=None):
# load .env
dot_env_path = dot_env_path or os.path.join(project_working_directory, ".env")
if os.path.isfile(dot_env_path):
project_mapping["env"] = load_dot_env_file(dot_env_path)
else:
project_mapping["env"] = {}
project_mapping["env"] = load_dot_env_file(dot_env_path)
# load api/testcases
project_mapping["def-api"] = load_api_folder(os.path.join(project_working_directory, "api"))
# TODO: replace suite with testcases
project_mapping["def-testcase"] = load_test_folder(os.path.join(project_working_directory, "suite"))

4
tests/test_loader.py
View File

@@ -153,8 +153,8 @@ class TestFileLoader(unittest.TestCase):
dot_env_path = os.path.join(
os.getcwd(), "tests", "data",
)
with self.assertRaises(exceptions.FileNotFound):
loader.load_dot_env_file(dot_env_path)
env_variables_mapping = loader.load_dot_env_file(dot_env_path)
self.assertEqual(env_variables_mapping, {})
def test_locate_file(self):
with self.assertRaises(exceptions.FileNotFound):