allow selection unstable, nightly, etc repo type (#31)

.gitignore

@@ -5,3 +5,5 @@ travis.log
 .vagrant
 spec/fixtures/*
 junit
+.bundle/
+vendor/

manifests/init.pp

@@ -57,6 +57,9 @@
 # [*manage_repo*]
 #   Boolean.  Whether or not to manage InfluxData's repo.
 #
+# [*repo_type*]
+#   String.  Which repo (stable, unstable, nightly) to use
+#
 class telegraf (
   $ensure                 = $telegraf::params::ensure,
   $config_file            = $telegraf::params::config_file,
@@ -75,6 +78,7 @@
   $global_tags            = $telegraf::params::global_tags,
   $manage_service         = $telegraf::params::manage_service,
   $manage_repo            = $telegraf::params::manage_repo,
+  $repo_type              = $telegraf::params::repo_type,
 ) inherits ::telegraf::params
 {
 
@@ -95,6 +99,7 @@
   validate_hash($global_tags)
   validate_bool($manage_service)
   validate_bool($manage_repo)
+  validate_string($repo_type)
 
   # currently the only way how to obtain merged hashes
   # from multiple files (`:merge_behavior: deeper` needs to be

manifests/install.pp

@@ -16,7 +16,7 @@
           comment  => 'Mirror for InfluxData packages',
           location => "https://repos.influxdata.com/${_operatingsystem}",
           release  => $::distcodename,
-          repos    => 'stable',
+          repos    => $::telegraf::repo_type,
           key      => {
             'id'     => '05CE15085FC09D18E99EFB22684A14CF2582E0C5',
             'source' => 'https://repos.influxdata.com/influxdb.key',
@@ -28,7 +28,7 @@
         yumrepo { 'influxdata':
           descr    => 'influxdata',
           enabled  => 1,
-          baseurl  => "https://repos.influxdata.com/rhel/${::operatingsystemmajrelease}/${::architecture}/stable",
+          baseurl  => "https://repos.influxdata.com/rhel/${::operatingsystemmajrelease}/${::architecture}/${::telegraf::repo_type}",
           gpgkey   => 'https://repos.influxdata.com/influxdb.key',
           gpgcheck => true,
         }

manifests/params.pp

@@ -19,6 +19,7 @@
   $global_tags            = {}
   $manage_service         = true
   $manage_repo            = true
+  $repo_type              = 'stable'
 
   $outputs = {
     'influxdb'  => {

spec/classes/telegraf_spec.rb

@@ -6,11 +6,13 @@
       [6,7].each do |releasenum|
         context "RedHat #{releasenum} release specifics" do
           let(:facts) {{
-            :osfamily               => 'RedHat',
-            :kernel                 => 'Linux',
-            :operatingsystem        => osfamily,
-            :operatingsystemrelease => releasenum,
-            :role                   => 'telegraf'
+            :osfamily                  => 'RedHat',
+            :architecture              => 'x86_64',
+            :kernel                    => 'Linux',
+            :operatingsystem           => osfamily,
+            :operatingsystemrelease    => releasenum,
+            :operatingsystemmajrelease => releasenum,
+            :role                      => 'telegraf'
           }}
           it { should compile.with_all_deps }
           it { should contain_class('telegraf::config') }
@@ -77,7 +79,20 @@
           it { should contain_file('/etc/telegraf/telegraf.conf') }
           it { should contain_package('telegraf') }
           it { should contain_service('telegraf') }
-          it { should contain_yumrepo('influxdata') }
+          it { should contain_yumrepo('influxdata')
+            .with(
+              :baseurl => "https://repos.influxdata.com/rhel/#{facts[:operatingsystemmajrelease]}/#{facts[:architecture]}/stable",
+            )
+          }
+
+          describe 'allow custom repo_type' do
+            let(:params) { {:repo_type => 'unstable' } }
+            it { should contain_yumrepo('influxdata')
+              .with(
+                :baseurl => "https://repos.influxdata.com/rhel/#{facts[:operatingsystemmajrelease]}/#{facts[:architecture]}/unstable",
+              )
+            }
+          end
         end
       end
     end