Converted Service node to use its decorator for icons/image

ManageIQ · Jan 10, 2017 · 814d3be · 814d3be
1 parent 63a79a6
commit 814d3be
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/app/presenters/tree_node/service.rb b/app/presenters/tree_node/service.rb
@@ -1,5 +1,12 @@
 module TreeNode
   class Service < Node
-    set_attribute(:image) { @object.picture ? "/pictures/#{@object.picture.basename}" : '100/service.png' }
+    set_attributes(:image, :icon) do
+      if @object.picture
+        image = @object.decorate.listicon_image
+      else
+        icon = 'product product-service'
+      end
+      [image, icon]
+    end
   end
 end
diff --git a/spec/presenters/tree_node/service_spec.rb b/spec/presenters/tree_node/service_spec.rb
@@ -5,5 +5,5 @@
   let(:object) { FactoryGirl.create(:service) }
 
   include_examples 'TreeNode::Node#key prefix', 's-'
-  include_examples 'TreeNode::Node#image', '100/service.png'
+  include_examples 'TreeNode::Node#icon', 'product product-service'
 end