summaryrefslogtreecommitdiff
path: root/spec/lib/gitlab/kubernetes/rollout_instances_spec.rb
blob: 3ac97ddc75d1cf6639e50abb669442fd04535504 (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
# frozen_string_literal: true

require 'spec_helper'

RSpec.describe Gitlab::Kubernetes::RolloutInstances do
  include KubernetesHelpers

  def setup(deployments_attrs, pods_attrs)
    deployments = deployments_attrs.map do |attrs|
      ::Gitlab::Kubernetes::Deployment.new(attrs, pods: pods_attrs)
    end

    pods = pods_attrs.map do |attrs|
      ::Gitlab::Kubernetes::Pod.new(attrs)
    end

    [deployments, pods]
  end

  describe '#pod_instances' do
    it 'returns an instance for a deployment with one pod' do
      deployments, pods = setup(
        [kube_deployment(name: 'one', track: 'stable', replicas: 1)],
        [kube_pod(name: 'one', status: 'Running', track: 'stable')]
      )
      rollout_instances = described_class.new(deployments, pods)

      expect(rollout_instances.pod_instances).to eq([{
        pod_name: 'one',
        stable: true,
        status: 'running',
        tooltip: 'one (Running)',
        track: 'stable'
      }])
    end

    it 'returns a pending pod for a missing replica' do
      deployments, pods = setup(
        [kube_deployment(name: 'one', track: 'stable', replicas: 1)],
        []
      )
      rollout_instances = described_class.new(deployments, pods)

      expect(rollout_instances.pod_instances).to eq([{
        pod_name: 'Not provided',
        stable: true,
        status: 'pending',
        tooltip: 'Not provided (Pending)',
        track: 'stable'
      }])
    end

    it 'returns instances when there are two stable deployments' do
      deployments, pods = setup([
        kube_deployment(name: 'one', track: 'stable', replicas: 1),
        kube_deployment(name: 'two', track: 'stable', replicas: 1)
      ], [
        kube_pod(name: 'one', status: 'Running', track: 'stable'),
        kube_pod(name: 'two', status: 'Running', track: 'stable')
      ])
      rollout_instances = described_class.new(deployments, pods)

      expect(rollout_instances.pod_instances).to eq([{
        pod_name: 'one',
        stable: true,
        status: 'running',
        tooltip: 'one (Running)',
        track: 'stable'
      }, {
        pod_name: 'two',
        stable: true,
        status: 'running',
        tooltip: 'two (Running)',
        track: 'stable'
      }])
    end

    it 'returns instances for two deployments with different tracks' do
      deployments, pods = setup([
        kube_deployment(name: 'one', track: 'mytrack', replicas: 1),
        kube_deployment(name: 'two', track: 'othertrack', replicas: 1)
      ], [
        kube_pod(name: 'one', status: 'Running', track: 'mytrack'),
        kube_pod(name: 'two', status: 'Running', track: 'othertrack')
      ])
      rollout_instances = described_class.new(deployments, pods)

      expect(rollout_instances.pod_instances).to eq([{
        pod_name: 'one',
        stable: false,
        status: 'running',
        tooltip: 'one (Running)',
        track: 'mytrack'
      }, {
        pod_name: 'two',
        stable: false,
        status: 'running',
        tooltip: 'two (Running)',
        track: 'othertrack'
      }])
    end

    it 'sorts stable tracks after canary tracks' do
      deployments, pods = setup([
        kube_deployment(name: 'one', track: 'stable', replicas: 1),
        kube_deployment(name: 'two', track: 'canary', replicas: 1)
      ], [
        kube_pod(name: 'one', status: 'Running', track: 'stable'),
        kube_pod(name: 'two', status: 'Running', track: 'canary')
      ])
      rollout_instances = described_class.new(deployments, pods)

      expect(rollout_instances.pod_instances).to eq([{
        pod_name: 'two',
        stable: false,
        status: 'running',
        tooltip: 'two (Running)',
        track: 'canary'
      }, {
        pod_name: 'one',
        stable: true,
        status: 'running',
        tooltip: 'one (Running)',
        track: 'stable'
      }])
    end
  end
end