Update services NetworkPolicy related

To include CiliumNetworkPolicy in their
logic

ee/app/controllers/projects/security/network_policies_controller.rb

@@ -39,9 +39,8 @@ module Projects
       end
 
       def create
-        policy = Gitlab::Kubernetes::NetworkPolicy.from_yaml(params[:manifest])
         response = NetworkPolicies::DeployResourceService.new(
-          policy: policy,
+          manifest: params[:manifest],
           environment: environment
         ).execute
 
@@ -49,15 +48,11 @@ module Projects
       end
 
       def update
-        policy = Gitlab::Kubernetes::NetworkPolicy.from_yaml(params[:manifest])
-        unless params[:enabled].nil?
-          params[:enabled] ? policy.enable : policy.disable
-        end
-
         response = NetworkPolicies::DeployResourceService.new(
           resource_name: params[:id],
-          policy: policy,
-          environment: environment
+          manifest: params[:manifest],
+          environment: environment,
+          enabled: params[:enabled]
         ).execute
 
         respond_with_service_response(response)
@@ -66,6 +61,7 @@ module Projects
       def destroy
         response = NetworkPolicies::DeleteResourceService.new(
           resource_name: params[:id],
+          manifest: params[:manifest],
           environment: environment
         ).execute
 

ee/app/services/network_policies/delete_resource_service.rb

@@ -3,17 +3,24 @@
 module NetworkPolicies
   class DeleteResourceService
     include NetworkPolicies::Responses
+    include NetworkPolicies::Types
 
-    def initialize(resource_name:, environment:)
+    def initialize(resource_name:, manifest:, environment:)
       @resource_name = resource_name
       @platform = environment.deployment_platform
       @kubernetes_namespace = environment.deployment_namespace
+      @has_cilium_policy = cilium_policy?(manifest)
     end
 
     def execute
       return no_platform_response unless @platform
 
-      @platform.kubeclient.delete_network_policy(@resource_name, @kubernetes_namespace)
+      if @has_cilium_policy
+        @platform.kubeclient.delete_cilium_network_policy(@resource_name, @kubernetes_namespace)
+      else
+        @platform.kubeclient.delete_network_policy(@resource_name, @kubernetes_namespace)
+      end
+
       ServiceResponse.success
     rescue Kubeclient::HttpError => e
       kubernetes_error_response(e)

ee/app/services/network_policies/deploy_resource_service.rb

@@ -3,9 +3,15 @@
 module NetworkPolicies
   class DeployResourceService
     include NetworkPolicies::Responses
+    include NetworkPolicies::Types
+
+    def initialize(manifest:, environment:, resource_name: nil, enabled: nil)
+      @has_cilium_policy = cilium_policy?(manifest)
+      @policy = policy_from_manifest(manifest)
+      unless enabled.nil?
+        enabled ? policy.enable : policy.disable
+      end
 
-    def initialize(policy:, environment:, resource_name: nil)
-      @policy = policy
       @platform = environment.deployment_platform
       @kubernetes_namespace = environment.deployment_namespace
       @resource_name = resource_name
@@ -16,8 +22,8 @@ module NetworkPolicies
       return no_platform_response unless platform
 
       setup_resource
-      resource = deploy_resource
-      policy = Gitlab::Kubernetes::NetworkPolicy.from_resource(resource)
+      deploy_resource
+      load_policy_from_resource
       ServiceResponse.success(payload: policy)
     rescue Kubeclient::HttpError => e
       kubernetes_error_response(e)
@@ -25,7 +31,7 @@ module NetworkPolicies
 
     private
 
-    attr_reader :platform, :policy, :resource_name, :resource, :kubernetes_namespace
+    attr_reader :platform, :policy, :resource_name, :resource, :kubernetes_namespace, :has_cilium_policy
 
     def setup_resource
       @resource = policy.generate
@@ -33,7 +39,31 @@ module NetworkPolicies
       resource[:metadata][:name] = resource_name if resource_name
     end
 
+    def load_policy_from_resource
+      @policy = if has_cilium_policy
+                  Gitlab::Kubernetes::CiliumNetworkPolicy.from_resource(resource)
+                else
+                  Gitlab::Kubernetes::NetworkPolicy.from_resource(resource)
+                end
+    end
+
     def deploy_resource
