From d83e3cf181dc277c0e5381fec76e6439164a5446 Mon Sep 17 00:00:00 2001
From: Zita Nemeckova <znemecko@redhat.com>
Date: Thu, 17 Aug 2017 15:02:19 +0200
Subject: [PATCH] Add new classes to BUTTON_CLASSES

Add new classes to have custom buttons
---
 app/models/custom_button.rb | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/app/models/custom_button.rb b/app/models/custom_button.rb
index c71c9741795..ca38240139b 100644
--- a/app/models/custom_button.rb
+++ b/app/models/custom_button.rb
@@ -14,13 +14,28 @@ class CustomButton < ApplicationRecord
   acts_as_miq_set_member
 
   BUTTON_CLASSES = [
+    AvailabilityZone,
+    CloudNetwork,
+    CloudObjectStoreContainer,
+    CloudSubnet,
     CloudTenant,
+    CloudVolume,
+    ContainerGroup,
+    ContainerImage,
+    ContainerProject,
+    ContainerTemplate,
+    ContainerVolume,
     EmsCluster,
     ExtManagementSystem,
     Host,
+    LoadBalancer,
     MiqTemplate,
+    NetworkRouter,
+    OrchestrationStack,
+    SecurityGroup,
     Service,
     Storage,
+    Switch,
     Vm,
   ].freeze