Enforce safe_mode value as boolean

sensu · May 7, 2013 · 381e45a · 381e45a
1 parent 823081c
commit 381e45a
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/lib/puppet/type/sensu_client_config.rb b/lib/puppet/type/sensu_client_config.rb
@@ -37,8 +37,9 @@ def initialize(*args)
     desc ""
   end
 
-  newproperty(:safe_mode) do
+  newproperty(:safe_mode, :boolean => true) do
     desc "Require checks to be defined on server and client"
+    newvalues(:true, :false)
   end
 
   autorequire(:package) do

diff --git a/manifests/client.pp b/manifests/client.pp
@@ -10,7 +10,7 @@
   $client_name    = $::fqdn,
   $enabled        = 'true',
   $purge_config   = 'false',
-  $safe_mode      = 'false',
+  $safe_mode      = false,
 ) {
 
   $ensure = $enabled ? {

diff --git a/spec/classes/sensu_client_spec.rb b/spec/classes/sensu_client_spec.rb
@@ -21,15 +21,15 @@
       :address                  => '1.2.3.4',
       :subscriptions            => ['all'],
       :client_name              => 'myclient',
-      :safe_mode                => 'true'
+      :safe_mode                => true
     } }
 
     it { should contain_sensu_client_config('host.domain.com').with(
       'client_name'   => 'myclient',
       'address'       => '1.2.3.4',
       'subscriptions' => ['all'],
       'ensure'        => 'present',
-      'safe_mode'     => 'true'
+      'safe_mode'     => true
     ) }
 
   end