gitlab-org--gitlab-foss/spec/services/clusters/applications/upgrade_service_spec.rb

# frozen_string_literal: true

require 'spec_helper'

describe Clusters::Applications::UpgradeService do
  describe '#execute' do
    let(:application) { create(:clusters_applications_helm, :scheduled) }
    let!(:install_command) { application.install_command }
    let(:service) { described_class.new(application) }
    let(:helm_client) { instance_double(Gitlab::Kubernetes::Helm::Api) }

    before do
      allow(service).to receive(:install_command).and_return(install_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(install_command)
        allow(ClusterWaitForAppInstallationWorker).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 'schedule async installation status check' do
        expect(ClusterWaitForAppInstallationWorker).to receive(:perform_in).once

        service.execute
      end
    end

    context 'when kubernetes cluster communication fails' do
      let(:error) { Kubeclient::HttpError.new(500, 'system failure', nil) }

      before do
        expect(helm_client).to receive(:update).with(install_command).and_raise(error)
      end

      it 'make the application errored' do
        service.execute

        expect(application).to be_update_errored
        expect(application.status_reason).to match('Kubernetes error: 500')
      end

      it 'logs errors' do
        expect(service.send(:logger)).to receive(:error).with(
          {
            exception: 'Kubeclient::HttpError',
            message: 'system failure',
            service: 'Clusters::Applications::UpgradeService',
            app_id: application.id,
            project_ids: application.cluster.project_ids,
            group_ids: [],
            error_code: 500
          }
        )

        expect(Gitlab::Sentry).to receive(:track_acceptable_exception).with(
          error,
          extra: {
            exception: 'Kubeclient::HttpError',
            message: 'system failure',
            service: 'Clusters::Applications::UpgradeService',
            app_id: application.id,
            project_ids: application.cluster.project_ids,
            group_ids: [],
            error_code: 500
          }
        )

        service.execute
      end
    end

    context 'a non kubernetes error happens' do
      let(:application) { create(:clusters_applications_helm, :scheduled) }
      let(:error) { StandardError.new('something bad happened') }

      before do
        expect(application).to receive(:make_updating!).once.and_raise(error)
      end

      it 'make the application errored' do
        expect(helm_client).not_to receive(:update)

        service.execute

        expect(application).to be_update_errored
        expect(application.status_reason).to eq("Can't start upgrade process.")
      end

      it 'logs errors' do
        expect(service.send(:logger)).to receive(:error).with(
          {
            exception: 'StandardError',
            error_code: nil,
            message: 'something bad happened',
            service: 'Clusters::Applications::UpgradeService',
            app_id: application.id,
            project_ids: application.cluster.projects.pluck(:id),
            group_ids: []
          }
        )

        expect(Gitlab::Sentry).to receive(:track_acceptable_exception).with(
          error,
          extra: {
            exception: 'StandardError',
            error_code: nil,
            message: 'something bad happened',
            service: 'Clusters::Applications::UpgradeService',
            app_id: application.id,
            project_ids: application.cluster.projects.pluck(:id),
            group_ids: []
          }
        )

        service.execute
      end
    end
  end
end