Skip to content

Commit e57954f

Browse files
author
pintaoz
committed
Add test_job test
1 parent 8722607 commit e57954f

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

test/unit_tests/service/test_list_training_jobs_service.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -273,6 +273,16 @@ def test_list_training_jobs_all_namespace_missing_status(
273273
result = self.mock_list_training_jobs.list_training_jobs(None, True, None, None)
274274
self.assertNotIn("State: null", result)
275275

276+
@mock.patch("hyperpod_cli.clients.kubernetes_client.KubernetesClient.__new__")
277+
def test_list_training_jobs_namespace_not_exist(
278+
self,
279+
mock_kubernetes_client: mock.Mock,
280+
):
281+
mock_kubernetes_client.return_value = self.mock_k8s_client
282+
self.mock_k8s_client.check_if_namespace_exists.return_value = False
283+
with self.assertRaises(ValueError):
284+
self.mock_list_training_jobs.list_training_jobs("abcdef", False, None, None)
285+
276286
def test_generate_table_with_no_priority_header_and_values(self):
277287
list_training_jobs = ListTrainingJobs()
278288
output_jobs = {

0 commit comments

Comments
 (0)