K8s + K8s_state integrations

manifests/integrations/kubernetes.pp

@@ -0,0 +1,50 @@
+# Class: datadog_agent::integrations::kubernetes
+#
+# This class will install the necessary configuration for the kubernetes integration
+#
+# Parameters:
+#   $url:
+#     The URL for kubernetes API
+#
+#   $tags:
+#     optional array of tags
+#
+#
+#
+# Sample Usage:
+#
+#   class { 'datadog_agent::integrations::kubernetes' :
+#     api_server_url       => 'https://kubernetes:443',
+#     apiserver_client_crt => '/etc/ssl/certs/crt',
+#     apiserver_client_key => '/etc/ssl/private/key',
+#     kubelet_client_crt   => '/etc/ssl/certs/crt',
+#     kubelet_client_key   => '/etc/ssl/private/key',
+#   }
+#
+class datadog_agent::integrations::kubernetes(
+  $api_server_url = 'Enter_Your_API_url',
+  $apiserver_client_crt = '/path/to/crt',
+  $apiserver_client_key = '/path/to/key',
+  $kubelet_client_crt = '/path/to/crt',
+  $kubelet_client_key = '/path/to/key',
+  $tags = [],
+
+) inherits datadog_agent::params {
+  include datadog_agent
+
+  if $::datadog_agent::agent6_enable {
+    $dst = "${datadog_agent::conf6_dir}/kubernetes.yaml"
+  } else {
+    $dst = "${datadog_agent::conf_dir}/kubernetes.yaml"
+  }
+
+  file { $dst:
+    ensure  => file,
+    owner   => $datadog_agent::params::dd_user,
+    group   => $datadog_agent::params::dd_group,
+    mode    => '0644',
+    require => Package[$datadog_agent::params::package_name],
+    notify  => Service[$datadog_agent::params::service_name],
+    content => template('datadog_agent/agent-conf.d/kubernetes.yaml.erb'),
+  }
+}

manifests/integrations/kubernetes_state.pp

@@ -0,0 +1,42 @@
+# Class: datadog_agent::integrations::kubernetes
+#
+# This class will install the necessary configuration for the kubernetes integration
+#
+# Parameters:
+#   $url:
+#     The URL for kubernetes metrics 
+#
+#   $tags:
+#     optional array of tags
+#
+#
+#
+# Sample Usage:
+#
+#   class { 'datadog_agent::integrations::kubernetes_state' :
+#     url  => 'http://kubernetes.com:8080/metrics',
+#   }
+#
+class datadog_agent::integrations::kubernetes_state(
+  $url = 'Enter_State_URL',
+  $tags = [],
+
+) inherits datadog_agent::params {
+  include datadog_agent
+
+  if $::datadog_agent::agent6_enable {
+    $dst = "${datadog_agent::conf6_dir}/kubernetes_state.yaml"
+  } else {
+    $dst = "${datadog_agent::conf_dir}/kubernetes_state.yaml"
+  }
+
+  file { $dst:
+    ensure  => file,
+    owner   => $datadog_agent::params::dd_user,
+    group   => $datadog_agent::params::dd_group,
+    mode    => '0644',
+    require => Package[$datadog_agent::params::package_name],
+    notify  => Service[$datadog_agent::params::service_name],
+    content => template('datadog_agent/agent-conf.d/kubernetes_state.yaml.erb'),
+  }
+}

spec/classes/datadog_agent_integrations_kubernetes_spec.rb