+      @resource = if has_cilium_policy
+                    deploy_cilium_network_policy
+                  else
+                    deploy_network_policy
+                  end
+    end
+
+    def deploy_cilium_network_policy
+      if resource_name
+        platform.kubeclient.update_cilium_network_policy(resource)
+      else
+        platform.kubeclient.create_cilium_network_policy(resource)
+      end
+    end
+
+    def deploy_network_policy
       if resource_name
         platform.kubeclient.update_network_policy(resource)
       else

ee/app/services/network_policies/resources_service.rb

@@ -15,6 +15,9 @@ module NetworkPolicies
       policies = @platform.kubeclient
         .get_network_policies(namespace: @kubernetes_namespace)
         .map { |resource| Gitlab::Kubernetes::NetworkPolicy.from_resource(resource) }
+      policies += @platform.kubeclient
+        .get_cilium_network_policies(namespace: @kubernetes_namespace)
+        .map { |resource| Gitlab::Kubernetes::CiliumNetworkPolicy.from_resource(resource) }
       ServiceResponse.success(payload: policies)
     rescue Kubeclient::HttpError => e
       kubernetes_error_response(e)

ee/app/services/network_policies/types.rb

+# frozen_string_literal: true
+
+module NetworkPolicies
+  module Types
+    def cilium_policy?(manifest)
+      manifest&.include?(Gitlab::Kubernetes::CiliumNetworkPolicy::KIND)
+    end
+
+    def policy_from_manifest(manifest)
+      cilium_policy?(manifest) ? Gitlab::Kubernetes::CiliumNetworkPolicy.from_yaml(manifest) : Gitlab::Kubernetes::NetworkPolicy.from_yaml(manifest)
+    end
+  end
+end

ee/changelogs/unreleased/add_cilium_network_policy_into_services.yml

+---
+title: add CiliumNetworkPolicy into services
+merge_request: 39127
+author:
+type: changed

ee/spec/controllers/projects/security/network_policies_controller_spec.rb

@@ -11,6 +11,27 @@ RSpec.describe Projects::Security::NetworkPoliciesController do
 
   let_it_be(:action_params) { { project_id: project, namespace_id: project.namespace, environment_id: environment.id } }
 
+  let_it_be(:manifest) do
+    <<~POLICY
+      apiVersion: networking.k8s.io/v1
+      kind: NetworkPolicy
+      metadata:
+        name: example-name
+        namespace: example-namespace
+      spec:
+        podSelector:
+          matchLabels:
+            role: db
+        policyTypes:
+        - Ingress
+        ingress:
+        - from:
+          - namespaceSelector:
+              matchLabels:
+                project: myproject
+    POLICY
+  end
+
   shared_examples 'CRUD service errors' do
     context 'with an error service response' do
       before do
@@ -189,33 +210,12 @@ RSpec.describe Projects::Security::NetworkPoliciesController do
       )
     end
 
-    let(:manifest) do
-      <<~POLICY
-        apiVersion: networking.k8s.io/v1
-        kind: NetworkPolicy
-        metadata:
-          name: example-name
-          namespace: example-namespace
-        spec:
-          podSelector:
-            matchLabels:
-              role: db
-          policyTypes:
-          - Ingress
-          ingress:
-          - from:
-            - namespaceSelector:
-                matchLabels:
-                  project: myproject
-      POLICY
-    end
-
     context 'with authorized user' do
       before do
         group.add_developer(user)
         allow(NetworkPolicies::DeployResourceService).to(
           receive(:new)
-            .with(policy: kind_of(Gitlab::Kubernetes::NetworkPolicy), environment: environment)
+            .with(manifest: manifest, environment: environment)
             .and_return(service)
         )
       end
@@ -253,33 +253,12 @@ RSpec.describe Projects::Security::NetworkPoliciesController do
       )
     end
 
-    let(:manifest) do
-      <<~POLICY
-        apiVersion: networking.k8s.io/v1
-        kind: NetworkPolicy
-        metadata:
-          name: example-name
-          namespace: example-namespace
-        spec:
-          podSelector:
-            matchLabels:
-              role: db
-          policyTypes:
-          - Ingress
-          ingress:
-          - from:
-            - namespaceSelector:
-                matchLabels:
-                  project: myproject
-      POLICY
-    end
-
     context 'with authorized user' do
       before do
         group.add_developer(user)
         allow(NetworkPolicies::DeployResourceService).to(
           receive(:new)
-            .with(policy: kind_of(Gitlab::Kubernetes::NetworkPolicy), environment: environment, resource_name: 'example-policy')
+            .with(manifest: manifest, environment: environment, enabled: enabled, resource_name: 'example-policy')
             .and_return(service)
         )
       end
