gitlab-org--gitlab-foss/spec/models/clusters/applications/elastic_stack_spec.rb

# frozen_string_literal: true

require 'spec_helper'

RSpec.describe Clusters::Applications::ElasticStack do
  include KubernetesHelpers

  include_examples 'cluster application core specs', :clusters_applications_elastic_stack
  include_examples 'cluster application status specs', :clusters_applications_elastic_stack
  include_examples 'cluster application version specs', :clusters_applications_elastic_stack
  include_examples 'cluster application helm specs', :clusters_applications_elastic_stack

  describe 'cluster.integration_elastic_stack state synchronization' do
    let!(:application) { create(:clusters_applications_elastic_stack) }
    let(:cluster) { application.cluster }
    let(:integration) { cluster.integration_elastic_stack }

    describe 'after_destroy' do
      it 'disables the corresponding integration' do
        application.destroy!

        expect(integration).not_to be_enabled
      end
    end

    describe 'on install' do
      it 'enables the corresponding integration' do
        application.make_scheduled!
        application.make_installing!
        application.make_installed!

        expect(integration).to be_enabled
      end
    end

    describe 'on uninstall' do
      it 'disables the corresponding integration' do
        application.make_scheduled!
        application.make_installing!
        application.make_installed!
        application.make_externally_uninstalled!

        expect(integration).not_to be_enabled
      end
    end
  end

  describe '#install_command' do
    let!(:elastic_stack) { create(:clusters_applications_elastic_stack) }

    subject { elastic_stack.install_command }

    it { is_expected.to be_an_instance_of(Gitlab::Kubernetes::Helm::V3::InstallCommand) }

    it 'is initialized with elastic stack arguments' do
      expect(subject.name).to eq('elastic-stack')
      expect(subject.chart).to eq('elastic-stack/elastic-stack')
      expect(subject.version).to eq('3.0.0')
      expect(subject.repository).to eq('https://charts.gitlab.io')
      expect(subject).to be_rbac
      expect(subject.files).to eq(elastic_stack.files)
      expect(subject.preinstall).to be_empty
    end

    context 'within values.yaml' do
      let(:values_yaml_content) {subject.files[:"values.yaml"]}

      it 'contains the disabled index lifecycle management' do
        expect(values_yaml_content).to include "setup.ilm.enabled: false"
      end

      it 'contains daily indices with respective template' do
        expect(values_yaml_content).to include "index: \"filebeat-%{[agent.version]}-%{+yyyy.MM.dd}\""
        expect(values_yaml_content).to include "setup.template.name: 'filebeat'"
        expect(values_yaml_content).to include "setup.template.pattern: 'filebeat-*'"
      end
    end

    context 'on a non rbac enabled cluster' do
      before do
        elastic_stack.cluster.platform_kubernetes.abac!
      end

      it { is_expected.not_to be_rbac }
    end

    context 'on versions older than 2' do
      before do
        elastic_stack.status = elastic_stack.status_states[:updating]
        elastic_stack.version = "1.9.0"
      end

      it 'includes a preinstall script' do
        expect(subject.preinstall).not_to be_empty
        expect(subject.preinstall.first).to include("helm uninstall")
      end
    end

    context 'on versions older than 3' do
      before do
        elastic_stack.status = elastic_stack.status_states[:updating]
        elastic_stack.version = "2.9.0"
      end

      it 'includes a preinstall script' do
        expect(subject.preinstall).not_to be_empty
        expect(subject.preinstall.first).to include("helm uninstall")
      end
    end

    context 'application failed to install previously' do
      let(:elastic_stack) { create(:clusters_applications_elastic_stack, :errored, version: '0.0.1') }

      it 'is initialized with the locked version' do
        expect(subject.version).to eq('3.0.0')
      end
    end
  end

  describe '#chart_above_v2?' do
    let(:elastic_stack) { create(:clusters_applications_elastic_stack, version: version) }

    subject { elastic_stack.chart_above_v2? }

    context 'on v1.9.0' do
      let(:version) { '1.9.0' }

      it { is_expected.to be_falsy }
    end

    context 'on v2.0.0' do
      let(:version) { '2.0.0' }

      it { is_expected.to be_truthy }
    end
  end

  describe '#chart_above_v3?' do
    let(:elastic_stack) { create(:clusters_applications_elastic_stack, version: version) }

    subject { elastic_stack.chart_above_v3? }

    context 'on v1.9.0' do
      let(:version) { '1.9.0' }

      it { is_expected.to be_falsy }
    end

    context 'on v3.0.0' do
      let(:version) { '3.0.0' }

      it { is_expected.to be_truthy }
    end
  end

  describe '#uninstall_command' do
    let!(:elastic_stack) { create(:clusters_applications_elastic_stack) }

    subject { elastic_stack.uninstall_command }

    it { is_expected.to be_an_instance_of(Gitlab::Kubernetes::Helm::V3::DeleteCommand) }

    it 'is initialized with elastic stack arguments' do
      expect(subject.name).to eq('elastic-stack')
      expect(subject).to be_rbac
      expect(subject.files).to eq(elastic_stack.files)
    end

    it 'specifies a post delete command to remove custom resource definitions' do
      expect(subject.postdelete).to eq([
        'kubectl delete pvc --selector app\\=elastic-stack-elasticsearch-master --namespace gitlab-managed-apps'
      ])
    end
  end

  it_behaves_like 'cluster-based #elasticsearch_client', :clusters_applications_elastic_stack
end