Override get_description method

ManageIQ · Feb 28, 2018 · 955ef67 · 955ef67
1 parent fbfaeec
commit 955ef67
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/app/models/service_template_transformation_plan_task.rb b/app/models/service_template_transformation_plan_task.rb
@@ -3,8 +3,13 @@ def self.base_model
     ServiceTemplateTransformationPlanTask
   end
 
+  def self.get_description(req_obj)
+    source_name = req_obj.source.name
+    req_obj.kind_of?(ServiceTemplateTransformationPlanRequest) ? source_name : "Transforming VM [#{source_name}]"
+  end
+
   def after_request_task_create
-    update_attributes(:description => "Transforming VM #{source.name}")
+    update_attributes(:description => get_description)
   end
 
   def resource_action

diff --git a/spec/models/service_template_transformation_plan_task_spec.rb b/spec/models/service_template_transformation_plan_task_spec.rb
@@ -72,5 +72,15 @@
         expect(plan.vm_resources.first.status).to eq('Completed')
       end
     end
+
+    describe '.get_description' do
+      it 'describes a task' do
+        expect(described_class.get_description(task)).to include("Transforming VM")
+      end
+
+      it 'describe a request' do
+        expect(described_class.get_description(request)).to eq(plan.name)
+      end
+    end
   end
 end