@@ -292,34 +271,6 @@ RSpec.describe Projects::Security::NetworkPoliciesController do
       end
 
       include_examples 'CRUD service errors'
-
-      context 'with enabled param' do
-        let(:enabled) { true }
-
-        before do
-          allow(Gitlab::Kubernetes::NetworkPolicy).to receive(:new) { policy }
-        end
-
-        it 'enables policy and responds with success' do
-          expect(policy).to receive(:enable)
-
-          subject
-
-          expect(response).to have_gitlab_http_status(:success)
-        end
-
-        context 'with enabled=false' do
-          let(:enabled) { false }
-
-          it 'disables policy and responds with success' do
-            expect(policy).to receive(:disable)
-
-            subject
-
-            expect(response).to have_gitlab_http_status(:success)
-          end
-        end
-      end
     end
 
     context 'with unauthorized user' do
@@ -332,7 +283,7 @@ RSpec.describe Projects::Security::NetworkPoliciesController do
   end
 
   describe 'DELETE #destroy' do
-    subject { delete :destroy, params: action_params.merge(id: 'example-policy'), format: :json }
+    subject { delete :destroy, params: action_params.merge(id: 'example-policy', manifest: manifest), format: :json }
 
     let(:service) { instance_double('NetworkPolicies::DeleteResourceService', execute: ServiceResponse.success) }
 
@@ -341,7 +292,7 @@ RSpec.describe Projects::Security::NetworkPoliciesController do
         group.add_developer(user)
         allow(NetworkPolicies::DeleteResourceService).to(
           receive(:new)
-            .with(environment: environment, resource_name: 'example-policy')
+            .with(environment: environment, manifest: manifest, resource_name: 'example-policy')
             .and_return(service)
         )
       end

ee/spec/services/network_policies/delete_resource_service_spec.rb

@@ -3,10 +3,32 @@
 require 'spec_helper'
 
 RSpec.describe NetworkPolicies::DeleteResourceService do
-  let(:service) { NetworkPolicies::DeleteResourceService.new(resource_name: 'policy', environment: environment) }
+  let(:service) { NetworkPolicies::DeleteResourceService.new(resource_name: 'policy', environment: environment, manifest: manifest) }
   let(:environment) { instance_double('Environment', deployment_platform: platform, deployment_namespace: 'namespace') }
   let(:platform) { instance_double('Clusters::Platforms::Kubernetes', kubeclient: kubeclient) }
   let(:kubeclient) { double('Kubeclient::Client') }
+  let(:manifest) do
+    <<~POLICY
+      apiVersion: networking.k8s.io/v1
+      kind: NetworkPolicy
+      metadata:
+        name: example-name
+        namespace: example-namespace
+      spec:
+        podSelector:
+          matchLabels:
+            role: db
+        policyTypes:
+        - Ingress
+        ingress:
+        - from:
+          - namespaceSelector:
+              matchLabels:
+                project: myproject
+    POLICY
+  end
+
+  include_examples 'different network policy types'
 
   describe '#execute' do
     subject { service.execute }
@@ -37,5 +59,31 @@ RSpec.describe NetworkPolicies::DeleteResourceService do
         expect(subject.message).not_to be_nil
       end
     end
+
+    context 'with CiliumNetworkPolicy' do
+      let(:manifest) do
+        <<~POLICY
+          apiVersion: cilium.io/v2
+          kind: CiliumNetworkPolicy
+          metadata:
+            name: example-name
+            namespace: example-namespace
+            resourceVersion: 101
+          spec:
+            endpointSelector:
+              matchLabels:
+                role: db
+            ingress:
+            - fromEndpoints:
+              - matchLabels:
+                  project: myproject
+        POLICY
+      end
+
+      it 'deletes resource from the deployment namespace and returns success response' do
+        expect(kubeclient).to receive(:delete_cilium_network_policy).with('policy', environment.deployment_namespace)
+        expect(subject).to be_success
+      end
+    end
   end
 end

ee/spec/services/network_policies/deploy_resource_service_spec.rb

