[UnitTest] Add unit test for cfn_hup_configuration resource

Author: Himani Anil Deshpande

cookbooks/aws-parallelcluster-environment/resources/config_cfn_hup.rb renamed to cookbooks/aws-parallelcluster-environment/resources/cfn_hup_configuration.rb

@@ -0,0 +1,116 @@
+require 'spec_helper'
+
+class ConvergeCfnHupConfiguration
+  def self.configure(chef_run)
+    chef_run.converge_dsl('aws-parallelcluster-environment') do
+      cfn_hup_configuration 'configure' do
+        action :configure
+      end
+    end
+  end
+end
+
+AWS_REGION = "AWS_REGION".freeze
+AWS_DOMAIN = "AWS_DOMAIN".freeze
+STACK_ID = "STACK_ID".freeze
+CLOUDFORMATION_URL = "https://cloudformation.#{AWS_REGION}.#{AWS_DOMAIN}".freeze
+INSTANCE_ROLE_NAME = "INSTANCE_ROLE_NAME".freeze
+LAUNCH_TEMPLATE_ID = "LAUNCH_TEMPLATE_ID".freeze
+SCRIPT_DIR = "SCRIPT_DIR".freeze
+MONITOR_SHARED_DIR = "MONITOR_SHARED_DIR".freeze
+
+describe 'cfn_hup_configuration:configure' do
+  for_all_oses do |platform, version|
+    context "on #{platform}#{version}" do
+      for_all_node_types do |node_type|
+        context "when #{node_type}" do
+          cached(:chef_run) do
+            runner = runner(platform: platform, version: version, step_into: ['cfn_hup_configuration']) do |node|
+              allow_any_instance_of(Object).to receive(:get_metadata_token).and_return("IMDS_TOKEN")
+              allow_any_instance_of(Object).to receive(:get_metadata_with_token)
+                                                 .with("IMDS_TOKEN", URI("http://169.254.169.254/latest/meta-data/iam/security-credentials"))
+                                                 .and_return(INSTANCE_ROLE_NAME)
+              node.override["cluster"]["node_type"] = node_type
+              node.override["cluster"]["region"] = AWS_REGION
+              node.override["cluster"]["aws_domain"] = AWS_DOMAIN
+              # TODO: We inject the stack id into the attribute stack_arn when generating the dna.json in the CLI.
+              #  This should be fixed at the CLI level first and adapt the cookbook accordingly.
+              node.override["cluster"]["stack_arn"] = STACK_ID
+              node.override["cluster"]["launch_template_id"] = LAUNCH_TEMPLATE_ID
+              node.override['cluster']['scripts_dir'] = SCRIPT_DIR
+              node.override['cluster']['shared_dir'] = MONITOR_SHARED_DIR
+            end
+            ConvergeCfnHupConfiguration.configure(runner)
+          end
+          cached(:node) { chef_run.node }
+
+          %w(/etc/cfn /etc/cfn/hooks.d).each do |dir|
+            it "creates the directory #{dir}" do
+              is_expected.to create_directory(dir)
+                               .with(owner: 'root')
+                               .with(group: 'root')
+                               .with(mode:  "0770")
+                               .with(recursive: true)
+            end
+          end
+
+          it "creates the file /etc/cfn/cfn-hup.conf" do
+            is_expected.to create_template("/etc/cfn/cfn-hup.conf")
+                             .with(source: 'cfn_hup/cfn-hup.conf.erb')
+                             .with(user: "root")
+                             .with(group: "root")
+                             .with(mode: "0400")
+                             .with(variables: {
+                               stack_id: STACK_ID,
+                               region: AWS_REGION,
+                               cloudformation_url: CLOUDFORMATION_URL,
+                               cfn_init_role: INSTANCE_ROLE_NAME,
+                             })
+          end
+
+          it "creates the file /etc/cfn/hooks.d/pcluster-update.conf" do
+            is_expected.to create_template("/etc/cfn/hooks.d/pcluster-update.conf")
+                             .with(source: 'cfn_hup/cfn-hook-update.conf.erb')
+                             .with(user: "root")
+                             .with(group: "root")
+                             .with(mode: "0400")
+                             .with(variables: {
+                               stack_id: STACK_ID,
+                               region: AWS_REGION,
+                               cloudformation_url: CLOUDFORMATION_URL,
+                               cfn_init_role: INSTANCE_ROLE_NAME,
+                               launch_template_resource_id: LAUNCH_TEMPLATE_ID,
+                               update_hook_script_dir: SCRIPT_DIR,
+                             })
+          end
+
+          if %(ComputeFleet).include?(node_type)
+            it "creates the file #{SCRIPT_DIR}/cfn-hup-update-action.sh" do
+              is_expected.to create_template("#{SCRIPT_DIR}/cfn-hup-update-action.sh")
+                               .with(source: "cfn_hup/#{node_type}/cfn-hup-update-action.sh.erb")
+                               .with(user: "root")
+                               .with(group: "root")
+                               .with(mode: "0744")
+                               .with(variables: {
+                                 monitor_shared_dir: "#{MONITOR_SHARED_DIR}/dna",
+                                 launch_template_resource_id: LAUNCH_TEMPLATE_ID,
+                               })
+            end
+          elsif node_type == 'HeadNode'
+            it "creates #{SCRIPT_DIR}/get_compute_user_data.py" do
+              is_expected.to create_if_missing_cookbook_file("#{SCRIPT_DIR}/get_compute_user_data.py")
+                               .with(source: 'cfn_hup/get_compute_user_data.py')
+                               .with(user: 'root')
+                               .with(group: 'root')
+                               .with(mode: '0755')
+            end
+
+            it "creates the directory #{MONITOR_SHARED_DIR}/dna" do
+              is_expected.to create_directory("#{MONITOR_SHARED_DIR}/dna")
+            end
+          end
+        end
+      end
+    end
+  end
+end