[NVIDIA-IMEX] Installing NVIDIA IMEx using install_packages resource

Author: Himani Anil Deshpande

cookbooks/aws-parallelcluster-platform/resources/nvidia_imex/partial/_nvidia_imex_common.rb

@@ -48,11 +48,13 @@
     action :create
   end
 
-  package "nvidia-imex-#{_nvidia_imex_version}" do
-    retries 3
-    retry_delay 5
-    # version _nvidia_imex_version
+  install_packages 'Install nvidia-imex' do
+    packages "nvidia-imex-#{_nvidia_imex_version}"
+    action :install
   end
+  # Save Imex version in Node Attributes for InSpec Tests
+  node.default['cluster']['nvidia']['imex']['version'] = _nvidia_imex_version
+  node_attributes 'dump node attributes'
 end
 
 action :configure do

cookbooks/aws-parallelcluster-platform/spec/unit/resources/nvidia_imex_spec.rb

@@ -176,6 +176,7 @@ def self.install(chef_run)
             "1_1.2.3-1"
           end
         end
+        cached(:node) { chef_run.node }
 
         before do
           chef_run.node.override['cluster']['shared_dir'] = shared_dir
@@ -203,10 +204,13 @@ def self.install(chef_run)
               .with(user: 'root')
               .with(group: 'root')
               .with(mode: '0644')
-            is_expected.not_to install_package('nvidia-imex')
-              .with(retries: 3)
-              .with(retry_delay: 5)
-              .with(version: nvidia_imex_version)
+            is_expected.not_to install_install_packages('Install nvidia-imex')
+              .with(packages: "nvidia-imex-#{nvidia_imex_version}")
+              .with(action: %i(install))
+          end
+          it 'does not set nvidia-imex version' do
+            expect(node.default['cluster']['nvidia']['imex']['version']).not_to eq(nvidia_imex_version)
+            is_expected.not_to write_node_attributes('dump node attributes')
           end
         else
           it 'installs nvidia-imex' do
@@ -228,10 +232,13 @@ def self.install(chef_run)
               .with(user: 'root')
               .with(group: 'root')
               .with(mode: '0644')
-            is_expected.to install_package('nvidia-imex')
-              .with(retries: 3)
-              .with(retry_delay: 5)
-              .with(version: nvidia_imex_version)
+            is_expected.to install_install_packages('Install nvidia-imex')
+              .with(packages: "nvidia-imex-#{nvidia_imex_version}")
+              .with(action: %i(install))
+          end
+          it 'sets nvidia-imex version' do
+            expect(node.default['cluster']['nvidia']['imex']['version']).to eq(nvidia_imex_version)
+            is_expected.to write_node_attributes('dump node attributes')
           end
         end
       end

cookbooks/aws-parallelcluster-platform/test/controls/nvidia_imex_spec.rb

@@ -14,7 +14,7 @@
 
   describe package('nvidia-imex') do
     it { should be_installed }
-    its('version') { should match /#{node['cluster']['nvidia']['driver_version']}/ }
+    its('version') { should match /#{node['cluster']['nvidia']['imex']['version']}/ }
   end
 
   %w(/usr/bin/nvidia-imex /usr/bin/nvidia-imex-ctl).each do |path|