deleted set_result_for_command

Author: DannyLiCom

src/xpk/commands/managed_ml_diagnostics_test.py

@@ -50,12 +50,12 @@ def mocks(mocker) -> _Mocks:
       commands_print_mock=commands_print_mock,
       commands_tester=CommandsTester(
           mocker,
-        #   run_command_with_updates_path=(
-        #       'xpk.commands.managed_ml_diagnostics.run_command_with_updates'
-        #   ),
-        #   run_command_for_value_path=(
-        #       'xpk.commands.managed_ml_diagnostics.run_command_for_value'
-        #   ),
+          run_command_with_updates_path=(
+              'xpk.commands.managed_ml_diagnostics.run_command_with_updates'
+          ),
+          run_command_for_value_path=(
+              'xpk.commands.managed_ml_diagnostics.run_command_for_value'
+          ),
       ),
   )
 
@@ -64,63 +64,63 @@ def test_install_mldiagnostics_prerequisites_commands_executed(
     mocker,
 ):
 
-  mocks.commands_tester.set_result_for_command(
-      (0, ''),
-      'kubectl',
-      'rollout',
-      'status',
-      'deployment/kueue-controller-manager',
-  )
-
-  mocks.commands_tester.set_result_for_command(
-      (0, ''),
-      'kubectl',
-      'rollout',
-      'status',
-      'deployment/cert-manager-webhook',
-  )
-
-  mocks.commands_tester.set_result_for_command(
-      (0, ''),
-      'kubectl',
-      'apply',
-      '-f',
-      'https://github.com/cert-manager/cert-manager/releases/',
-  )
-
-  mocks.commands_tester.set_result_for_command(
-      (0, ''),
-      'gcloud',
-      'artifacts',
-      'generic',
-      'download',
-  )
-
-  mocks.commands_tester.set_result_for_command(
-      (0, ''),
-      'kubectl',
-      'create',
-      'namespace',
-      'gke-mldiagnostics',
-  )
-
-  mocks.commands_tester.set_result_for_command(
-      (0, ''),
-      'kubectl',
-      'apply',
-      '-f',
-      '-n',
-      'gke-mldiagnostics',
-  )
-
-  mocks.commands_tester.set_result_for_command(
-      (0, ''),
-      'kubectl',
-      'label',
-      'namespace',
-      'default',
-      'managed-mldiagnostics-gke=true',
-  )
+#   mocks.commands_tester.set_result_for_command(
+#       (0, ''),
+#       'kubectl',
+#       'rollout',
+#       'status',
+#       'deployment/kueue-controller-manager',
+#   )
+
+#   mocks.commands_tester.set_result_for_command(
+#       (0, ''),
+#       'kubectl',
+#       'rollout',
+#       'status',
+#       'deployment/cert-manager-webhook',
+#   )
+
+#   mocks.commands_tester.set_result_for_command(
+#       (0, ''),
+#       'kubectl',
+#       'apply',
+#       '-f',
+#       'https://github.com/cert-manager/cert-manager/releases/',
+#   )
+
+#   mocks.commands_tester.set_result_for_command(
+#       (0, ''),
+#       'gcloud',
+#       'artifacts',
+#       'generic',
+#       'download',
+#   )
+
+#   mocks.commands_tester.set_result_for_command(
+#       (0, ''),
+#       'kubectl',
+#       'create',
+#       'namespace',
+#       'gke-mldiagnostics',
+#   )
+
+#   mocks.commands_tester.set_result_for_command(
+#       (0, ''),
+#       'kubectl',
+#       'apply',
+#       '-f',
+#       '-n',
+#       'gke-mldiagnostics',
+#   )
+
+#   mocks.commands_tester.set_result_for_command(
+#       (0, ''),
+#       'kubectl',
+#       'label',
+#       'namespace',
+#       'default',
+#       'managed-mldiagnostics-gke=true',
+#   )
 
   install_mldiagnostics_prerequisites()