added ensure variable to install.pp

1 files changed, 4 insertions, 3 deletions

diff --git a/manifests/install.pp b/manifests/install.pp
index c545339..47ab022 100644
--- a/manifests/install.pp
+++ b/manifests/install.pp
@@ -15,9 +15,10 @@
 # == Requires:
 #
 #   If use_rvm = 'true':
-#   include rvm::system
+#   include rvm
 #
 class bundler::install (
+  $ensure       = 'present',
   $use_rvm      = $bundler::params::use_rvm,
   $ruby_version
 ) inherits bundler::params {
@@ -25,13 +26,13 @@ class bundler::install (
   if $use_rvm == 'true' {
     #Install bundler with correct RVM
     rvm_gem { 'bundler':
-        ensure       => 'present',
+        ensure       => $ensure,
         ruby_version => $ruby_version,
      }
   }
   else {
     package { 'bundler':
-      ensure   => 'present',
+      ensure   => $ensure,
       provider => 'gem',
     }
   }