Port Helm::Api EE extensions to CE

We will need these utility level code in the future to help upgrade all
helm applications.

ee/lib/ee/gitlab/kubernetes/helm/api.rb

-module EE
-  module Gitlab
-    module Kubernetes
-      module Helm
-        module Api
-          def get_config_map(config_map_name)
-            namespace.ensure_exists!
-
-            kubeclient.get_config_map(config_map_name, namespace.name)
-          end
-
-          def update(command)
-            namespace.ensure_exists!
-            update_config_map(command)
-            kubeclient.create_pod(command.pod_resource)
-          end
-
-          private
-
-          def update_config_map(command)
-            command.config_map_resource.tap do |config_map_resource|
-              kubeclient.update_config_map(config_map_resource)
-            end
-          end
-        end
-      end
-    end
-  end
-end

ee/spec/lib/ee/gitlab/kubernetes/helm/api_spec.rb

-require 'spec_helper'
-
-describe Gitlab::Kubernetes::Helm::Api do
-  let(:kubeclient)  { spy }
-  let(:namespace)   { spy }
-  let(:application) { build(:clusters_applications_prometheus) }
-
-  subject { described_class.new(kubeclient) }
-
-  before do
-    allow(Gitlab::Kubernetes::Namespace)
-      .to receive(:new)
-      .with(Gitlab::Kubernetes::Helm::NAMESPACE, kubeclient)
-      .and_return(namespace)
-  end
-
-  describe '#get_config_map' do
-    it 'ensures the namespace exists before retrieving the config map' do
-      expect(namespace).to receive(:ensure_exists!).once
-
-      subject.get_config_map(application.name)
-    end
-
-    it 'gets the config map on kubeclient' do
-      expect(kubeclient).to receive(:get_config_map)
-        .with("example-config-map-name", namespace.name)
-        .once
-
-      subject.get_config_map("example-config-map-name")
-    end
-  end
-
-  describe '#update' do
-    let(:rbac) { false }
-    let(:command) do
-      Gitlab::Kubernetes::Helm::UpgradeCommand.new(
-        application.name,
-        chart: application.chart,
-        files: application.files,
-        rbac: rbac
-      )
-    end
-
-    it 'ensures the namespace exists before creating the pod' do
-      expect(namespace).to receive(:ensure_exists!).once.ordered
-      expect(kubeclient).to receive(:create_pod).once.ordered
-
-      subject.update(command)
-    end
-
-    it 'updates the config map on kubeclient when one exists' do
-      resource = Gitlab::Kubernetes::ConfigMap.new(
-        application.name, application.files
-      ).generate
-
-      expect(kubeclient).to receive(:update_config_map).with(resource).once
-
-      subject.update(command)
-    end
-  end
-end

lib/gitlab/kubernetes/helm/api.rb

@@ -2,8 +2,6 @@ module Gitlab
   module Kubernetes
     module Helm
       class Api
-        prepend EE::Gitlab::Kubernetes::Helm::Api
-
         def initialize(kubeclient)
           @kubeclient = kubeclient
           @namespace = Gitlab::Kubernetes::Namespace.new(Gitlab::Kubernetes::Helm::NAMESPACE, kubeclient)
@@ -19,6 +17,12 @@ module Gitlab
           kubeclient.create_pod(command.pod_resource)
         end
 
+        def update(command)
+          namespace.ensure_exists!
+          update_config_map(command)
+          kubeclient.create_pod(command.pod_resource)
+        end
+
         ##
         # Returns Pod phase
         #
@@ -38,6 +42,12 @@ module Gitlab
           kubeclient.delete_pod(pod_name, namespace.name)
         end
 
+        def get_config_map(config_map_name)
+          namespace.ensure_exists!
+
+          kubeclient.get_config_map(config_map_name, namespace.name)
+        end
+
         private
 
         attr_reader :kubeclient, :namespace
@@ -48,6 +58,12 @@ module Gitlab
           end
         end
 
+        def update_config_map(command)
+          command.config_map_resource.tap do |config_map_resource|
+            kubeclient.update_config_map(config_map_resource)
+          end
+        end
+
         def create_service_account(command)
           command.service_account_resource.tap do |service_account_resource|
             break unless service_account_resource

spec/lib/gitlab/kubernetes/helm/api_spec.rb

@@ -150,6 +150,43 @@ describe Gitlab::Kubernetes::Helm::Api do
     end
   end
 
+  describe '#update' do
+    let(:rbac) { false }
+
+    let(:command) do
+      Gitlab::Kubernetes::Helm::UpgradeCommand.new(
+        application_name,
+        chart: 'chart-name',
+        files: files,
+        rbac: rbac
+      )
+    end
+
+    before do
+      allow(namespace).to receive(:ensure_exists!).once
+
+      allow(client).to receive(:update_config_map).and_return(nil)
+      allow(client).to receive(:create_pod).and_return(nil)
+    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.update(command)
+    end
+
+    it 'updates the config map on kubeclient when one exists' do
+      resource = Gitlab::Kubernetes::ConfigMap.new(
+        application_name, files
+      ).generate
+
+      expect(client).to receive(:update_config_map).with(resource).once
+
+      subject.update(command)
+    end
+  end
+
   describe '#status' do
     let(:phase) { Gitlab::Kubernetes::Pod::RUNNING }
     let(:pod) { Kubeclient::Resource.new(status: { phase: phase }) } # partial representation
@@ -179,4 +216,25 @@ describe Gitlab::Kubernetes::Helm::Api do
       subject.delete_pod!(command.pod_name)
     end
   end
+
+  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
 end