MiqProvisionTask based provisioning

Author: agrare

app/models/manageiq/providers/terraform_enterprise/automation_manager/configuration_script.rb

@@ -9,6 +9,10 @@ def self.stack_type
     "OrchestrationStack"
   end
 
+  def my_zone
+    manager&.my_zone
+  end
+
   def run(options = {})
     variables = options[:variables] || []
 

app/models/manageiq/providers/terraform_enterprise/automation_manager/orchestration_stack/status.rb

@@ -16,6 +16,10 @@ def canceled?
     %w[discarded canceled force_canceled].include?(status)
   end
 
+  def running?
+    !succeeded? && !failed? && !canceled?
+  end
+
   # Lookup table of run state descriptions
   # https://developer.hashicorp.com/terraform/cloud-docs/api-docs/run#run-states
   def self.run_state_to_description(status)

app/models/manageiq/providers/terraform_enterprise/automation_manager/provision.rb

@@ -0,0 +1,17 @@
+class ManageIQ::Providers::TerraformEnterprise::AutomationManager::Provision < MiqProvisionTask
+  include StateMachine
+
+  TASK_DESCRIPTION = N_("Terraform Enterprise Workspace Provision")
+
+  def self.request_class
+    MiqProvisionConfigurationScriptRequest
+  end
+
+  def my_role(*)
+    "ems_operations"
+  end
+
+  def my_queue_name
+    source.manager&.queue_name_for_ems_operations
+  end
+end

app/models/manageiq/providers/terraform_enterprise/automation_manager/provision/state_machine.rb

@@ -0,0 +1,45 @@
+module ManageIQ::Providers::TerraformEnterprise::AutomationManager::Provision::StateMachine
+  def run_provision
+    signal :provision
+  end
+
+  def provision
+    stack_class = "#{source.class.module_parent}::#{source.class.stack_type}".constantize
+    stack       = stack_class.create_stack(source)
+
+    phase_context[:stack_id] = stack.id
+    save!
+
+    signal :check_provisioned
+  end
+
+  def check_provisioned
+    if running?
+      stack.refresh_ems
+      requeue_phase
+    else
+      signal :post_provision
+    end
+  end
+
+  def post_provision
+    signal :mark_as_completed
+  end
+
+  def running?
+    stack.raw_status.running?
+  end
+
+  def mark_as_completed
+    update_and_notify_parent(:state => 'provisioned')
+    signal :finish
+  end
+
+  def finish
+    mark_execution_servers
+  end
+
+  def stack
+    @stack ||= source.manager.orchestration_stacks.find(phase_context[:stack_id])
+  end
+end

app/models/manageiq/providers/terraform_enterprise/automation_manager/provision_workflow.rb

@@ -0,0 +1,6 @@
+class ManageIQ::Providers::TerraformEnterprise::AutomationManager::ProvisionWorkflow < ManageIQ::Providers::AutomationManager::ProvisionWorkflow
+  def dialog_name_from_automate(message = 'get_dialog_name', extra_attrs = {})
+    extra_attrs['platform_category'] ||= 'terraform_enterprise'
+    super
+  end
+end

spec/factories/miq_request_task.rb

@@ -0,0 +1,3 @@
+FactoryBot.define do
+  factory :miq_provision_terraform_enterprise, :parent => :miq_provision, :class => "ManageIQ::Providers::TerraformEnterprise::AutomationManager::Provision"
+end

spec/models/manageiq/providers/terraform_enterprise/automation_manager/provision_spec.rb

@@ -0,0 +1,27 @@
+describe ManageIQ::Providers::TerraformEnterprise::AutomationManager::Provision do
+  let(:admin)       { FactoryBot.create(:user_admin) }
+  let(:ems)         { FactoryBot.create(:ems_terraform_enterprise) }
+  let(:workspace)   { FactoryBot.create(:configuration_script_terraform_enterprise, :manager => ems) }
+  let(:miq_request) { FactoryBot.create(:miq_provision_request, :requester => admin, :source => workspace)}
+  let(:options)     { {:source => [workspace.id, workspace.name]} }
+  let(:subject)     do
+    FactoryBot.create(
+      :miq_provision_terraform_enterprise,
+      :userid       => admin.userid,
+      :miq_request  => miq_request,
+      :source       => workspace,
+      :request_type => 'template',
+      :state        => 'pending',
+      :status       => 'Ok',
+      :options      => options
+    )
+  end
+
+  it ".my_role" do
+    expect(subject.my_role).to eq("ems_operations")
+  end
+
+  it ".my_queue_name" do
+    expect(subject.my_queue_name).to eq(ems.queue_name_for_ems_operations)
+  end
+end