Add source to retire request to parse in parse_prov_category in engine

d-m-u · May 6, 2019 · f579458 · f579458
1 parent 2891bc3
commit f579458
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/app/models/vm_retire_request.rb b/app/models/vm_retire_request.rb
@@ -3,6 +3,9 @@ class VmRetireRequest < MiqRetireRequest
   SOURCE_CLASS_NAME = 'Vm'.freeze
   ACTIVE_STATES     = %w(retired) + base_class::ACTIVE_STATES
 
+  default_value_for(:source_id)    { |r| r.get_option(:src_ids) }
+  default_value_for :source_type,  SOURCE_CLASS_NAME
+
   def my_zone
     vm = Vm.find_by(:id => options[:src_ids])
     vm.nil? ? super : vm.my_zone

diff --git a/spec/models/miq_request_spec.rb b/spec/models/miq_request_spec.rb
@@ -558,6 +558,11 @@ def approvals
     end
   end
 
+  let(:vm_retire_request) { FactoryBot.create(:vm_retire_request, :requester => fred) }
+  it "retire_request has source" do
+    expect(vm_retire_request.source_type).not_to eq(nil)
+  end
+
   context "retire request source classes" do
     let(:vm_retire_request)      { FactoryBot.create(:vm_retire_request, :requester => fred) }
     let(:service_retire_request) { FactoryBot.create(:service_retire_request, :requester => fred) }