First pass of adding network module (DataDog#346)

Author: jameynelson

manifests/integrations/network.pp

@@ -0,0 +1,43 @@
+# Class: datadog_agent::integrations::network
+#
+# This class will install the network integration
+#
+# Parameters:
+#   $collect_connection_state
+#       Enable TCP connection state counts
+#
+# Sample Usage:
+#
+#  class { 'datadog_agent::integrations::network' :
+#    collect_connection_state     => false,
+#    excluded_interfaces => [
+#     'lo0',
+#     'lo',
+#    ],
+#    excluded_interface_re => [
+#    'eth1.*'
+#    ]
+#    combine_connection_states => true
+#  }
+#
+#
+class datadog_agent::integrations::network(
+  $collect_connection_state = false,
+  $excluded_interfaces = ['lo','lo0'],
+  $excluded_interface_re = [],
+  $combine_connection_states = true,
+) inherits datadog_agent::params {
+  include ::datadog_agent
+
+  validate_array($excluded_interfaces)
+
+  file { "${datadog_agent::params::conf_dir}/network.yaml":
+    ensure  => file,
+    owner   => $datadog_agent::params::dd_user,
+    group   => $datadog_agent::params::dd_group,
+    mode    => '0600',
+    content => template('datadog_agent/agent-conf.d/network.yaml.erb'),
+    require => Package[$datadog_agent::params::package_name],
+    notify  => Service[$datadog_agent::params::service_name],
+  }
+}

spec/classes/datadog_agent_intergrations_network_spec.rb

@@ -0,0 +1,43 @@
+require 'spec_helper'
+
+describe 'datadog_agent::integrations::network' do
+  let(:facts) {{
+    operatingsystem: 'Ubuntu',
+  }}
+  let(:conf_dir) { '/etc/dd-agent/conf.d' }
+  let(:dd_user) { 'dd-agent' }
+  let(:dd_group) { 'root' }
+  let(:dd_package) { 'datadog-agent' }
+  let(:dd_service) { 'datadog-agent' }
+  let(:conf_file) { "#{conf_dir}/network.yaml" }
+
+  context 'with default parameters' do
+    it { should compile }
+  end
+
+  context 'with collect_connection_state set' do
+    let(:params) {{
+      collect_connection_state: true,
+    }}
+
+    it { should compile.with_all_deps }
+    it { should contain_file(conf_file).with(
+      owner: dd_user,
+      group: dd_group,
+      mode: '0600',
+    )}
+    it { should contain_file(conf_file).that_requires("Package[#{dd_package}]") }
+    it { should contain_file(conf_file).that_notifies("Service[#{dd_service}]") }
+
+    it { should contain_file(conf_file).with_content(%r{collect_connection_state: true}) }
+    it { should contain_file(conf_file).without_content(%r{tags: }) }
+
+    context 'with excluded_interfaces parameter array' do
+      let(:params) {{
+        excluded_interfaces: %w{ lo lo0 eth0 },
+      }}
+      it { should contain_file(conf_file).with_content(/excluded_interfaces:[^-]+- lo\s+- lo0\s+- eth0\s*?[^-]/m) }
+    end
+
+  end
+end

templates/agent-conf.d/network.yaml.erb

@@ -0,0 +1,20 @@
+#                   #
+# MANAGED BY PUPPET #
+#                   #
+init_config:
+
+instances:
+  - collect_connection_state: <%= @collect_connection_state %>
+<% if !@excluded_interfaces.empty? -%>
+    excluded_interfaces:
+<%- (@excluded_interfaces.each do |interface| -%>
+      - <%= interface %>
+<% end ) -%>
+<% end -%>
+<% if !@excluded_interface_re.empty? -%>
+    excluded_interface_re:
+<%- (@excluded_interfaces_re.each do |interface| -%>
+      - <%= interface %>
+<% end ) -%>
+<% end -%>
+    combine_connection_states: <%= @combine_connection_states %>