Merge pull request #14641 from lfu/prevent_retirement_request

Add policy checking for retirement request.

db/fixtures/miq_event_definition_sets.csv

@@ -8,6 +8,7 @@ vm_operational,VM Operation
 vm_configurational,VM Configuration
 vm_process,VM Lifecycle
 service_process,Service Lifecycle
+orchestration_process,Orchestration Lifecycle
 storage_operational,Datastore Operation
 auth_validation,Authentication Validation
-container_operations,Container Operation
+container_operations,Container Operation

db/fixtures/miq_event_definitions.csv

@@ -165,6 +165,11 @@ service_started,Service Started,Default,service_process
 request_service_stop,Service Stop Request,Default,service_process
 service_stopped,Service Stopped,Default,service_process
 
+#
+# Orchestration
+#
+request_orchestration_stack_retire,Orchestration Stack Retire Request,Default,orchestration_process
+
 #
 # Container Operations
 #

lib/miq_automation_engine/engine/miq_ae_engine/miq_ae_builtin_method.rb

@@ -153,6 +153,14 @@ def self.miq_event_enforce_policy(obj, _inputs)
       event_object_from_workspace(obj).process_evm_event
     end
 
+    def self.miq_check_policy_prevent(obj, _inputs)
+      event = event_object_from_workspace(obj)
+      if event.full_data && event.full_data[:policy][:prevented]
+        msg = "Event #{event.event_type} for #{event.target} was terminated: #{event.message}"
+        raise MiqAeException::StopInstantiation, msg
+      end
+    end
+
     def self.event_object_from_workspace(obj)
       event = obj.workspace.get_obj_from_path("/")['event_stream']
       raise MiqAeException::MethodParmMissing, "Event not specified" if event.nil?

lib/miq_automation_engine/service_models/miq_ae_service_event_stream.rb

@@ -9,6 +9,7 @@ class MiqAeServiceEventStream < MiqAeServiceModelBase
     expose :dest_vm,               :association => true, :method => :dest_vm_or_template
     expose :dest_host,             :association => true
     expose :service,               :association => true
+    expose :target,                :association => true
 
     def event_namespace
       object_class.name

spec/lib/miq_automation_engine/engine/miq_ae_builtin_method_spec.rb

@@ -0,0 +1,19 @@
+describe MiqAeEngine::MiqAeBuiltinMethod do
+  describe '.miq_check_policy_prevent' do
+    let(:event)     { FactoryGirl.create(:miq_event) }
+    let(:svc_event) { MiqAeMethodService::MiqAeServiceEventStream.find(event.id) }
+    let(:workspace) { double('WORKSPACE', :get_obj_from_path => { 'event_stream' => svc_event }) }
+    let(:obj)       { double('OBJ', :workspace => workspace) }
+
+    subject { described_class.send(:miq_check_policy_prevent, obj, {}) }
+
+    it 'with policy not prevented' do
+      expect { subject }.not_to raise_error
+    end
+
+    it 'with policy prevented' do
+      event.update_attributes(:full_data => {:policy => {:prevented => true}})
+      expect { subject }.to raise_error(MiqAeException::StopInstantiation)
+    end
+  end
+end