diff options
Diffstat (limited to 'spec/services/pod_logs/base_service_spec.rb')
-rw-r--r-- | spec/services/pod_logs/base_service_spec.rb | 229 |
1 files changed, 229 insertions, 0 deletions
diff --git a/spec/services/pod_logs/base_service_spec.rb b/spec/services/pod_logs/base_service_spec.rb new file mode 100644 index 00000000000..a18fda544df --- /dev/null +++ b/spec/services/pod_logs/base_service_spec.rb @@ -0,0 +1,229 @@ +# frozen_string_literal: true + +require 'spec_helper' + +describe ::PodLogs::BaseService do + include KubernetesHelpers + + let_it_be(:cluster) { create(:cluster, :provided_by_gcp, environment_scope: '*') } + let(:namespace) { 'autodevops-deploy-9-production' } + + let(:pod_name) { 'pod-1' } + let(:container_name) { 'container-0' } + let(:params) { {} } + let(:raw_pods) do + JSON.parse([ + kube_pod(name: pod_name) + ].to_json, object_class: OpenStruct) + end + + subject { described_class.new(cluster, namespace, params: params) } + + describe '#initialize' do + let(:params) do + { + 'container_name' => container_name, + 'another_param' => 'foo' + } + end + + it 'filters the parameters' do + expect(subject.cluster).to eq(cluster) + expect(subject.namespace).to eq(namespace) + expect(subject.params).to eq({ + 'container_name' => container_name + }) + expect(subject.params.equal?(params)).to be(false) + end + end + + describe '#check_arguments' do + context 'when cluster and namespace are provided' do + it 'returns success' do + result = subject.send(:check_arguments, {}) + + expect(result[:status]).to eq(:success) + end + end + + context 'when cluster is nil' do + let(:cluster) { nil } + + it 'returns an error' do + result = subject.send(:check_arguments, {}) + + expect(result[:status]).to eq(:error) + expect(result[:message]).to eq('Cluster does not exist') + end + end + + context 'when namespace is nil' do + let(:namespace) { nil } + + it 'returns an error' do + result = subject.send(:check_arguments, {}) + + expect(result[:status]).to eq(:error) + expect(result[:message]).to eq('Namespace is empty') + end + end + + context 'when namespace is empty' do + let(:namespace) { '' } + + it 'returns an error' do + result = subject.send(:check_arguments, {}) + + expect(result[:status]).to eq(:error) + expect(result[:message]).to eq('Namespace is empty') + end + end + end + + describe '#check_param_lengths' do + context 'when pod_name and container_name are provided' do + let(:params) do + { + 'pod_name' => pod_name, + 'container_name' => container_name + } + end + + it 'returns success' do + result = subject.send(:check_param_lengths, {}) + + expect(result[:status]).to eq(:success) + expect(result[:pod_name]).to eq(pod_name) + expect(result[:container_name]).to eq(container_name) + end + end + + context 'when pod_name is too long' do + let(:params) do + { + 'pod_name' => "a very long string." * 15 + } + end + + it 'returns an error' do + result = subject.send(:check_param_lengths, {}) + + expect(result[:status]).to eq(:error) + expect(result[:message]).to eq('pod_name cannot be larger than 253 chars') + end + end + + context 'when container_name is too long' do + let(:params) do + { + 'container_name' => "a very long string." * 15 + } + end + + it 'returns an error' do + result = subject.send(:check_param_lengths, {}) + + expect(result[:status]).to eq(:error) + expect(result[:message]).to eq('container_name cannot be larger than 253 chars') + end + end + end + + describe '#get_raw_pods' do + let(:service) { create(:cluster_platform_kubernetes, :configured) } + + it 'returns success with passthrough k8s response' do + stub_kubeclient_pods(namespace) + + result = subject.send(:get_raw_pods, {}) + + expect(result[:status]).to eq(:success) + expect(result[:raw_pods].first).to be_a(Kubeclient::Resource) + end + end + + describe '#get_pod_names' do + it 'returns success with a list of pods' do + result = subject.send(:get_pod_names, raw_pods: raw_pods) + + expect(result[:status]).to eq(:success) + expect(result[:pods]).to eq([pod_name]) + end + end + + describe '#check_pod_name' do + it 'returns success if pod_name was specified' do + result = subject.send(:check_pod_name, pod_name: pod_name, pods: [pod_name]) + + expect(result[:status]).to eq(:success) + expect(result[:pod_name]).to eq(pod_name) + end + + it 'returns success if pod_name was not specified but there are pods' do + result = subject.send(:check_pod_name, pod_name: nil, pods: [pod_name]) + + expect(result[:status]).to eq(:success) + expect(result[:pod_name]).to eq(pod_name) + end + + it 'returns error if pod_name was not specified and there are no pods' do + result = subject.send(:check_pod_name, pod_name: nil, pods: []) + + expect(result[:status]).to eq(:error) + expect(result[:message]).to eq('No pods available') + end + + it 'returns error if pod_name was specified but does not exist' do + result = subject.send(:check_pod_name, pod_name: 'another_pod', pods: [pod_name]) + + expect(result[:status]).to eq(:error) + expect(result[:message]).to eq('Pod does not exist') + end + end + + describe '#check_container_name' do + it 'returns success if container_name was specified' do + result = subject.send(:check_container_name, + container_name: container_name, + pod_name: pod_name, + raw_pods: raw_pods + ) + + expect(result[:status]).to eq(:success) + expect(result[:container_name]).to eq(container_name) + end + + it 'returns success if container_name was not specified and there are containers' do + result = subject.send(:check_container_name, + pod_name: pod_name, + raw_pods: raw_pods + ) + + expect(result[:status]).to eq(:success) + expect(result[:container_name]).to eq(container_name) + end + + it 'returns error if container_name was not specified and there are no containers on the pod' do + raw_pods.first.spec.containers = [] + + result = subject.send(:check_container_name, + pod_name: pod_name, + raw_pods: raw_pods + ) + + expect(result[:status]).to eq(:error) + expect(result[:message]).to eq('No containers available') + end + + it 'returns error if container_name was specified but does not exist' do + result = subject.send(:check_container_name, + container_name: 'foo', + pod_name: pod_name, + raw_pods: raw_pods + ) + + expect(result[:status]).to eq(:error) + expect(result[:message]).to eq('Container does not exist') + end + end +end |