@@ -3,10 +3,12 @@
 require 'spec_helper'
 
 RSpec.describe NetworkPolicies::DeployResourceService do
-  let(:service) { NetworkPolicies::DeployResourceService.new(policy: policy, environment: environment) }
+  let(:service) { NetworkPolicies::DeployResourceService.new(resource_name: resource_name, manifest: manifest, environment: environment, enabled: enabled) }
   let(:environment) { instance_double('Environment', deployment_platform: platform, deployment_namespace: 'namespace') }
   let(:platform) { instance_double('Clusters::Platforms::Kubernetes', kubeclient: kubeclient) }
   let(:kubeclient) { double('Kubeclient::Client') }
+  let(:enabled) { nil }
+  let(:resource_name) { nil }
   let(:policy) do
     Gitlab::Kubernetes::NetworkPolicy.new(
       name: 'policy',
@@ -16,6 +18,34 @@ RSpec.describe NetworkPolicies::DeployResourceService do
     )
   end
 
+  let(:manifest) do
+    <<~POLICY
+      apiVersion: networking.k8s.io/v1
+      kind: NetworkPolicy
+      metadata:
+        name: policy
+        namespace: another
+      spec:
+        podSelector:
+          matchLabels:
+            role: db
+        policyTypes:
+        - Ingress
+        ingress:
+        - from:
+          - namespaceSelector:
+              matchLabels:
+                project: myproject
+    POLICY
+  end
+
+  include_examples 'different network policy types'
+
+  before do
+    allow(Gitlab::Kubernetes::NetworkPolicy).to receive(:from_resource).and_return policy
+    allow(Gitlab::Kubernetes::NetworkPolicy).to receive(:from_yaml).and_return policy
+  end
+
   describe '#execute' do
     subject { service.execute }
 
@@ -29,7 +59,7 @@ RSpec.describe NetworkPolicies::DeployResourceService do
     end
 
     context 'with resource_name' do
-      let(:service) { NetworkPolicies::DeployResourceService.new(policy: policy, environment: environment, resource_name: 'policy2') }
+      let(:resource_name) { 'policy2' }
 
       it 'updates resource in the deployment namespace and returns success response with a policy' do
         namespaced_policy = policy.generate
@@ -42,7 +72,8 @@ RSpec.describe NetworkPolicies::DeployResourceService do
       end
     end
 
-    context 'without policy' do
+    context 'without manifest' do
+      let(:manifest) { nil }
       let(:policy) { nil }
 
       it 'returns error response' do
@@ -73,5 +104,79 @@ RSpec.describe NetworkPolicies::DeployResourceService do
         expect(subject.message).not_to be_nil
       end
     end
+
+    context 'with cilium network policy' do
+      let(:manifest) do
+        <<~POLICY
+          apiVersion: cilium.io/v2
+          kind: CiliumNetworkPolicy
+          metadata:
+            name: policy
+            namespace: another
+            resourceVersion: 101
+          spec:
+            endpointSelector:
+              matchLabels:
+                role: db
+            ingress:
+            - fromEndpoints:
+              - matchLabels:
+                  project: myproject
+        POLICY
+      end
+
+      let(:policy) do
+        Gitlab::Kubernetes::CiliumNetworkPolicy.new(
+          name: 'policy',
+          namespace: 'namespace',
+          resource_version: 101,
+          selector: { matchLabels: { role: 'db' } },
+          ingress: [{ fromEndpoints: [{ matchLabels: { project: 'myproject' } }] }]
+        )
+      end
+
+      it 'creates resource in the deployment namespace and return success response with a policy' do
+        namespaced_policy = policy.generate
+        namespaced_policy[:metadata][:namespace] = environment.deployment_namespace
+
+        expect(kubeclient).to receive(:create_cilium_network_policy).with(namespaced_policy) { policy.generate }
+        expect(subject).to be_success
+        expect(subject.payload.as_json).to eq(policy.as_json)
+      end
+
+      context 'with resource_name' do
+        let(:resource_name) { 'policy' }
+
+        it 'updates resource in the deployment namespace and returns success response with a policy' do
+          namespaced_policy = policy.generate
+          namespaced_policy[:metadata][:namespace] = environment.deployment_namespace
+          namespaced_policy[:metadata][:name] = resource_name
+
+          expect(kubeclient).to receive(:update_cilium_network_policy).with(namespaced_policy) { policy.generate }
+          expect(subject).to be_success
+          expect(subject.payload.as_json).to eq(policy.as_json)
+        end
+      end
+    end
+
+    context 'with enabled set to true' do
+      let(:enabled) { true }
+
+      it 'enables policy before deploying it' do
+        expect(policy).to receive(:enable)
+        expect(kubeclient).to receive(:create_network_policy) { policy.generate }
+        expect(subject).to be_success
+      end
+    end
+
+    context 'with enabled set to false' do
+      let(:enabled) { false }
+
+      it 'disables policy before deploying it' do
+        expect(policy).to receive(:disable)
+        expect(kubeclient).to receive(:create_network_policy) { policy.generate }
+        expect(subject).to be_success
+      end
+    end
   end
 end

