summaryrefslogtreecommitdiff
path: root/spec/lib/gitlab/kubernetes/helm/install_command_spec.rb
blob: 63997a40d52c705ddbd82647e6aaab88c5890fc9 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
require 'rails_helper'

describe Gitlab::Kubernetes::Helm::InstallCommand do
  let(:prometheus) { create(:clusters_applications_prometheus) }

  describe "#initialize" do
    context "With all the params" do
      subject { described_class.new(prometheus.name, install_helm: true, chart: prometheus.chart, chart_values_file: prometheus.chart_values_file) }

      it 'should assign all parameters' do
        expect(subject.name).to eq(prometheus.name)
        expect(subject.install_helm).to be_truthy
        expect(subject.chart).to eq(prometheus.chart)
        expect(subject.chart_values_file).to eq("#{Rails.root}/vendor/prometheus/values.yaml")
      end
    end

    context 'when install_helm is not set' do
      subject { described_class.new(prometheus.name, chart: prometheus.chart, chart_values_file: true) }

      it 'should set install_helm as false' do
        expect(subject.install_helm).to be_falsy
      end
    end

    context 'when chart is not set' do
      subject { described_class.new(prometheus.name, install_helm: true) }

      it 'should set chart as nil' do
        expect(subject.chart).to be_falsy
      end
    end

    context 'when chart_values_file is not set' do
      subject { described_class.new(prometheus.name, install_helm: true, chart: prometheus.chart) }

      it 'should set chart_values_file as nil' do
        expect(subject.chart_values_file).to be_falsy
      end
    end
  end

  describe "#generate_script" do
    let(:install_command) { described_class.new(prometheus.name, install_helm: install_helm) }
    let(:client) { double('kubernetes client') }
    let(:namespace) { Gitlab::Kubernetes::Namespace.new(Gitlab::Kubernetes::Helm::NAMESPACE, client) }
    subject { install_command.send(:generate_script, namespace.name) }

    context 'when install helm is true' do
      let(:install_helm) { true }
      let(:command) do
        <<~MSG
        set -eo pipefail
        apk add -U ca-certificates openssl >/dev/null
        wget -q -O - https://kubernetes-helm.storage.googleapis.com/helm-v2.7.0-linux-amd64.tar.gz | tar zxC /tmp >/dev/null
        mv /tmp/linux-amd64/helm /usr/bin/

        helm init >/dev/null
        MSG
      end

      it 'should return appropriate command' do
        is_expected.to eq(command)
      end
    end

    context 'when install helm is false' do
      let(:install_helm) { false }
      let(:command) do
        <<~MSG
        set -eo pipefail
        apk add -U ca-certificates openssl >/dev/null
        wget -q -O - https://kubernetes-helm.storage.googleapis.com/helm-v2.7.0-linux-amd64.tar.gz | tar zxC /tmp >/dev/null
        mv /tmp/linux-amd64/helm /usr/bin/

        helm init --client-only >/dev/null
        MSG
      end

      it 'should return appropriate command' do
        is_expected.to eq(command)
      end
    end

    context 'when chart is present' do
      let(:install_command) { described_class.new(prometheus.name, chart: prometheus.chart) }
      let(:command) do
        <<~MSG.chomp
        set -eo pipefail
        apk add -U ca-certificates openssl >/dev/null
        wget -q -O - https://kubernetes-helm.storage.googleapis.com/helm-v2.7.0-linux-amd64.tar.gz | tar zxC /tmp >/dev/null
        mv /tmp/linux-amd64/helm /usr/bin/

        helm init --client-only >/dev/null
        helm install #{prometheus.chart} --name #{prometheus.name} --namespace #{namespace.name} >/dev/null
        MSG
      end

      it 'should return appropriate command' do
        is_expected.to eq(command)
      end
    end

    context 'when chart values file is present' do
      let(:install_command) { described_class.new(prometheus.name, chart: prometheus.chart, chart_values_file: prometheus.chart_values_file) }
      let(:command) do
        <<~MSG.chomp
        set -eo pipefail
        apk add -U ca-certificates openssl >/dev/null
        wget -q -O - https://kubernetes-helm.storage.googleapis.com/helm-v2.7.0-linux-amd64.tar.gz | tar zxC /tmp >/dev/null
        mv /tmp/linux-amd64/helm /usr/bin/

        helm init --client-only >/dev/null
        helm install #{prometheus.chart} --name #{prometheus.name} --namespace #{namespace.name} -f /data/helm/#{prometheus.name}/config/values.yaml >/dev/null
        MSG
      end

      it 'should return appropriate command' do
        is_expected.to eq(command)
      end
    end
  end

  describe "#pod_name" do
    let(:install_command) { described_class.new(prometheus.name, install_helm: true, chart: prometheus.chart, chart_values_file: true) }
    subject { install_command.send(:pod_name) }

    it { is_expected.to eq('install-prometheus') }
  end
end