[apm] adding apm tests.

Author: truthbk

manifests/init.pp

@@ -398,6 +398,12 @@
     order   => '05',
   }
 
+  concat::fragment{ 'datadog apm footer':
+    target  => '/etc/dd-agent/datadog.conf',
+    content => template('datadog_agent/datadog_apm_footer.conf.erb'),
+    order   => '06',
+  }
+
 
   if $puppet_run_reports {
     class { 'datadog_agent::reports':

spec/classes/datadog_agent_spec.rb

@@ -206,7 +206,7 @@
                 end
 
                 context 'for APM' do
-                    it { should contain_file('/etc/dd-agent/datadog.conf').with(
+                    it { should contain_concat__fragment('datadog footer').with(
                     'content' => /^apm_enabled: false\n/,
                     )}
                 end
@@ -509,7 +509,7 @@
             end
             context 'with apm_enabled set to true' do
                 let(:params) {{:apm_enabled  => true }}
-                it { should contain_file('/etc/dd-agent/datadog.conf').with(
+                it { should contain_concat__fragment('datadog footer').with(
                     'content' => /^apm_enabled: true\n/,
                 )}
             end
@@ -518,13 +518,13 @@
                     {:apm_enabled  => true,
                      :apm_env => 'foo',
                 }}
-                it { should contain_file('/etc/dd-agent/datadog.conf').with(
+                it { should contain_concat__fragment('datadog footer').with(
                     'content' => /^apm_enabled: true\n/,
                 )}
-                it { should contain_file('/etc/dd-agent/datadog.conf').with(
+                it { should contain_concat__fragment('datadog apm footer').with(
                     'content' => /^\[trace.agent\]\n/,
                 )}
-                it { should contain_file('/etc/dd-agent/datadog.conf').with(
+                it { should contain_concat__fragment('datadog apm footer').with(
                     'content' => /^env: foo\n/,
                 )}
             end

templates/datadog.conf.erb

@@ -433,8 +433,3 @@ apm_enabled: <%= @apm_enabled %>
 # ========================================================================== #
 
 <% end -%>
-
-<% if not @apm_env.empty? -%>
-[trace.agent]
-env: <%= @apm_env %>
-<% end -%>

templates/datadog_apm_footer.conf.erb

@@ -0,0 +1,4 @@
+<% if not @apm_env.empty? -%>
+[trace.agent]
+env: <%= @apm_env %>
+<% end -%>

templates/datadog_footer.conf.erb

@@ -343,6 +343,14 @@ sd_jmx_enable: <%= @sd_jmx_enable %>
 consul_token: <%= @consul_token %>
 <% end -%>
 
+# ========================================================================== #
+# Trace                                                                      #
+# ========================================================================== #
+
+# Enable the trace agent.
+apm_enabled: <%= @apm_enabled %>
+
+
 <% if not @extra_template.empty? -%>
 # ========================================================================== #
 # Custom Templates from Puppet                                               #