Skip to content

Commit 2b542f3

Browse files
seelmannChris Fei
authored and
Chris Fei
committed
Assign permission get_logs_with_metadata to viewer role (apache#3913)
1 parent 078c7b0 commit 2b542f3

File tree

1 file changed

+55
-0
lines changed

1 file changed

+55
-0
lines changed

tests/www_rbac/test_views.py

+55
Original file line numberDiff line numberDiff line change
@@ -1003,6 +1003,17 @@ def login(self, username=None, password=None):
10031003
role=role_admin,
10041004
password='test')
10051005

1006+
role_user = self.appbuilder.sm.find_role('User')
1007+
test_user = self.appbuilder.sm.find_user(username='test_user')
1008+
if not test_user:
1009+
self.appbuilder.sm.add_user(
1010+
username='test_user',
1011+
first_name='test_user',
1012+
last_name='test_user',
1013+
email='test_user@fab.org',
1014+
role=role_user,
1015+
password='test_user')
1016+
10061017
dag_acl_role = self.appbuilder.sm.add_role('dag_acl_tester')
10071018
dag_tester = self.appbuilder.sm.find_user(username='dag_tester')
10081019
if not dag_tester:
@@ -1446,6 +1457,50 @@ def test_tree_success_for_read_only_role(self):
14461457
resp = self.client.get(url, follow_redirects=True)
14471458
self.check_content_in_response('runme_1', resp)
14481459

1460+
def test_log_success(self):
1461+
self.logout()
1462+
self.login()
1463+
url = ('log?task_id=runme_0&dag_id=example_bash_operator&execution_date={}'
1464+
.format(self.percent_encode(self.default_date)))
1465+
resp = self.client.get(url, follow_redirects=True)
1466+
self.check_content_in_response('Log by attempts', resp)
1467+
url = ('get_logs_with_metadata?task_id=runme_0&dag_id=example_bash_operator&'
1468+
'execution_date={}&try_number=1&metadata=null'
1469+
.format(self.percent_encode(self.default_date)))
1470+
resp = self.client.get(url, follow_redirects=True)
1471+
self.check_content_in_response('"message":', resp)
1472+
self.check_content_in_response('"metadata":', resp)
1473+
1474+
def test_log_failure(self):
1475+
self.logout()
1476+
self.login(username='dag_faker',
1477+
password='dag_faker')
1478+
url = ('log?task_id=runme_0&dag_id=example_bash_operator&execution_date={}'
1479+
.format(self.percent_encode(self.default_date)))
1480+
resp = self.client.get(url, follow_redirects=True)
1481+
self.check_content_not_in_response('Log by attempts', resp)
1482+
url = ('get_logs_with_metadata?task_id=runme_0&dag_id=example_bash_operator&'
1483+
'execution_date={}&try_number=1&metadata=null'
1484+
.format(self.percent_encode(self.default_date)))
1485+
resp = self.client.get(url, follow_redirects=True)
1486+
self.check_content_not_in_response('"message":', resp)
1487+
self.check_content_not_in_response('"metadata":', resp)
1488+
1489+
def test_log_success_for_user(self):
1490+
self.logout()
1491+
self.login(username='test_user',
1492+
password='test_user')
1493+
url = ('log?task_id=runme_0&dag_id=example_bash_operator&execution_date={}'
1494+
.format(self.percent_encode(self.default_date)))
1495+
resp = self.client.get(url, follow_redirects=True)
1496+
self.check_content_in_response('Log by attempts', resp)
1497+
url = ('get_logs_with_metadata?task_id=runme_0&dag_id=example_bash_operator&'
1498+
'execution_date={}&try_number=1&metadata=null'
1499+
.format(self.percent_encode(self.default_date)))
1500+
resp = self.client.get(url, follow_redirects=True)
1501+
self.check_content_in_response('"message":', resp)
1502+
self.check_content_in_response('"metadata":', resp)
1503+
14491504

14501505
if __name__ == '__main__':
14511506
unittest.main()

0 commit comments

Comments
 (0)