@@ -35,11 +35,14 @@ def setUp(self):
3535
3636 @mock .patch ("hyperpod_cli.service.get_logs.GetLogs" )
3737 @mock .patch ("hyperpod_cli.service.get_logs.GetLogs.get_training_job_logs" )
38+ @mock .patch ("hyperpod_cli.service.get_logs.GetLogs.generate_cloudwatch_link" )
3839 def test_get_logs_happy_case (
3940 self ,
41+ mock_get_logs_service_and_generate_cloudwatch_link : mock .Mock ,
4042 mock_get_logs_service_and_get_logs : mock .Mock ,
4143 mock_get_logs_service : mock .Mock ,
4244 ):
45+ mock_get_logs_service_and_generate_cloudwatch_link .return_value = 'link'
4346 mock_get_logs_service .return_value = self .mock_get_job_log
4447 mock_get_logs_service_and_get_logs .return_value = "{}"
4548 result = self .runner .invoke (
@@ -55,15 +58,19 @@ def test_get_logs_happy_case(
5558
5659 @mock .patch ("hyperpod_cli.service.get_logs.GetLogs" )
5760 @mock .patch ("hyperpod_cli.service.get_logs.GetLogs.get_training_job_logs" )
61+ @mock .patch ("hyperpod_cli.service.get_logs.GetLogs.generate_cloudwatch_link" )
5862 @mock .patch ("logging.Logger.debug" )
5963 def test_get_logs_happy_case_debug_mode (
6064 self ,
6165 mock_debug : mock .Mock ,
66+ mock_get_logs_service_and_generate_cloudwatch_link : mock .Mock ,
6267 mock_get_logs_service_and_get_logs : mock .Mock ,
6368 mock_get_logs_service : mock .Mock ,
6469 ):
6570 mock_get_logs_service .return_value = self .mock_get_job_log
6671 mock_get_logs_service_and_get_logs .return_value = "{}"
72+ mock_get_logs_service_and_generate_cloudwatch_link .return_value = 'link'
73+
6774 result = self .runner .invoke (
6875 get_log ,
6976 [
@@ -79,13 +86,17 @@ def test_get_logs_happy_case_debug_mode(
7986
8087 @mock .patch ("hyperpod_cli.service.get_logs.GetLogs" )
8188 @mock .patch ("hyperpod_cli.service.get_logs.GetLogs.get_training_job_logs" )
89+ @mock .patch ("hyperpod_cli.service.get_logs.GetLogs.generate_cloudwatch_link" )
8290 def test_describe_job_happy_case_with_namespace (
8391 self ,
92+ mock_get_logs_service_and_generate_cloudwatch_link : mock .Mock ,
8493 mock_get_logs_service_and_get_logs : mock .Mock ,
8594 mock_get_logs_service : mock .Mock ,
8695 ):
8796 mock_get_logs_service .return_value = self .mock_get_job_log
8897 mock_get_logs_service_and_get_logs .return_value = "{}"
98+ mock_get_logs_service_and_generate_cloudwatch_link .return_value = 'link'
99+
89100 result = self .runner .invoke (
90101 get_log ,
91102 [
0 commit comments