Skip to content

Commit

Permalink
Fix the dynamic task naming generation for subclasses
Browse files Browse the repository at this point in the history
for subclasses like ServiceOrchestration we need the base class name
for Vms we just need the class name
  • Loading branch information
d-m-u committed Mar 4, 2019
1 parent e3e7c38 commit cc245ab
Show file tree
Hide file tree
Showing 2 changed files with 50 additions and 19 deletions.
13 changes: 10 additions & 3 deletions app/models/service_retire_task.rb
Original file line number Diff line number Diff line change
@@ -1,3 +1,4 @@

class ServiceRetireTask < MiqRetireTask
default_value_for :request_type, "service_retire"

Expand Down Expand Up @@ -51,18 +52,24 @@ def create_retire_subtasks(parent_service, parent_task)
end

def create_task(svc_rsc, parent_service, nh, parent_task)
resource_type = svc_rsc.resource.type.presence || "Service"
(resource_type.demodulize + "RetireTask").constantize.new(nh).tap do |task|
@task_type = retire_task_type(svc_rsc.resource.class)
@task_type.new(nh).tap do |task|
task.options.merge!(
:src_ids => [svc_rsc.resource.id],
:service_resource_id => svc_rsc.id,
:parent_service_id => parent_service.id,
:parent_task_id => parent_task.id,
)
task.request_type = resource_type.demodulize.underscore.downcase + "_retire"
task.request_type = @task_type.name.underscore[0..-6]
task.source = svc_rsc.resource
parent_task.miq_request_tasks << task
task.save!
end
end

private

def retire_task_type(resource_type)
(resource_type.base_class.name + "RetireTask").safe_constantize || (resource_type.name.demodulize + "RetireTask").safe_constantize
end
end
56 changes: 40 additions & 16 deletions spec/models/service_retire_task_spec.rb
Original file line number Diff line number Diff line change
Expand Up @@ -37,6 +37,7 @@

expect(service_retire_task.description).to eq("Service Retire for: #{service.name} - ")
expect(VmRetireTask.count).to eq(0)
expect(ServiceRetireTask.count).to eq(1)
end
end

Expand All @@ -47,26 +48,49 @@
miq_request.approve(approver, reason)
end

context "resource lacks type" do
it "creates service retire subtask" do
resource = FactoryBot.create(:service_resource, :resource_type => nil, :service_id => service.id, :resource_id => vm.id)
service.service_resources << resource
service_retire_task.after_request_task_create
it "creates service retire subtask" do
service.add_resource!(FactoryBot.create(:service_orchestration))
service_retire_task.after_request_task_create

expect(service_retire_task.description).to eq("Service Retire for: #{service.name} - ")
expect(ServiceRetireTask.count).to eq(1)
end
expect(service_retire_task.description).to eq("Service Retire for: #{service.name} - ")
expect(ServiceRetireTask.count).to eq(2)
end

context "resource has type" do
it "creates vm retire subtask" do
resource = FactoryBot.create(:service_resource, :resource_type => "VmOrTemplate", :service_id => service.id, :resource_id => vm.id)
service.service_resources << resource
service_retire_task.after_request_task_create
it "creates service retire subtask" do
service.add_resource!(FactoryBot.create(:service))
service_retire_task.after_request_task_create

expect(service_retire_task.description).to eq("Service Retire for: #{service.name} - ")
expect(VmRetireTask.count).to eq(1)
end
expect(service_retire_task.description).to eq("Service Retire for: #{service.name} - ")
expect(ServiceRetireTask.count).to eq(2)
end

it "creates stack retire subtask" do
service.add_resource!(FactoryBot.create(:orchestration_stack))
service_retire_task.after_request_task_create

expect(service_retire_task.description).to eq("Service Retire for: #{service.name} - ")
expect(OrchestrationStackRetireTask.count).to eq(1)
expect(ServiceRetireTask.count).to eq(1)
end

it "doesn't create subtask for miq_provision_request_template" do
admin = FactoryBot.create(:user_admin)
vm_template = FactoryBot.create(:vm_openstack, :ext_management_system => FactoryBot.create(:ext_management_system))
service.add_resource!(FactoryBot.create(:miq_provision_request_template, :requester => admin, :src_vm_id => vm_template.id))
service_retire_task.after_request_task_create

expect(service_retire_task.description).to eq("Service Retire for: #{service.name} - ")
expect(MiqRetireTask.count).to eq(1)
expect(ServiceRetireTask.count).to eq(1)
end

it "creates vm retire subtask" do
service.add_resource!(FactoryBot.create(:vm_openstack))
service_retire_task.after_request_task_create

expect(service_retire_task.description).to eq("Service Retire for: #{service.name} - ")
expect(VmRetireTask.count).to eq(1)
expect(ServiceRetireTask.count).to eq(1)
end
end

Expand Down

0 comments on commit cc245ab

Please sign in to comment.