Support DD EU in reporter (#468)

* Write api_url in the report config

* Use api_url to choose datadog host

* Update spec tests

* Add a default value for datadog_site

Author: remicalixte

lib/puppet/reports/datadog_reports.rb

@@ -13,6 +13,7 @@
   raise(Puppet::ParseError, "Datadog report config file #{configfile} not readable") unless File.readable?(configfile)
   config = YAML.load_file(configfile)
   API_KEY = config[:datadog_api_key]
+  API_URL = config[:api_url]
 
   # if need be initialize the regex
   if !config[:hostname_extraction_regex].nil?
@@ -110,7 +111,7 @@ def process
     end
 
     Puppet.debug "Sending metrics for #{@msg_host} to Datadog"
-    @dog = Dogapi::Client.new(API_KEY)
+    @dog = Dogapi::Client.new(API_KEY, nil, nil, nil, nil, nil, API_URL)
     @dog.batch_metrics do
       self.metrics.each { |metric,data|
         data.values.each { |val|

manifests/init.pp

@@ -663,6 +663,7 @@
   if $puppet_run_reports {
     class { 'datadog_agent::reports':
       api_key                   => $api_key,
+      datadog_site              => $datadog_site,
       dogapi_version            => $datadog_agent::params::dogapi_version,
       puppetmaster_user         => $puppetmaster_user,
       hostname_extraction_regex => $hostname_extraction_regex,

manifests/reports.pp

@@ -17,7 +17,8 @@
   $api_key,
   $puppetmaster_user,
   $dogapi_version,
-  $hostname_extraction_regex = nil
+  $hostname_extraction_regex = nil,
+  $datadog_site = 'datadoghq.com',
 ) {
 
   include datadog_agent

spec/classes/datadog_agent_reports_spec.rb

@@ -95,4 +95,42 @@
       end
     end
   end
+
+  context 'EU site in report' do
+    let(:params) do
+      {
+        api_key: 'notanapikey',
+        hostname_extraction_regex: nil,
+        puppetmaster_user: 'puppet',
+        dogapi_version: 'installed',
+        datadog_site: 'datadoghq.eu',
+      }
+    end
+    describe "datadog_agent class dogapi version override" do
+      let(:facts) do
+        {
+          operatingsystem: 'Debian',
+          osfamily: 'debian'
+        }
+      end
+
+      it { should contain_class('ruby').with_rubygems_update(false) }
+      it { should contain_class('ruby::params') }
+      it { should contain_package('ruby').with_ensure('installed') }
+      it { should contain_package('rubygems').with_ensure('installed') }
+
+      it do
+        should contain_package('ruby-dev')\
+          .with_ensure('installed')\
+          .that_comes_before('Package[dogapi]')
+      end
+
+      it do
+        should contain_file('/etc/datadog-agent/datadog-reports.yaml')\
+          .with_owner('puppet')\
+          .with_group('root')\
+          .with_content(/:api_url: api.datadoghq.eu/)
+      end
+    end
+  end
 end

templates/datadog-reports.yaml.erb

@@ -3,6 +3,7 @@
 #
 ---
 :datadog_api_key: '<%= @api_key %>'
+:api_url: api.<%= @datadog_site %>
 <% if !@hostname_extraction_regex.nil? -%>
 :hostname_extraction_regex: '<%= @hostname_extraction_regex %>'
 <% end -%>