@@ -1003,6 +1003,17 @@ def login(self, username=None, password=None):
1003
1003
role = role_admin ,
1004
1004
password = 'test' )
1005
1005
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
+
1006
1017
dag_acl_role = self .appbuilder .sm .add_role ('dag_acl_tester' )
1007
1018
dag_tester = self .appbuilder .sm .find_user (username = 'dag_tester' )
1008
1019
if not dag_tester :
@@ -1446,6 +1457,50 @@ def test_tree_success_for_read_only_role(self):
1446
1457
resp = self .client .get (url , follow_redirects = True )
1447
1458
self .check_content_in_response ('runme_1' , resp )
1448
1459
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
+
1449
1504
1450
1505
if __name__ == '__main__' :
1451
1506
unittest .main ()
0 commit comments