# frozen_string_literal: true require 'spec_helper' RSpec.describe Clusters::Applications::PrometheusUpdateService do describe '#execute' do let(:project) { create(:project) } let(:environment) { create(:environment, project: project) } let(:cluster) { create(:cluster, :provided_by_user, :with_installed_helm, projects: [project]) } let(:application) { create(:clusters_applications_prometheus, :installed, cluster: cluster) } let(:empty_alerts_values_update_yaml) { "---\nalertmanager:\n enabled: false\nserverFiles:\n alerts: {}\n" } let(:helm_client) { instance_double(::Gitlab::Kubernetes::Helm::API) } subject(:service) { described_class.new(application, project) } context 'when prometheus is a Clusters::Integrations::Prometheus' do let(:application) { create(:clusters_integrations_prometheus, cluster: cluster) } it 'raises NotImplementedError' do expect { service.execute }.to raise_error(NotImplementedError) end end context 'when prometheus is externally installed' do let(:application) { create(:clusters_applications_prometheus, :externally_installed, cluster: cluster) } it 'raises NotImplementedError' do expect { service.execute }.to raise_error(NotImplementedError) end end context 'when prometheus is a Clusters::Applications::Prometheus' do let!(:patch_command) { application.patch_command(empty_alerts_values_update_yaml) } before do allow(service).to receive(:patch_command).with(empty_alerts_values_update_yaml).and_return(patch_command) allow(service).to receive(:helm_api).and_return(helm_client) end context 'when there are no errors' do before do expect(helm_client).to receive(:update).with(patch_command) allow(::ClusterWaitForAppUpdateWorker) .to receive(:perform_in) .and_return(nil) end it 'make the application updating' do expect(application.cluster).not_to be_nil service.execute expect(application).to be_updating end it 'updates current config' do prometheus_config_service = spy(:prometheus_config_service) expect(Clusters::Applications::PrometheusConfigService) .to receive(:new) .with(project, cluster, application) .and_return(prometheus_config_service) expect(prometheus_config_service) .to receive(:execute) .and_return(YAML.safe_load(empty_alerts_values_update_yaml)) service.execute end it 'schedules async update status check' do expect(::ClusterWaitForAppUpdateWorker).to receive(:perform_in).once service.execute end end context 'when k8s cluster communication fails' do before do error = ::Kubeclient::HttpError.new(500, 'system failure', nil) allow(helm_client).to receive(:update).and_raise(error) end it 'make the application update errored' do service.execute expect(application).to be_update_errored expect(application.status_reason).to match(/kubernetes error:/i) end end context 'when application cannot be persisted' do let(:application) { build(:clusters_applications_prometheus, :installed) } before do allow(application).to receive(:make_updating!).once .and_raise(ActiveRecord::RecordInvalid.new(application)) end it 'make the application update errored' do expect(helm_client).not_to receive(:update) service.execute expect(application).to be_update_errored end end end end end