Allow removing check config files

Adds a new 'ensure' argument to 'datadog_agent::integration' that can be
'present' (default) or 'absent' (to remove the file).

Author: albertvaka

manifests/integration.pp

@@ -1,8 +1,9 @@
 define datadog_agent::integration (
-  Array $instances,
-  Optional[Hash] $init_config = undef,
-  Optional[Array] $logs       = undef,
-  String $integration         = $title,
+  Array $instances                  = [],
+  Optional[Hash] $init_config       = undef,
+  Optional[Array] $logs             = undef,
+  String $integration               = $title,
+  Enum['present', 'absent'] $ensure = 'present',
 ){
 
   include datadog_agent
@@ -21,7 +22,7 @@
   }
 
   file { $dst:
-    ensure  => file,
+    ensure  => $ensure ? { default => file, 'absent' => absent },
     owner   => $datadog_agent::dd_user,
     group   => $datadog_agent::dd_group,
     mode    => $datadog_agent::params::permissions_file,

spec/defines/datadog_agent__integration_spec.rb

@@ -4,38 +4,41 @@
   context 'supported agents' do
     ALL_SUPPORTED_AGENTS.each do |agent_major_version|
       let(:pre_condition) { "class {'::datadog_agent': agent_major_version => #{agent_major_version}}" }
+
+      let(:title) { 'test' }
+
       if agent_major_version == 5
         let(:conf_file) { '/etc/dd-agent/conf.d/test.yaml' }
-
       else
         let(:conf_dir) { "#{CONF_DIR}/test.d" }
         let(:conf_file) { "#{conf_dir}/conf.yaml" }
       end
 
-      let(:title) { 'test' }
-      let(:params) do
-        {
-          instances: [
-            {
-              one: 'two',
-            },
-          ],
-        }
-      end
+      gem_spec = Gem.loaded_specs['puppet']
 
-      it { is_expected.to compile }
       if agent_major_version == 5
         it { is_expected.to contain_file(conf_dir.to_s).that_comes_before("File[#{conf_file}]") }
       end
-      it { is_expected.to contain_file(conf_file.to_s).with_content(%r{init_config: }) }
-      gem_spec = Gem.loaded_specs['puppet']
-      if gem_spec.version >= Gem::Version.new('4.0.0')
-        it { is_expected.to contain_file(conf_file.to_s).with_content(%r{---\ninit_config: \ninstances:\n- one: two\n}) }
-      else
-        it { is_expected.to contain_file(conf_file.to_s).with_content(%r{--- \n  init_config: \n  instances: \n    - one: two}) }
-      end
       it { is_expected.to contain_file(conf_file.to_s).that_notifies("Service[#{SERVICE_NAME}]") }
 
+      context 'with instances' do
+        let(:params) do
+          {
+            instances: [
+              {
+                one: 'two',
+              },
+            ],
+          }
+        end
+        it { is_expected.to compile }
+        if gem_spec.version >= Gem::Version.new('4.0.0')
+          it { is_expected.to contain_file(conf_file.to_s).with_content(%r{---\ninit_config:\ninstances:\n- one: two\n}) }
+        else
+          it { is_expected.to contain_file(conf_file.to_s).with_content(%r{--- \n  init_config:\n  instances: \n    - one: two}) }
+        end
+      end
+
       context 'with logs' do
         let(:params) do
           {
@@ -47,13 +50,24 @@
             logs: ['one', 'two'],
           }
         end
-
+        it { is_expected.to compile }
         if gem_spec.version >= Gem::Version.new('4.0.0')
           it { is_expected.to contain_file(conf_file).with_content(%r{logs:\n- one\n- two}) }
         else
           it { is_expected.to contain_file(conf_file).with_content(%r{logs:\n  - one\n  - two}) }
         end
       end
+
+      context 'with ensure absent' do
+        let(:params) do
+          {
+            ensure: 'absent',
+          }
+        end
+        it { is_expected.to compile }
+        it { is_expected.to contain_file(conf_file).with_ensure('absent') }
+      end
+
     end
   end
 end