From 3e244bdf91740170b5b61e35bf57b84459571470 Mon Sep 17 00:00:00 2001
From: Trevor Vaughan <trevor@sicura.us>
Date: Sat, 22 Jan 2022 15:49:48 -0500
Subject: [PATCH] Amazon Linux 2 Support

Closes #84
---
 .gitlab-ci.yml                     | 13 +++++++++++++
 CHANGELOG                          |  3 +++
 metadata.json                      |  8 +++++++-
 spec/acceptance/nodesets/amzn2.yml | 23 +++++++++++++++++++++++
 spec/spec_helper_acceptance.rb     |  6 +++++-
 5 files changed, 51 insertions(+), 2 deletions(-)
 create mode 100644 spec/acceptance/nodesets/amzn2.yml

diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index bbb1bee..b727ea1 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -368,6 +368,12 @@ pup6.pe-oel:
   script:
     - 'bundle exec rake beaker:suites[default,oel]'
 
+pup6.pe-amzn2:
+  <<: *pup_6_pe
+  <<: *acceptance_base
+  script:
+    - 'bundle exec rake beaker:suites[default,amzn2]'
+
 pup6.pe-oel-fips:
   <<: *pup_6_pe
   <<: *acceptance_base
@@ -394,6 +400,13 @@ pup7.x-oel:
   script:
     - 'bundle exec rake beaker:suites[default,oel]'
 
+pup7.x-amzn2:
+  <<: *pup_7_x
+  <<: *acceptance_base
+  <<: *with_SIMP_ACCEPTANCE_MATRIX_LEVEL_2
+  script:
+    - 'bundle exec rake beaker:suites[default,amzn2]'
+
 pup7.x-oel-fips:
   <<: *pup_7_x
   <<: *acceptance_base
diff --git a/CHANGELOG b/CHANGELOG
index 424f01f..5c40657 100644
--- a/CHANGELOG
+++ b/CHANGELOG
@@ -1,3 +1,6 @@
+* Sat Sep 17 2022 Trevor Vaughan <trevor@sicura.us> - 6.7.0
+- Added support for Amazon Linux 2
+
 * Fri Jun 03 2022 Chris Tessmer <chris.tessmer@onyxpoint.com> - 6.6.0
 - Update from camptocamp/systemd to puppet/systemd
 
diff --git a/metadata.json b/metadata.json
index 35bf55b..a09ae27 100644
--- a/metadata.json
+++ b/metadata.json
@@ -1,6 +1,6 @@
 {
   "name": "simp-aide",
-  "version": "6.6.0",
+  "version": "6.7.0",
   "author": "SIMP Team",
   "summary": "manages AIDE",
   "license": "Apache-2.0",
@@ -46,6 +46,12 @@
     ]
   },
   "operatingsystem_support": [
+    {
+      "operatingsystem": "Amazon",
+      "operatingsystemrelease": [
+        "2"
+      ]
+    },
     {
       "operatingsystem": "CentOS",
       "operatingsystemrelease": [
diff --git a/spec/acceptance/nodesets/amzn2.yml b/spec/acceptance/nodesets/amzn2.yml
new file mode 100644
index 0000000..1368180
--- /dev/null
+++ b/spec/acceptance/nodesets/amzn2.yml
@@ -0,0 +1,23 @@
+<%
+  if ENV['BEAKER_HYPERVISOR']
+    hypervisor = ENV['BEAKER_HYPERVISOR']
+  else
+    hypervisor = 'vagrant'
+  end
+-%>
+HOSTS:
+  amzn2:
+    roles:
+      - default
+    platform: el-7-x86_64
+    box: gbailey/amzn2
+    hypervisor: <%= hypervisor %>
+
+CONFIG:
+  log_level: verbose
+  synced_folder : disabled
+  type: aio
+  vagrant_memsize: 512
+<% if ENV['BEAKER_PUPPET_COLLECTION'] -%>
+  puppet_collection: <%= ENV['BEAKER_PUPPET_COLLECTION'] %>
+<% end -%>
diff --git a/spec/spec_helper_acceptance.rb b/spec/spec_helper_acceptance.rb
index 505d29d..9631ed6 100644
--- a/spec/spec_helper_acceptance.rb
+++ b/spec/spec_helper_acceptance.rb
@@ -32,7 +32,11 @@
     begin
       # Install modules and dependencies from spec/fixtures/modules
       copy_fixture_modules_to( hosts )
-      server = only_host_with_role(hosts, 'server')
+      begin
+        server = only_host_with_role(hosts, 'server')
+      rescue ArgumentError =>e
+        server = only_host_with_role(hosts, 'default')
+      end
 
       # Generate and install PKI certificates on each SUT
       Dir.mktmpdir do |cert_dir|