added support to any kind of hosts

MaysaMacedo · Jun 26, 2017 · 1e147e7 · 1e147e7
1 parent 91ccea0
commit 1e147e7
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/app/services/physical_infra_topology_service.rb b/app/services/physical_infra_topology_service.rb
@@ -14,6 +14,14 @@ class PhysicalInfraTopologyService < TopologyService
 
   @kinds = %i(PhysicalInfraManager PhysicalServer Host Vm Tag)
 
+  def entity_type(entity)
+    if entity.kind_of?(Host)
+      entity.class.base_class.name.demodulize
+    else
+      super
+    end
+  end
+
   def entity_display_type(entity)
     if entity.kind_of?(ManageIQ::Providers::PhysicalInfraManager)
       entity.class.short_token