Get proper record class for VMs of a Resource Pool

Author: Hilda Stastna

app/controllers/resource_pool_controller.rb

@@ -13,6 +13,14 @@ def self.display_methods
     %w[vms all_vms resource_pools]
   end
 
+  def record_class
+    if %w[all_vms vms].include?(params[:display])
+      VmOrTemplate
+    else
+      ResourcePool
+    end
+  end
+
   # handle buttons pressed on the button bar
   def button
     @edit = session[:edit] # Restore @edit for adv search box