test: added unit tests for output command

Author: sai-sankalp-reddy

test/unit_tests/test_job.py

@@ -183,6 +183,56 @@ def test_list_job_happy_case_debug_mode(
         self.assertIn("jobs", result.output)
         mock_debug.assert_called()
 
+    @mock.patch("hyperpod_cli.service.list_training_jobs.ListTrainingJobs")
+    @mock.patch(
+        "hyperpod_cli.service.list_training_jobs.ListTrainingJobs.list_training_jobs"
+    )
+    @mock.patch("logging.Logger.debug")
+    def test_list_job_happy_case_default_output_mode_json(
+        self,
+        mock_debug,
+        mock_list_training_job_service_and_list_jobs: mock.Mock,
+        mock_list_training_job_service: mock.Mock,
+    ):
+        mock_list_training_job_service.return_value = self.mock_list_jobs
+        mock_list_training_job_service_and_list_jobs.return_value = json.dumps({"jobs": [{
+            "Name": "test-job-name",
+            "Namespace": "test_namespace",
+            "CreationTime": "2025-01-01T01:01:01Z",
+            "State": "Succeeded"
+        }]})
+        result = self.runner.invoke(list_jobs, ["--output", "json"])
+        self.assertEqual(result.exit_code, 0)
+        print(result.output)
+        expected_output = '{\n    "jobs": [\n        {\n            "Name": "test-job-name",\n            "Namespace": "test_namespace",\n            "CreationTime": "2025-01-01T01:01:01Z",\n            "State": "Succeeded"\n        }\n    ]\n}\n'
+        self.assertEqual(expected_output, result.output)
+        mock_debug.assert_called()
+
+    @mock.patch("hyperpod_cli.service.list_training_jobs.ListTrainingJobs")
+    @mock.patch(
+        "hyperpod_cli.service.list_training_jobs.ListTrainingJobs.list_training_jobs"
+    )
+    @mock.patch("logging.Logger.debug")
+    def test_list_job_happy_case_default_output_mode_table(
+        self,
+        mock_debug,
+        mock_list_training_job_service_and_list_jobs: mock.Mock,
+        mock_list_training_job_service: mock.Mock,
+    ):
+        mock_list_training_job_service.return_value = self.mock_list_jobs
+        mock_list_training_job_service_and_list_jobs.return_value = json.dumps({"jobs": [{
+            "Name": "test-job-name",
+            "Namespace": "test_namespace",
+            "CreationTime": "2025-01-01T01:01:01Z",
+            "State": "Succeeded"
+        }]})
+        result = self.runner.invoke(list_jobs, ["--output", "table"])
+        self.assertEqual(result.exit_code, 0)
+        print(result.output)
+        expected_output = ' Name          | Namespace      | CreationTime         | State\n---------------+----------------+----------------------+-----------\n test-job-name | test_namespace | 2025-01-01T01:01:01Z | Succeeded\n'
+        self.assertEqual(expected_output, result.output)
+        mock_debug.assert_called()
+
     @mock.patch("hyperpod_cli.service.list_training_jobs.ListTrainingJobs")
     @mock.patch(
         "hyperpod_cli.service.list_training_jobs.ListTrainingJobs.list_training_jobs"