Skip to content

Commit 3b92cfc

Browse files
committed
fix unit tests
1 parent f753ece commit 3b92cfc

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

test/unit_tests/test_pod.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)