ee/spec/services/network_policies/resources_service_spec.rb

@@ -16,14 +16,26 @@ RSpec.describe NetworkPolicies::ResourcesService do
     )
   end
 
+  let(:cilium_policy) do
+    Gitlab::Kubernetes::CiliumNetworkPolicy.new(
+      name: 'cilium_policy',
+      namespace: 'another',
+      resource_version: '102',
+      selector: { matchLabels: { role: 'db' } },
+      ingress: [{ endpointFrom: [{ matchLabels: { project: 'myproject' } }] }]
+    )
+  end
+
   describe '#execute' do
     subject { service.execute }
 
     it 'returns success response with policies from the deployment namespace' do
       expect(kubeclient).to receive(:get_network_policies).with(namespace: environment.deployment_namespace) { [policy.generate] }
+      expect(kubeclient).to receive(:get_cilium_network_policies).with(namespace: environment.deployment_namespace) { [cilium_policy.generate] }
       expect(subject).to be_success
-      expect(subject.payload.count).to eq(1)
+      expect(subject.payload.count).to eq(2)
       expect(subject.payload.first.as_json).to eq(policy.as_json)
+      expect(subject.payload.last.as_json).to eq(cilium_policy.as_json)
     end
 
     context 'without deployment_platform' do

ee/spec/support/shared_examples/services/network_policies/types_shared_examples.rb

+# frozen_string_literal: true
+
+RSpec.shared_examples 'different network policy types' do
+  let(:network_policy_manifest) do
+    <<~POLICY
+      apiVersion: networking.k8s.io/v1
+      kind: NetworkPolicy
+      metadata:
+        name: example-name
+        namespace: example-namespace
+      spec:
+        podSelector:
+          matchLabels:
+            role: db
+        policyTypes:
+        - Ingress
+        ingress:
+        - from:
+          - namespaceSelector:
+              matchLabels:
+                project: myproject
+    POLICY
+  end
+
+  let(:cilium_network_policy_manifest) do
+    <<~POLICY
+      apiVersion: cilium.io/v2
+      kind: CiliumNetworkPolicy
+      metadata:
+        name: example-name
+        namespace: example-namespace
+        resourceVersion: 101
+      spec:
+        endpointSelector:
+          matchLabels:
+            role: db
+        ingress:
+        - fromEndpoints:
+          - matchLabels:
+              project: myproject
+    POLICY
+  end
+
+  describe 'cilium_policy?' do
+    subject { service.cilium_policy?(manifest) }
+
+    context 'with nil as parameter' do
+      let(:manifest) { nil }
+
+      it { is_expected.to be_nil }
+    end
+
+    context 'with a manifest of NetworkPolicy' do
+      let(:manifest) { network_policy_manifest }
+
+      it { is_expected.to be false }
+    end
+
+    context 'with a manifest of CiliumNetworkPolicy' do
+      let(:manifest) { cilium_network_policy_manifest }
+
+      it { is_expected.to be true }
+    end
+  end
+
+  describe 'policy_from_manifest' do
+    subject { service.policy_from_manifest(manifest) }
+
+    context 'with a manifest of a NetworkPolicy' do
+      let(:manifest) { network_policy_manifest }
+
+      it { is_expected.to be_an_instance_of Gitlab::Kubernetes::NetworkPolicy }
+    end
+
+    context 'with a manifest of a CiliumNetworkPolicy' do
+      let(:manifest) { cilium_network_policy_manifest }
+
+      it { is_expected.to be_an_instance_of Gitlab::Kubernetes::CiliumNetworkPolicy }
+    end
+  end
+end