diff --git a/ee/app/services/cloud_connector/status_checks/probes/self_hosted/duo_agent_platform_probe.rb b/ee/app/services/cloud_connector/status_checks/probes/self_hosted/duo_agent_platform_probe.rb index dc607dc0593b854b6cc84550683f63d624d7054a..f2d020871f3c80a46f187539f26c4eb6de559783 100644 --- a/ee/app/services/cloud_connector/status_checks/probes/self_hosted/duo_agent_platform_probe.rb +++ b/ee/app/services/cloud_connector/status_checks/probes/self_hosted/duo_agent_platform_probe.rb @@ -21,7 +21,7 @@ def result duo_workflow_service_url: @host, current_user: @user, secure: Gitlab::DuoWorkflow::Client.secure? - ).generate_token + ).list_tools end def verify_request_success diff --git a/ee/spec/services/cloud_connector/status_checks/probes/self_hosted/duo_agent_platform_probe_spec.rb b/ee/spec/services/cloud_connector/status_checks/probes/self_hosted/duo_agent_platform_probe_spec.rb index c955d2d690b552c2f0ed30906697288d4627f9f6..b7641c0f432c73aac8860acaf5952dbb1bf60019 100644 --- a/ee/spec/services/cloud_connector/status_checks/probes/self_hosted/duo_agent_platform_probe_spec.rb +++ b/ee/spec/services/cloud_connector/status_checks/probes/self_hosted/duo_agent_platform_probe_spec.rb @@ -17,11 +17,11 @@ allow(Ai::DuoWorkflow::DuoWorkflowService::Client).to receive(:new).and_return(duo_workflow_client) end - context 'when generate_token succeeds' do - let(:success_response) { { status: :success, message: 'JWT Generated', payload: { token: 'test-token' } } } + context 'when list_tools succeeds' do + let(:success_response) { { status: :success, message: 'Tools listed successfully', payload: { tools: [] } } } before do - allow(duo_workflow_client).to receive(:generate_token).and_return(success_response) + allow(duo_workflow_client).to receive(:list_tools).and_return(success_response) end it 'returns a successful result' do @@ -40,15 +40,15 @@ current_user: user, secure: true ) - expect(duo_workflow_client).to have_received(:generate_token) + expect(duo_workflow_client).to have_received(:list_tools) end end - context 'when generate_token fails' do + context 'when list_tools fails' do let(:error_response) { { status: :error, message: 'Connection failed' } } before do - allow(duo_workflow_client).to receive(:generate_token).and_return(error_response) + allow(duo_workflow_client).to receive(:list_tools).and_return(error_response) end it 'returns a failed result' do