Fixes #39532 - All hosts - add action - bulk add capsule

app/controllers/concerns/foreman_openscap/api/v2/hosts_bulk_actions_controller_extensions.rb

@@ -0,0 +1,78 @@
+module ForemanOpenscap
+  module Api
+    module V2
+      module HostsBulkActionsControllerExtensions
+        extend ActiveSupport::Concern
+
+        included do
+          before_action :find_editable_hosts, only: [:change_openscap_proxy]
+        end
+
+        def change_openscap_proxy
+          openscap_proxy_id = params[:openscap_proxy_id]
+
+          if openscap_proxy_id.blank?
+            return render json: {
+              error: {
+                message: _("No OpenSCAP Proxy selected."),
+              },
+            }, status: :unprocessable_entity
+          end
+
+          smart_proxy = begin
+            ::SmartProxy.find(openscap_proxy_id)
+          rescue ActiveRecord::RecordNotFound
+            nil
+          end
+
+          if smart_proxy.nil?
+            return render json: {
+              error: {
+                message: _("OpenSCAP proxy with id %s not found") % openscap_proxy_id,
+              },
+            }, status: :unprocessable_entity
+          end
+
+          unless smart_proxy.has_feature?('Openscap')
+            return render json: {
+              error: {
+                message: _("The selected proxy does not have the OpenSCAP feature enabled."),
+              },
+            }, status: :unprocessable_entity
+          end
+
+          failed_hosts = []
+          @hosts.each do |host|
+            host.openscap_proxy = smart_proxy
+            failed_hosts << host unless host.save
+          end
+
+          if failed_hosts.empty?
+            process_response(true, {
+              message: _("OpenSCAP Proxy set to %s") % smart_proxy.name,
+            })
+          else
+            render_error(:bulk_hosts_error, status: :unprocessable_entity,
+                         locals: {
+                           message: n_("Failed to assign OpenSCAP Proxy to %s host",
+                                       "Failed to assign OpenSCAP Proxy to %s hosts",
+                                       failed_hosts.count) % failed_hosts.count,
+                           failed_host_ids: failed_hosts.map(&:id),
+                         })
+          end
+        end
+
+        private
+
+        def action_permission
+          case params[:action]
+          when 'change_openscap_proxy'
+            'edit'
+          else
+            super
+          end
+        end
+      end
+    end
+  end
+end

config/routes.rb

@@ -103,6 +103,8 @@
           end
         end
       end
+
+      match 'hosts/bulk/change_openscap_proxy', :to => 'hosts_bulk_actions#change_openscap_proxy', :via => [:put]
     end
   end
 end

lib/foreman_openscap/engine.rb

@@ -93,7 +93,8 @@ class Engine < ::Rails::Engine
                       :resource_type => 'ForemanOpenscap::ScapContent'
             permission :edit_hosts, { :hosts => %i[openscap_proxy_changed
                                                   select_multiple_openscap_proxy
-                                                  update_multiple_openscap_proxy] },
+                                                  update_multiple_openscap_proxy],
+                                      'api/v2/hosts_bulk_actions' => [:change_openscap_proxy] },
                       :resource_type => "Host"
             permission :view_hosts, { 'api/v2/hosts' => [:policies_enc] }, :resource_type => 'Host'
             permission :edit_hostgroups, { :hostgroups => [:openscap_proxy_changed] }, :resource_type => "Hostgroup"
@@ -195,6 +196,7 @@ class Engine < ::Rails::Engine
     # Include concerns in this config.to_prepare block
     config.to_prepare do
       ::Api::V2::HostsController.send(:include, ForemanOpenscap::Api::V2::HostsControllerExtensions)
+      ::Api::V2::HostsBulkActionsController.send(:include, ForemanOpenscap::Api::V2::HostsBulkActionsControllerExtensions)
       ::Host::Managed.send(:include, ForemanOpenscap::OpenscapProxyExtensions)
       ::Host::Managed.send(:include, ForemanOpenscap::OpenscapProxyCoreExtensions)
       ::Host::Managed.send(:prepend, ForemanOpenscap::HostExtensions)

test/functional/api/v2/hosts_bulk_actions_controller_test.rb

@@ -0,0 +1,98 @@
+require 'test_plugin_helper'
+
+class Api::V2::HostsBulkActionsControllerTest < ActionController::TestCase
+  def setup
+    as_admin do
+      @organization = FactoryBot.create(:organization)
+      @location = FactoryBot.create(:location)
+      @proxy = FactoryBot.create(:openscap_proxy,
+                                 :organizations => [@organization],
+                                 :locations => [@location])
+      @host1 = FactoryBot.create(:host, :managed,
+                                 :organization => @organization,
+                                 :location => @location)
+      @host2 = FactoryBot.create(:host, :managed,
+                                 :organization => @organization,
+                                 :location => @location)
+      @host_ids = [@host1.id, @host2.id]
+    end
+  end
+
+  def valid_bulk_params(host_ids = @host_ids)
+    {
+      :organization_id => @organization.id,
+      :location_id => @location.id,
+      :included => {
+        :ids => host_ids,
+      },
+      :excluded => {
+        :ids => [],
+      },
+    }
+  end
+
+  test "should assign openscap proxy to selected hosts" do
+    put :change_openscap_proxy,
+        params: valid_bulk_params.merge(:openscap_proxy_id => @proxy.id),
+        session: set_session_user
+
+    assert_response :success
+    response = ActiveSupport::JSON.decode(@response.body)
+    assert_match(/OpenSCAP Proxy set to/, response['message'])
+    assert_includes response['message'], @proxy.name
+
+    [@host1, @host2].each do |host|
+      host.reload
+      assert_equal @proxy.id, host.openscap_proxy_id
+    end
+  end
+
+  test "should require openscap_proxy_id" do
+    put :change_openscap_proxy,
+        params: valid_bulk_params,
+        session: set_session_user
+
+    assert_response :unprocessable_entity
+    response = ActiveSupport::JSON.decode(@response.body)
+    assert_match(/No OpenSCAP Proxy selected/, response['error']['message'])
+  end
+
+  test "should return error when proxy is not found" do
+    put :change_openscap_proxy,
+        params: valid_bulk_params.merge(:openscap_proxy_id => 0),
+        session: set_session_user
+
+    assert_response :unprocessable_entity
+    response = ActiveSupport::JSON.decode(@response.body)
+    assert_match(/not found/, response['error']['message'])
+  end
+
+  test "should return error when proxy lacks Openscap feature" do
+    other_proxy = FactoryBot.create(:smart_proxy,
+                                    :organizations => [@organization],
+                                    :locations => [@location])
+    openscap_feature = Feature.find_by(:name => 'Openscap')
+    other_proxy.features.delete(openscap_feature) if openscap_feature
+    refute other_proxy.reload.has_feature?('Openscap')
+
+    put :change_openscap_proxy,
+        params: valid_bulk_params.merge(:openscap_proxy_id => other_proxy.id),
+        session: set_session_user
+
+    assert_response :unprocessable_entity
+    response = ActiveSupport::JSON.decode(@response.body)
+    assert_match(/OpenSCAP feature/, response['error']['message'])
+  end
+
+  test "should assign openscap proxy for a single host" do
+    put :change_openscap_proxy,
+        params: valid_bulk_params([@host1.id]).merge(:openscap_proxy_id => @proxy.id),
+        session: set_session_user
+
+    assert_response :success
+    @host1.reload
+    assert_equal @proxy.id, @host1.openscap_proxy_id
+    @host2.reload
+    assert_nil @host2.openscap_proxy_id
+  end
+end