Update spec/functions/stdlib_ensure_spec.rb

Co-authored-by: Ewoud Kohl van Wijngaarden <ewoud@kohlvanwijngaarden.nl>

Author: b4ldr

functions/ensure.pp

@@ -3,7 +3,7 @@
 # @return [String]
 function stdlib::ensure(
   Variant[Boolean, Enum['present', 'absent']] $ensure,
-  Enum['present', 'directory', 'link', 'mounted', 'service', 'file', 'package'] $resource = 'present',
+  Optional[Enum['directory', 'link', 'mounted', 'service', 'file', 'package']] $resource = undef,
 ) >> String {
   $_ensure = $ensure ? {
     Boolean => $ensure.bool2str('present', 'absent'),
@@ -22,6 +22,7 @@ function stdlib::ensure(
         default   => 'stopped',
       }
     }
+    undef: { $_ensure }
     default: {
       $_ensure ? {
         'present' => $resource,

spec/functions/stdlib_ensure_spec.rb

@@ -3,11 +3,9 @@
 require 'spec_helper'
 
 describe 'stdlib::ensure' do
-  context 'work with generi resource' do
+  context 'work with generic resource' do
     it { is_expected.to run.with_params(true).and_return('present') }
     it { is_expected.to run.with_params(false).and_return('absent') }
-    it { is_expected.to run.with_params(true, 'present').and_return('present') }
-    it { is_expected.to run.with_params(false, 'present').and_return('absent') }
   end
   context 'work with service resource' do
     it { is_expected.to run.with_params('present', 'service').and_return('running') }