Skip to content

Commit 66eb89e

Browse files
committed
Updated e2e test for file_name
1 parent c237ba2 commit 66eb89e

File tree

1 file changed

+10
-2
lines changed

1 file changed

+10
-2
lines changed

tests/endtoend/test_file_name_functions.py

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,9 +20,17 @@ class TestHttpFunctionsFileName(testutils.WebHostTestCase):
2020
Compared to the unittests/test_http_functions.py, this file is more focus
2121
on testing the E2E flow scenarios.
2222
"""
23+
24+
@classmethod
25+
def setUpClass(cls):
26+
os.environ["PYTHON_SCRIPT_FILE_NAME"] = "main.py"
27+
super().setUpClass()
28+
2329
@classmethod
24-
def get_script_name(cls):
25-
return "main.py"
30+
def tearDownClass(cls):
31+
# Remove the WEBSITE_HOSTNAME environment variable
32+
os.environ.pop('PYTHON_SCRIPT_FILE_NAME')
33+
super().tearDownClass()
2634

2735
@classmethod
2836
def get_script_dir(cls):

0 commit comments

Comments
 (0)