Merge pull request #18715 from lfu/src_vm_or_dest_host_1696889

Add src_vm_or_dest_host_refresh_target.

(cherry picked from commit 8a87d0f)

https://bugzilla.redhat.com/show_bug.cgi?id=1712595

app/models/ems_event.rb

@@ -264,6 +264,10 @@ def vm_refresh_target
   end
   alias_method :src_vm_refresh_target, :vm_refresh_target
 
+  def src_vm_or_dest_host_refresh_target
+    vm_or_template ? vm_refresh_target : dest_host_refresh_target
+  end
+
   def host_refresh_target
     (host && host.ext_management_system ? host : ems_refresh_target)
   end

spec/models/ems_event_spec.rb

@@ -393,4 +393,23 @@
       end
     end
   end
+
+  context 'refresh target' do
+    describe 'src_vm_or_dest_host_refresh_target' do
+      let(:ems)   { FactoryBot.create(:ems_vmware) }
+      let(:vm)    { FactoryBot.create(:vm_vmware, :ext_management_system => ems) }
+      let(:host1) { FactoryBot.create(:host, :ext_management_system => ems) }
+      let(:host2) { FactoryBot.create(:host, :ext_management_system => ems) }
+
+      it 'returns src_vm when it exists' do
+        event = FactoryBot.create(:ems_event, :vm_or_template => vm, :host => host1, :dest_host => host2)
+        expect(event.get_target("src_vm_or_dest_host_refresh_target")).to eq(vm)
+      end
+
+      it 'returns dest_host when src_vm does not exists' do
+        event = FactoryBot.create(:ems_event, :vm_or_template_id => 123, :host => host1, :dest_host => host2)
+        expect(event.get_target("src_vm_or_dest_host_refresh_target")).to eq(host2)
+      end
+    end
+  end
 end