debian-mirror-gitlab/spec/lib/gitlab/kubernetes/helm/api_spec.rb

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

270 lines
9.1 KiB
Ruby
Raw Normal View History

2020-01-01 13:55:28 +05:30
# frozen_string_literal: true
2018-03-17 18:26:18 +05:30
require 'spec_helper'
2020-07-28 23:09:34 +05:30
RSpec.describe Gitlab::Kubernetes::Helm::API do
2018-03-17 18:26:18 +05:30
let(:client) { double('kubernetes client') }
let(:helm) { described_class.new(client) }
let(:gitlab_namespace) { Gitlab::Kubernetes::Helm::NAMESPACE }
2020-03-13 15:44:24 +05:30
let(:gitlab_namespace_labels) { Gitlab::Kubernetes::Helm::NAMESPACE_LABELS }
let(:namespace) { Gitlab::Kubernetes::Namespace.new(gitlab_namespace, client, labels: gitlab_namespace_labels) }
2018-11-20 20:47:30 +05:30
let(:application_name) { 'app-name' }
let(:rbac) { false }
let(:files) { {} }
let(:command) do
2021-01-29 00:20:46 +05:30
Gitlab::Kubernetes::Helm::V2::InstallCommand.new(
2018-11-20 20:47:30 +05:30
name: application_name,
chart: 'chart-name',
rbac: rbac,
2020-10-24 23:57:45 +05:30
files: files
2018-11-20 20:47:30 +05:30
)
end
2018-03-27 19:54:05 +05:30
2018-03-17 18:26:18 +05:30
subject { helm }
before do
2020-03-13 15:44:24 +05:30
allow(Gitlab::Kubernetes::Namespace).to(
receive(:new).with(gitlab_namespace, client, labels: gitlab_namespace_labels).and_return(namespace)
)
2018-03-27 19:54:05 +05:30
allow(client).to receive(:create_config_map)
2018-03-17 18:26:18 +05:30
end
describe '#initialize' do
it 'creates a namespace object' do
2020-03-13 15:44:24 +05:30
expect(Gitlab::Kubernetes::Namespace).to(
receive(:new).with(gitlab_namespace, client, labels: gitlab_namespace_labels)
)
2018-03-17 18:26:18 +05:30
subject
end
end
2019-07-31 22:56:46 +05:30
describe '#uninstall' do
before do
allow(client).to receive(:create_pod).and_return(nil)
allow(client).to receive(:get_config_map).and_return(nil)
allow(client).to receive(:create_config_map).and_return(nil)
allow(client).to receive(:delete_pod).and_return(nil)
allow(namespace).to receive(:ensure_exists!).once
end
it 'ensures the namespace exists before creating the POD' do
expect(namespace).to receive(:ensure_exists!).once.ordered
expect(client).to receive(:create_pod).once.ordered
subject.uninstall(command)
end
it 'removes an existing pod before installing' do
expect(client).to receive(:delete_pod).with('install-app-name', 'gitlab-managed-apps').once.ordered
expect(client).to receive(:create_pod).once.ordered
subject.uninstall(command)
end
context 'with a ConfigMap' do
let(:resource) { Gitlab::Kubernetes::ConfigMap.new(application_name, files).generate }
it 'creates a ConfigMap on kubeclient' do
expect(client).to receive(:create_config_map).with(resource).once
subject.install(command)
end
context 'config map already exists' do
before do
expect(client).to receive(:get_config_map).with("values-content-configuration-#{application_name}", gitlab_namespace).and_return(resource)
end
it 'updates the config map' do
expect(client).to receive(:update_config_map).with(resource).once
subject.install(command)
end
end
end
end
2018-03-17 18:26:18 +05:30
describe '#install' do
before do
allow(client).to receive(:create_pod).and_return(nil)
2019-02-15 15:39:39 +05:30
allow(client).to receive(:get_config_map).and_return(nil)
2018-03-27 19:54:05 +05:30
allow(client).to receive(:create_config_map).and_return(nil)
2018-11-20 20:47:30 +05:30
allow(client).to receive(:create_service_account).and_return(nil)
2019-02-15 15:39:39 +05:30
allow(client).to receive(:delete_pod).and_return(nil)
2018-03-17 18:26:18 +05:30
allow(namespace).to receive(:ensure_exists!).once
end
it 'ensures the namespace exists before creating the POD' do
expect(namespace).to receive(:ensure_exists!).once.ordered
expect(client).to receive(:create_pod).once.ordered
subject.install(command)
end
2018-03-27 19:54:05 +05:30
2019-02-15 15:39:39 +05:30
it 'removes an existing pod before installing' do
expect(client).to receive(:delete_pod).with('install-app-name', 'gitlab-managed-apps').once.ordered
expect(client).to receive(:create_pod).once.ordered
subject.install(command)
end
2018-03-27 19:54:05 +05:30
context 'with a ConfigMap' do
2018-11-20 20:47:30 +05:30
let(:resource) { Gitlab::Kubernetes::ConfigMap.new(application_name, files).generate }
2018-03-27 19:54:05 +05:30
it 'creates a ConfigMap on kubeclient' do
expect(client).to receive(:create_config_map).with(resource).once
subject.install(command)
end
2019-02-15 15:39:39 +05:30
context 'config map already exists' do
before do
expect(client).to receive(:get_config_map).with("values-content-configuration-#{application_name}", gitlab_namespace).and_return(resource)
end
it 'updates the config map' do
expect(client).to receive(:update_config_map).with(resource).once
subject.install(command)
end
end
2018-03-27 19:54:05 +05:30
end
2018-11-20 20:47:30 +05:30
context 'without a service account' do
it 'does not create a service account on kubeclient' do
expect(client).not_to receive(:create_service_account)
2020-05-24 23:13:21 +05:30
expect(client).not_to receive(:update_cluster_role_binding)
2018-11-20 20:47:30 +05:30
subject.install(command)
end
end
context 'with a service account' do
2021-01-29 00:20:46 +05:30
let(:command) { Gitlab::Kubernetes::Helm::V2::InitCommand.new(name: application_name, files: files, rbac: rbac) }
2018-11-20 20:47:30 +05:30
context 'rbac-enabled cluster' do
let(:rbac) { true }
let(:service_account_resource) do
Kubeclient::Resource.new(metadata: { name: 'tiller', namespace: 'gitlab-managed-apps' })
end
let(:cluster_role_binding_resource) do
Kubeclient::Resource.new(
metadata: { name: 'tiller-admin' },
roleRef: { apiGroup: 'rbac.authorization.k8s.io', kind: 'ClusterRole', name: 'cluster-admin' },
subjects: [{ kind: 'ServiceAccount', name: 'tiller', namespace: 'gitlab-managed-apps' }]
)
end
2020-05-24 23:13:21 +05:30
context 'service account does not exist' do
2018-11-20 20:47:30 +05:30
before do
2019-02-15 15:39:39 +05:30
expect(client).to receive(:get_service_account).with('tiller', 'gitlab-managed-apps').and_raise(Kubeclient::ResourceNotFoundError.new(404, 'Not found', nil))
2018-11-20 20:47:30 +05:30
end
it 'creates a service account, followed the cluster role binding on kubeclient' do
expect(client).to receive(:create_service_account).with(service_account_resource).once.ordered
2020-05-24 23:13:21 +05:30
expect(client).to receive(:update_cluster_role_binding).with(cluster_role_binding_resource).once.ordered
2018-11-20 20:47:30 +05:30
subject.install(command)
end
end
context 'service account already exists' do
before do
2019-02-15 15:39:39 +05:30
expect(client).to receive(:get_service_account).with('tiller', 'gitlab-managed-apps').and_return(service_account_resource)
2018-11-20 20:47:30 +05:30
end
it 'updates the service account, followed by creating the cluster role binding' do
expect(client).to receive(:update_service_account).with(service_account_resource).once.ordered
expect(client).to receive(:update_cluster_role_binding).with(cluster_role_binding_resource).once.ordered
subject.install(command)
end
end
context 'a non-404 error is thrown' do
before do
2019-02-15 15:39:39 +05:30
expect(client).to receive(:get_service_account).with('tiller', 'gitlab-managed-apps').and_raise(Kubeclient::HttpError.new(401, 'Unauthorized', nil))
2018-11-20 20:47:30 +05:30
end
it 'raises an error' do
expect { subject.install(command) }.to raise_error(Kubeclient::HttpError)
end
end
end
context 'legacy abac cluster' do
it 'does not create a service account on kubeclient' do
expect(client).not_to receive(:create_service_account)
2020-05-24 23:13:21 +05:30
expect(client).not_to receive(:update_cluster_role_binding)
2018-11-20 20:47:30 +05:30
subject.install(command)
end
end
end
2018-03-17 18:26:18 +05:30
end
2018-11-18 11:00:15 +05:30
describe '#status' do
2018-03-17 18:26:18 +05:30
let(:phase) { Gitlab::Kubernetes::Pod::RUNNING }
let(:pod) { Kubeclient::Resource.new(status: { phase: phase }) } # partial representation
it 'fetches POD phase from kubernetes cluster' do
expect(client).to receive(:get_pod).with(command.pod_name, gitlab_namespace).once.and_return(pod)
2018-11-18 11:00:15 +05:30
expect(subject.status(command.pod_name)).to eq(phase)
2018-03-17 18:26:18 +05:30
end
end
2018-11-18 11:00:15 +05:30
describe '#log' do
2018-03-17 18:26:18 +05:30
let(:log) { 'some output' }
let(:response) { RestClient::Response.new(log) }
it 'fetches POD phase from kubernetes cluster' do
expect(client).to receive(:get_pod_log).with(command.pod_name, gitlab_namespace).once.and_return(response)
2018-11-18 11:00:15 +05:30
expect(subject.log(command.pod_name)).to eq(log)
2018-03-17 18:26:18 +05:30
end
end
2018-11-18 11:00:15 +05:30
describe '#delete_pod!' do
2018-03-17 18:26:18 +05:30
it 'deletes the POD from kubernetes cluster' do
2019-02-15 15:39:39 +05:30
expect(client).to receive(:delete_pod).with('install-app-name', 'gitlab-managed-apps').once
subject.delete_pod!('install-app-name')
end
2018-03-17 18:26:18 +05:30
2019-02-15 15:39:39 +05:30
context 'when the resource being deleted does not exist' do
it 'catches the error' do
expect(client).to receive(:delete_pod).with('install-app-name', 'gitlab-managed-apps')
.and_raise(Kubeclient::ResourceNotFoundError.new(404, 'Not found', nil))
subject.delete_pod!('install-app-name')
end
2018-03-17 18:26:18 +05:30
end
end
2018-12-05 23:21:45 +05:30
describe '#get_config_map' do
before do
allow(namespace).to receive(:ensure_exists!).once
allow(client).to receive(:get_config_map).and_return(nil)
end
it 'ensures the namespace exists before retrieving the config map' do
expect(namespace).to receive(:ensure_exists!).once
subject.get_config_map('example-config-map-name')
end
it 'gets the config map on kubeclient' do
expect(client).to receive(:get_config_map)
.with('example-config-map-name', namespace.name)
.once
subject.get_config_map('example-config-map-name')
end
end
2018-03-17 18:26:18 +05:30
end