Skip to content

Commit

Permalink
Separate out tests to cater of changes in Python 3.8.8 (#14698)
Browse files Browse the repository at this point in the history
python/cpython#24297 change was included in
Python 3.8.8 to fix a vulnerability (bpo-42967)

Depending on which Base Python Image is run in our CI, two of the tests
can fail or succeed.

Our Previous two attempts:

- apache/airflow@061cd23
- apache/airflow@49952e7

We might for a while get different base python version depending on the changes of a PR (whether or not it includes a change to dockerfiler).
a) when you have PR which do not have changes in the Dockerfile, they will use the older python version as base (for example Python 3.8.7)
b) when you have PR that touches the Dockerfile and have setup.py changes in master, it should pull Python 3.8.8 first.

GitOrigin-RevId: ffe3bd29574d62a0a692cd8f63995856bbff8c0b
  • Loading branch information
kaxil authored and Cloud Composer Team committed Sep 17, 2021
1 parent 2f8c96d commit ce954f2
Showing 1 changed file with 66 additions and 6 deletions.
72 changes: 66 additions & 6 deletions tests/www/test_views.py
Original file line number Diff line number Diff line change
Expand Up @@ -2784,7 +2784,38 @@ def test_trigger_dag_form(self):
("%2Fgraph%3Fdag_id%3Dexample_bash_operator", "/graph?dag_id=example_bash_operator"),
]
)
def test_trigger_dag_form_origin_url(self, test_origin, expected_origin):
@pytest.mark.skipif(
sys.version_info < (3, 8, 8),
reason='Vulnerability was fixed in Python 3.8.8 which changed the query string separator: bpo-42967',
)
def test_trigger_dag_form_origin_url_py_lte_387(self, test_origin, expected_origin):
test_dag_id = "example_bash_operator"

resp = self.client.get(f'trigger?dag_id={test_dag_id}&origin={test_origin}')
self.check_content_in_response(
'<button type="button" class="btn" onclick="location.href = \'{}\'; return false">'.format(
expected_origin
),
resp,
)

@parameterized.expand(
[
("javascript:alert(1)", "/home"),
("http://google.com", "/home"),
(
"%2Ftree%3Fdag_id%3Dexample_bash_operator';alert(33)//",
"/tree?dag_id=example_bash_operator%27%3Balert%2833%29%2F%2F",
),
("%2Ftree%3Fdag_id%3Dexample_bash_operator", "/tree?dag_id=example_bash_operator"),
("%2Fgraph%3Fdag_id%3Dexample_bash_operator", "/graph?dag_id=example_bash_operator"),
]
)
@pytest.mark.skipif(
sys.version_info > (3, 8, 7),
reason='Vulnerability was fixed in Python 3.8.8 which changed the query string separator: bpo-42967',
)
def test_trigger_dag_form_origin_url_py_gt_387(self, test_origin, expected_origin):
test_dag_id = "example_bash_operator"

resp = self.client.get(f'trigger?dag_id={test_dag_id}&origin={test_origin}')
Expand Down Expand Up @@ -3329,11 +3360,40 @@ class TestHelperFunctions(TestBase):
),
]
)
@mock.patch("airflow.www.views.url_for")
def test_get_safe_url(self, test_url, expected_url, mock_url_for):
mock_url_for.return_value = "/home"
with self.app.test_request_context(base_url="http://localhost:8080"):
assert get_safe_url(test_url) == expected_url
@pytest.mark.skipif(
sys.version_info < (3, 8, 8),
reason='Vulnerability was fixed in Python 3.8.8 which changed the query string separator: bpo-42967',
)
def test_get_safe_url_py_lte_387(self, test_url, expected_url):
with mock.patch("airflow.www.views.url_for") as mock_url_for:
mock_url_for.return_value = "/home"
with self.app.test_request_context(base_url="http://localhost:8080"):
assert get_safe_url(test_url) == expected_url

@parameterized.expand(
[
("", "/home"),
("http://google.com", "/home"),
(
"http://localhost:8080/trigger?dag_id=test_dag&origin=%2Ftree%3Fdag_id%test_dag';alert(33)//",
"http://localhost:8080/trigger?dag_id=test_dag&origin=%2Ftree%3F"
"dag_id%25test_dag%27%3Balert%2833%29%2F%2F",
),
(
"http://localhost:8080/trigger?dag_id=test_dag&origin=%2Ftree%3Fdag_id%test_dag",
"http://localhost:8080/trigger?dag_id=test_dag&origin=%2Ftree%3Fdag_id%25test_dag",
),
]
)
@pytest.mark.skipif(
sys.version_info > (3, 8, 7),
reason='Vulnerability was fixed in Python 3.8.8 which changed the query string separator: bpo-42967',
)
def test_get_safe_url_py_gt_387(self, test_url, expected_url):
with mock.patch("airflow.www.views.url_for") as mock_url_for:
mock_url_for.return_value = "/home"
with self.app.test_request_context(base_url="http://localhost:8080"):
assert get_safe_url(test_url) == expected_url

@parameterized.expand(
[
Expand Down

0 comments on commit ce954f2

Please sign in to comment.