@@ -0,0 +1,87 @@
+require 'spec_helper'
+
+describe 'datadog_agent::integrations::kubernetes' do
+  context 'supported agents - v5 and v6' do
+    agents = { '5' => false, '6' => true }
+    agents.each do |_, enabled|
+      let(:pre_condition) { "class {'::datadog_agent': agent6_enable => #{enabled}}" }
+      let(:facts) {{
+        operatingsystem: 'Ubuntu',
+      }}
+      if !enabled
+        let(:conf_dir) { '/etc/dd-agent/conf.d' }
+      else
+        let(:conf_dir) { '/etc/datadog-agent/conf.d' }
+      end
+      let(:dd_user) { 'dd-agent' }
+      let(:dd_group) { 'root' }
+      let(:dd_package) { 'datadog-agent' }
+      let(:dd_service) { 'datadog-agent' }
+      if !enabled
+        let(:conf_file) { "#{conf_dir}/kubernetes.yaml" }
+      else
+        let(:conf_file) { "#{conf_dir}/kubernetes.yaml" }
+      end
+
+      it { should compile.with_all_deps }
+      it { should contain_file(conf_file).with(
+        owner: dd_user,
+        group: dd_group,
+        mode: '0644',
+      )}
+      it { should contain_file(conf_file).that_requires("Package[#{dd_package}]") }
+      it { should contain_file(conf_file).that_notifies("Service[#{dd_service}]") }
+
+      context 'with default parameters' do
+        it { should contain_file(conf_file).with_content(%r{api_server_url: Enter_Your_API_url}) }
+        it { should contain_file(conf_file).with_content(%r{apiserver_client_crt: /path/to/crt}) }
+        it { should contain_file(conf_file).with_content(%r{apiserver_client_key: /path/to/key}) }
+        it { should contain_file(conf_file).with_content(%r{kubelet_client_crt: /path/to/crt}) }
+        it { should contain_file(conf_file).with_content(%r{kubelet_client_key: /path/to/key}) }
+      end
+
+      context 'with parameters set' do
+        let(:params) {{
+          api_server_url: 'unix://foo/bar/baz.sock',
+        }}
+        it { should contain_file(conf_file).with_content(%r{api_server_url: unix://foo/bar/baz.sock}) }
+      end
+
+      context 'with tags parameter array' do
+        let(:params) {{
+          tags: %w{ foo bar baz },
+        }}
+        it { should contain_file(conf_file).with_content(/tags:\s+- foo\s+- bar\s+- baz\s*?[^-]/m) }
+      end
+
+      context 'with tags parameter with an empty tag' do
+      end
+
+      context 'with tags parameter empty values' do
+        context 'mixed in with other tags' do
+          let(:params) {{
+            tags: [ 'foo', '', 'baz' ]
+          }}
+
+          it { should contain_file(conf_file).with_content(/tags:\s+- foo\s+- baz\s*?[^-]/m) }
+        end
+
+        context 'single element array of an empty string' do
+          let(:params) {{
+            tags: [''],
+          }}
+
+          skip("undefined behavior")
+        end
+
+        context 'single value empty string' do
+          let(:params) {{
+            tags: '',
+          }}
+
+          skip("doubly undefined behavior")
+        end
+      end
+    end
+  end
+end

spec/classes/datadog_agent_integrations_kubernetes_state_spec.rb

@@ -0,0 +1,83 @@
+require 'spec_helper'
+
+describe 'datadog_agent::integrations::kubernetes_state' do
+  context 'supported agents - v5 and v6' do
+    agents = { '5' => false, '6' => true }
+    agents.each do |_, enabled|
+      let(:pre_condition) { "class {'::datadog_agent': agent6_enable => #{enabled}}" }
+      let(:facts) {{
+        operatingsystem: 'Ubuntu',
+      }}
+      if !enabled
+        let(:conf_dir) { '/etc/dd-agent/conf.d' }
+      else
+        let(:conf_dir) { '/etc/datadog-agent/conf.d' }
+      end
+      let(:dd_user) { 'dd-agent' }
+      let(:dd_group) { 'root' }
+      let(:dd_package) { 'datadog-agent' }
+      let(:dd_service) { 'datadog-agent' }
+      if !enabled
+        let(:conf_file) { "#{conf_dir}/kubernetes_state.yaml" }
+      else
+        let(:conf_file) { "#{conf_dir}/kubernetes_state.yaml" }
+      end
+
+      it { should compile.with_all_deps }
+      it { should contain_file(conf_file).with(
+        owner: dd_user,
+        group: dd_group,
+        mode: '0644',
+      )}
+      it { should contain_file(conf_file).that_requires("Package[#{dd_package}]") }
+      it { should contain_file(conf_file).that_notifies("Service[#{dd_service}]") }
+
+      context 'with default parameters' do
+        it { should contain_file(conf_file).with_content(%r{kube_state_url: Enter_State_URL}) }
+      end
+
+      context 'with parameters set' do
+        let(:params) {{
+          url: 'Some_Other_State_URL',
+        }}
+        it { should contain_file(conf_file).with_content(%r{kube_state_url: Some_Other_State_URL}) }
+      end
+
+      context 'with tags parameter array' do
+        let(:params) {{
+          tags: %w{ foo bar baz },
+        }}
+        it { should contain_file(conf_file).with_content(/tags:\s+- foo\s+- bar\s+- baz\s*?[^-]/m) }
+      end
+
+      context 'with tags parameter with an empty tag' do
+      end
+
+      context 'with tags parameter empty values' do
+        context 'mixed in with other tags' do
+          let(:params) {{
+            tags: [ 'foo', '', 'baz' ]
+          }}
+
+          it { should contain_file(conf_file).with_content(/tags:\s+- foo\s+- baz\s*?[^-]/m) }
+        end
+
+        context 'single element array of an empty string' do
+          let(:params) {{
+            tags: [''],
+          }}
+
+          skip("undefined behavior")
+        end
+
+        context 'single value empty string' do
+          let(:params) {{
+            tags: '',
+          }}
+
+          skip("doubly undefined behavior")
+        end
+      end
+    end
+  end
+end