Move repo_class to parameter and put value in module hiera data

elfranne · Jun 1, 2018 · af9a588 · af9a588
1 parent fc141ea
commit af9a588
Show file tree

Hide file tree

Showing 4 changed files with 19 additions and 2 deletions.
diff --git a/data/os/RedHat.yaml b/data/os/RedHat.yaml
@@ -0,0 +1,3 @@
+---
+sensu::repo_class: 'sensu::repo::yum'
+
diff --git a/hiera.yaml b/hiera.yaml
@@ -0,0 +1,11 @@
+---
+version: 5
+defaults:
+  datadir: data
+  data_hash: yaml_data
+hierarchy:
+  - name: "osfamily"
+    path: "os/%{facts.os.family}.yaml"
+  - name: "common"
+    path: "common.yaml"
+
diff --git a/manifests/init.pp b/manifests/init.pp
@@ -22,8 +22,10 @@
   },
   Boolean $etc_dir_purge = true,
   Boolean $manage_repo = true,
+  Optional[String] $repo_class = undef,
 ) {
 
+  include ::sensu::repo
   include ::sensu::agent
 
   file { 'sensu_etc_dir':
@@ -35,11 +37,9 @@
 
   case $osfamily {
     'RedHat': {
-      $repo_class = '::sensu::repo::yum'
     }
     default: {
       fail("Detected osfamily <${::osfamily}>. Only RedHat is supported.")
     }
   }
-  include ::sensu::repo
 }
diff --git a/spec/spec_helper.rb b/spec/spec_helper.rb
@@ -30,6 +30,9 @@
     :ipaddress                 => '127.0.0.1',
     :kernel                    => 'Linux',
     :osfamily                  => 'RedHat',
+    :os                        => {
+      :family => 'RedHat',
+    },
     :operatingsystem           => 'RedHat',
     :operatingsystemmajrelease => '7',
     :fqdn                      => 'testfqdn.example.com',