Support compliance_engine::enforcement as a String

This fixes backwards-compatibility with a compliance_markup edge case.

Also fix running `lookup()` tests directly.

Author: silug

.gitignore

@@ -16,3 +16,4 @@
 
 /Gemfile.lock
 /vendor/
+/spec/fixtures/modules/

Rakefile

@@ -11,6 +11,12 @@ RSpec::Core::RakeTask.new('spec:standalone') do |t|
   t.pattern = 'spec/{classes,functions}/**/*_spec.rb'
 end
 
+desc 'Prepare fixtures for testing'
+task spec_prep: :'fixtures:prep'
+
+desc 'Clean up fixtures after testing'
+task spec_clean: :'fixtures:clean'
+
 desc 'Run spec tests and clean the fixtures directory if successful'
 task spec: :'fixtures:prep' do |_t, args|
   Rake::Task['spec:standalone'].invoke(*args.extras)

lib/puppet/functions/compliance_engine/enforcement.rb

@@ -60,10 +60,10 @@ def enforcement(key, options, context)
   end
 
   def profiles
-    profile_list = call_function('lookup', 'compliance_engine::enforcement', { 'default_value' => [] })
+    profile_list = Array(call_function('lookup', 'compliance_engine::enforcement', { 'default_value' => [] }))
 
     # For backwards compatibility with compliance_markup.
-    profile_list += call_function('lookup', 'compliance_markup::enforcement', { 'default_value' => [] }) if @compat
+    profile_list += Array(call_function('lookup', 'compliance_markup::enforcement', { 'default_value' => [] })) if @compat
 
     profile_list.uniq
   end

spec/functions/lookup/10_enforce_spec.rb

@@ -98,7 +98,6 @@ def write_hieradata(hieradata_dir, hieradata_file, policy_order)
           let(:policy_order) { 'disa_stig' }
 
           it 'returns /bin/disa' do
-            skip('String value for compliance_engine::enforcement not supported, must be Array')
             result = lookup.execute('useradd::shells')
             expect(result).to be_instance_of(Array)
             expect(result).to include('/bin/disa')