diff --git a/ee/app/services/cloud_connector/status_checks/probes/registry.rb b/ee/app/services/cloud_connector/status_checks/probes/registry.rb index e66ce72f0c611ce096fd591e8ac2cd00d44eea44..c5e6ef3c4c4f3ce7324a8e225cf08008b8c58124 100644 --- a/ee/app/services/cloud_connector/status_checks/probes/registry.rb +++ b/ee/app/services/cloud_connector/status_checks/probes/registry.rb @@ -54,17 +54,11 @@ def probes_for_vendored_features end def self_hosted_only_probes - probes = [ + [ ::CloudConnector::StatusChecks::Probes::SelfHosted::AiGatewayUrlPresenceProbe.new, ::CloudConnector::StatusChecks::Probes::HostProbe.new(::Gitlab::AiGateway.self_hosted_url), ::CloudConnector::StatusChecks::Probes::SelfHosted::CodeSuggestionsLicenseProbe.new(@user) ] - - if Gitlab::DuoWorkflow::Client.self_hosted_url.present? - probes << ::CloudConnector::StatusChecks::Probes::SelfHosted::DuoAgentPlatformProbe.new(@user) - end - - probes end def at_least_one_vendored_feature? diff --git a/ee/spec/services/cloud_connector/status_checks/probes/registry_spec.rb b/ee/spec/services/cloud_connector/status_checks/probes/registry_spec.rb index 81adf7acf4693cc0200c8c7aaa83f479df9c8c35..2bb6d860675c222acf2c11b0517b25b24f94e784 100644 --- a/ee/spec/services/cloud_connector/status_checks/probes/registry_spec.rb +++ b/ee/spec/services/cloud_connector/status_checks/probes/registry_spec.rb @@ -120,20 +120,5 @@ end end end - - context 'when Duo Agent Platform URL is set up' do - before do - Ai::Setting.instance.update!(duo_agent_platform_service_url: 'localhost:50052') - end - - it 'returns self-hosted probes combined with default probes including DuoAgentPlatformProbe' do - probes = registry.self_hosted_probes - - expect(probes).to match( - self_hosted_only_probe_types + - [an_instance_of(CloudConnector::StatusChecks::Probes::SelfHosted::DuoAgentPlatformProbe)] - ) - end - end end end