Added a define that wraps the agent integration command

Author: albertvaka

README.md

@@ -147,6 +147,22 @@ Once the `datadog_agent` module is installed on your `puppetserver`/`puppetmaste
     }
     ```
 
+Installing and pinning specific versions of integrations
+--------------------------------------------------------
+
+    You can specify a given integration and version number to be installed by using `datadog_agent::install_integration`. This will use the `datadog-agent integration` command to ensure a specific integration is installed or uninstalled.
+
+    ```
+    datadog_agent::install_integration { "mongo-1.9":
+      ensure => present,
+      integration_name => 'datadog-mongo',
+      version => '1.9.0',
+    }
+    ```
+
+    The field `ensure` can be either `present` (default) or `absent`, the later being useful to remove a previously pinned version of an integration.
+
+
 Reporting
 ---------
 Ensure `dogapi-rb` is available on your system as explained earlier.

manifests/install_integration.pp

@@ -0,0 +1,21 @@
+define datadog_agent::install_integration (
+  Enum['present', 'absent'] $ensure           = 'present',
+  String                    $integration_name = undef,
+  String                    $version          = undef,
+){
+
+  include datadog_agent
+
+  if $ensure == 'present' {
+    $command = 'install'
+  } else {
+    $command = 'remove'
+  }
+
+  exec { "${command} ${integration_name}==${version}":
+    command => "/opt/datadog-agent/bin/agent/agent integration ${command} ${integration_name}==${version}",
+    user    => 'dd-agent',
+    require => Package[$datadog_agent::params::package_name],
+  }
+
+}

spec/defines/datadog_agent__install_integration_spec.rb

@@ -0,0 +1,33 @@
+require 'spec_helper'
+
+describe "datadog_agent::install_integration" do
+
+  describe "installing an integration" do
+    let(:pre_condition) { "class {'::datadog_agent': }" }
+    let (:title) { "test" }
+    let (:params) {{
+      :integration_name => "datadog-mongo",
+      :version => "1.9.0",
+    }}
+
+    it { should compile }
+
+    it { is_expected.to contain_exec("install datadog-mongo==1.9.0").with_command("/opt/datadog-agent/bin/agent/agent integration install datadog-mongo==1.9.0") }
+
+  end
+
+  describe "removing an integration" do
+    let(:pre_condition) { "class {'::datadog_agent': }" }
+    let (:title) { "test" }
+    let (:params) {{
+      :integration_name => "datadog-mongo",
+      :version => "1.9.0",
+      :ensure => "absent",
+    }}
+
+    it { should compile }
+
+    it { is_expected.to contain_exec("remove datadog-mongo==1.9.0").with_command("/opt/datadog-agent/bin/agent/agent integration remove datadog-mongo==1.9.0") }
+
+  end
+end