Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add queue_name to queued method options for CloudVolume model #19570

Merged
merged 2 commits into from
Dec 3, 2019
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
22 changes: 21 additions & 1 deletion app/models/cloud_volume.rb
Original file line number Diff line number Diff line change
Expand Up @@ -35,18 +35,26 @@ def self.class_by_ems(ext_management_system)
ext_management_system && ext_management_system.class::CloudVolume
end

# Create a cloud volume as a queued task and return the task id. The queue
# name and the queue zone are derived from the provided EMS instance. The EMS
# instance and a userid are mandatory. Any +options+ are forwarded as
# arguments to the +create_volume+ method.
#
def self.create_volume_queue(userid, ext_management_system, options = {})
task_opts = {
:action => "creating Cloud Volume for user #{userid}",
:userid => userid
}

queue_opts = {
:class_name => "CloudVolume",
:class_name => 'CloudVolume',
:method_name => 'create_volume',
:role => 'ems_operations',
:queue_name => ext_management_system.queue_name_for_ems_operations,
:zone => ext_management_system.my_zone,
:args => [ext_management_system.id, options]
}

MiqTask.generic_action_with_callback(task_opts, queue_opts)
end

Expand All @@ -70,19 +78,25 @@ def self.raw_create_volume(_ext_management_system, _options = {})
raise NotImplementedError, _("raw_create_volume must be implemented in a subclass")
end

# Update a cloud volume as a queued task and return the task id. The queue
# name and the queue zone are derived from the EMS, and a userid is mandatory.
#
def update_volume_queue(userid, options = {})
task_opts = {
:action => "updating Cloud Volume for user #{userid}",
:userid => userid
}

queue_opts = {
:class_name => self.class.name,
:method_name => 'update_volume',
:instance_id => id,
:role => 'ems_operations',
:queue_name => ext_management_system.queue_name_for_ems_operations,
:zone => ext_management_system.my_zone,
:args => [options]
}

MiqTask.generic_action_with_callback(task_opts, queue_opts)
end

Expand All @@ -98,19 +112,25 @@ def raw_update_volume(_options = {})
raise NotImplementedError, _("raw_update_volume must be implemented in a subclass")
end

# Delete a cloud volume as a queued task and return the task id. The queue
# name and the queue zone are derived from the EMS, and a userid is mandatory.
#
def delete_volume_queue(userid)
task_opts = {
:action => "deleting Cloud Volume for user #{userid}",
:userid => userid
}

queue_opts = {
:class_name => self.class.name,
:method_name => 'delete_volume',
:instance_id => id,
:role => 'ems_operations',
:queue_name => ext_management_system.queue_name_for_ems_operations,
:zone => ext_management_system.my_zone,
:args => []
}

MiqTask.generic_action_with_callback(task_opts, queue_opts)
end

Expand Down
85 changes: 80 additions & 5 deletions spec/models/cloud_volume_spec.rb
Original file line number Diff line number Diff line change
@@ -1,9 +1,84 @@
describe CloudVolume do
RSpec.describe CloudVolume do
let(:disks) { FactoryBot.create_list(:disk, 2) }
let(:ems) { FactoryBot.create(:ems_vmware) }
let(:cloud_volume) { FactoryBot.create(:cloud_volume, :ext_management_system => ems, :attachments => disks) }
let(:user) { FactoryBot.create(:user, :userid => 'test') }

it ".available" do
disk = FactoryBot.create(:disk)
FactoryBot.create(:cloud_volume, :attachments => [disk])
cv2 = FactoryBot.create(:cloud_volume)
cloud_volumes_no_backing_disks = FactoryBot.create_list(:cloud_volume, 2)
expect(described_class.available).to eq(cloud_volumes_no_backing_disks)
end

context 'queued methods' do
it 'queues a create task with create_volume_queue' do
task_id = described_class.create_volume_queue(user.userid, ems)

expect(MiqTask.find(task_id)).to have_attributes(
:name => "creating Cloud Volume for user #{user.userid}",
:state => "Queued",
:status => "Ok"
)

expect(MiqQueue.where(:class_name => described_class.name).first).to have_attributes(
:class_name => described_class.name,
:method_name => 'create_volume',
:role => 'ems_operations',
:queue_name => 'generic',
:zone => ems.my_zone,
:args => [ems.id, {}]
)
end

it 'requires a userid and ems for a queued create task' do
expect { described_class.create_volume_queue }.to raise_error(ArgumentError)
expect { described_class.create_volume_queue(user.userid) }.to raise_error(ArgumentError)
end

it 'queues an update task with update_volume_queue' do
options = {:name => 'updated_volume_name'}
task_id = cloud_volume.update_volume_queue(user.userid, options)

expect(MiqTask.find(task_id)).to have_attributes(
:name => "updating Cloud Volume for user #{user.userid}",
:state => "Queued",
:status => "Ok"
)

expect(MiqQueue.where(:class_name => described_class.name).first).to have_attributes(
:class_name => described_class.name,
:method_name => 'update_volume',
:role => 'ems_operations',
:queue_name => 'generic',
:zone => ems.my_zone,
:args => [options]
)
end

it 'requires a userid for a queued update task' do
expect { cloud_volume.update_volume_queue }.to raise_error(ArgumentError)
end

it 'queues a delete task with delete_volume_queue' do
task_id = cloud_volume.delete_volume_queue(user.userid)

expect(MiqTask.find(task_id)).to have_attributes(
:name => "deleting Cloud Volume for user #{user.userid}",
:state => "Queued",
:status => "Ok"
)

expect(MiqQueue.where(:class_name => described_class.name).first).to have_attributes(
:class_name => described_class.name,
:method_name => 'delete_volume',
:role => 'ems_operations',
:queue_name => 'generic',
:zone => ems.my_zone,
:args => []
)
end

expect(described_class.available).to eq([cv2])
it 'requires a userid for a queued delete task' do
expect { cloud_volume.delete_volume_queue }.to raise_error(ArgumentError)
end
end
end