From afa4d45ee904092ef17f4cb1a6189817819adc67 Mon Sep 17 00:00:00 2001
From: "Callahan, Michael Joseph" <callahm3@miamioh.edu>
Date: Tue, 4 Aug 2015 15:58:32 -0400
Subject: [PATCH] Added an ensure parameter for user control of proxy presence.
 Defaults to undef to resist breaking any current setup.

added spec entry

Added validation

fixed ensure hash
---
 README.md                | 2 ++
 manifests/init.pp        | 6 +++++-
 manifests/params.pp      | 7 ++++---
 spec/classes/apt_spec.rb | 8 ++++++++
 4 files changed, 19 insertions(+), 4 deletions(-)

diff --git a/README.md b/README.md
index 9506495..4220775 100644
--- a/README.md
+++ b/README.md
@@ -229,6 +229,8 @@ Main class, includes all other classes.
 
   * 'https': Specifies whether to enable https proxies. Valid options: 'true' and 'false'. Default: 'false'.
 
+  * 'ensure': Optional parameter. Valid options: 'file', 'present', and 'absent'. Default: 'undef'. Prefer 'file' over 'present'.
+
 * `purge`: Specifies whether to purge any existing settings that aren't managed by Puppet. Valid options: a hash made up from the following keys:
 
   * 'sources.list': Specifies whether to purge any unmanaged entries from `sources.list`. Valid options: 'true' and 'false'. Default: 'false'.
diff --git a/manifests/init.pp b/manifests/init.pp
index 578d733..b9087cf 100644
--- a/manifests/init.pp
+++ b/manifests/init.pp
@@ -45,6 +45,9 @@ class apt(
   $_purge = merge($::apt::purge_defaults, $purge)
 
   validate_hash($proxy)
+  if $proxy['ensure'] {
+    validate_re($proxy['ensure'], ['file', 'present', 'absent'])
+  }
   if $proxy['host'] {
     validate_string($proxy['host'])
   }
@@ -64,8 +67,9 @@ class apt(
   validate_hash($settings)
   validate_hash($ppas)
 
-  if $proxy['host'] {
+  if $_proxy['ensure'] == 'absent' or $_proxy['host'] {
     apt::setting { 'conf-proxy':
+      ensure   => $_proxy['ensure'],
       priority => '01',
       content  => template('apt/_conf_header.erb', 'apt/proxy.erb'),
     }
diff --git a/manifests/params.pp b/manifests/params.pp
index cdf8e6e..ea18460 100644
--- a/manifests/params.pp
+++ b/manifests/params.pp
@@ -76,9 +76,10 @@ class apt::params {
   }
 
   $proxy_defaults = {
-    'host'  => undef,
-    'port'  => 8080,
-    'https' => false,
+    'ensure' => undef,
+    'host'   => undef,
+    'port'   => 8080,
+    'https'  => false,
   }
 
   $purge_defaults = {
diff --git a/spec/classes/apt_spec.rb b/spec/classes/apt_spec.rb
index 5a71fb5..7578d2e 100644
--- a/spec/classes/apt_spec.rb
+++ b/spec/classes/apt_spec.rb
@@ -91,6 +91,14 @@ describe 'apt' do
         /Acquire::https::proxy "https:\/\/localhost:8080\/";/
       )}
     end
+
+    context 'ensure=absent' do
+      let(:params) { { :proxy => { 'ensure' => 'absent'} } }
+      it { is_expected.to contain_apt__setting('conf-proxy').with({
+        :ensure   => 'absent',
+        :priority => '01',
+      })}
+    end
   end
   context 'lots of non-defaults' do
     let :params do
-- 
2.32.3