puppet-datadog-agent/spec/classes/datadog_agent_integrations_elasticsearch_spec.rb at 2593d46aa8ef1143ae30ccbf88f42b18e0adf632 · charles-ferguson/puppet-datadog-agent · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
require 'spec_helper'

describe 'datadog_agent::integrations::elasticsearch' do
  context 'supported agents' do
    ALL_SUPPORTED_AGENTS.each do |agent_major_version|
      let(:pre_condition) { "class {'::datadog_agent': agent_major_version => #{agent_major_version}}" }

      if agent_major_version == 5
        let(:conf_file) { '/etc/dd-agent/conf.d/elastic.yaml' }
      else
        let(:conf_file) { "#{CONF_DIR}/elastic.d/conf.yaml" }
      end

      it { is_expected.to compile.with_all_deps }
      it {
        is_expected.to contain_file(conf_file).with(
          owner: DD_USER,
          group: DD_GROUP,
          mode: PERMISSIONS_FILE,
        )
      }

      it { is_expected.to contain_file(conf_file).that_requires("Package[#{PACKAGE_NAME}]") }
      it { is_expected.to contain_file(conf_file).that_notifies("Service[#{SERVICE_NAME}]") }

      context 'with default parameters' do
        it { is_expected.to contain_file(conf_file).with_content(%r{    - url: http://localhost:9200}) }
        it { is_expected.to contain_file(conf_file).with_content(%r{      cluster_stats: false}) }
        it { is_expected.to contain_file(conf_file).with_content(%r{      index_stats: false}) }
        it { is_expected.to contain_file(conf_file).with_content(%r{      pending_task_stats: true}) }
        it { is_expected.to contain_file(conf_file).with_content(%r{      pshard_stats: false}) }
        it { is_expected.not_to contain_file(conf_file).with_content(%r{      username}) }
        it { is_expected.not_to contain_file(conf_file).with_content(%r{      password}) }
        it { is_expected.not_to contain_file(conf_file).with_content(%r{      ssl_verify}) }
        it { is_expected.not_to contain_file(conf_file).with_content(%r{      ssl_cert}) }
        it { is_expected.not_to contain_file(conf_file).with_content(%r{      ssl_key}) }
        it { is_expected.not_to contain_file(conf_file).with_content(%r{      tags:}) }
      end

      context 'with parameters set' do
        let(:params) do
          {
            password:           'password',
            pending_task_stats: false,
            url:                'https://foo:4242',
            username:           'username',
            ssl_cert:           '/etc/ssl/certs/client.pem',
            ssl_key:            '/etc/ssl/private/client.key',
            tags:               ['tag1:key1'],
          }
        end

        it { is_expected.to contain_file(conf_file).with_content(%r{    - url: https://foo:4242}) }
        it { is_expected.to contain_file(conf_file).with_content(%r{      pending_task_stats: false}) }
        it { is_expected.to contain_file(conf_file).with_content(%r{      username: username}) }
        it { is_expected.to contain_file(conf_file).with_content(%r{      password: password}) }
        it { is_expected.to contain_file(conf_file).with_content(%r{      ssl_verify: true}) }
        it { is_expected.to contain_file(conf_file).with_content(%r{      ssl_cert: /etc/ssl/certs/client.pem}) }
        it { is_expected.to contain_file(conf_file).with_content(%r{      ssl_key: /etc/ssl/private/client.key}) }
        it { is_expected.to contain_file(conf_file).with_content(%r{      tags:}) }
        it { is_expected.to contain_file(conf_file).with_content(%r{        - tag1:key1}) }
      end

      context 'with multiple instances set' do
        let(:params) do
          {
            instances: [
              {
                'cluster_stats'      => true,
                'index_stats'        => false,
                'password'           => 'password',
                'pending_task_stats' => false,
                'pshard_stats'       => true,
                'url'                => 'https://foo:4242',
                'username'           => 'username',
                'ssl_verify'         => true,
                'ssl_cert'           => '/etc/ssl/certs/client.pem',
                'ssl_key'            => '/etc/ssl/private/client.key',
                'tags'               => ['tag1:key1'],
              },
              {
                'cluster_stats'      => false,
                'index_stats'        => true,
                'password'           => 'password_2',
                'pending_task_stats' => true,
                'pshard_stats'       => false,
                'url'                => 'https://bar:2424',
                'username'           => 'username_2',
                'ssl_verify'         => false,
                'tags'               => ['tag2:key2'],
              },
            ],
          }
        end


        it { is_expected.to contain_file(conf_file).with_content(%r{instances:}) }
        it { is_expected.to contain_file(conf_file).with_content(%r{    - url: https://foo:4242}) }
        it { is_expected.to contain_file(conf_file).with_content(%r{      cluster_stats: true}) }
        it { is_expected.to contain_file(conf_file).with_content(%r{      index_stats: false}) }
        it { is_expected.to contain_file(conf_file).with_content(%r{      pending_task_stats: false}) }
        it { is_expected.to contain_file(conf_file).with_content(%r{      username: username}) }
        it { is_expected.to contain_file(conf_file).with_content(%r{      password: password}) }
        it { is_expected.to contain_file(conf_file).with_content(%r{      pshard_stats: true}) }
        it { is_expected.to contain_file(conf_file).with_content(%r{      ssl_verify: true}) }
        it { is_expected.to contain_file(conf_file).with_content(%r{      ssl_cert: /etc/ssl/certs/client.pem}) }
        it { is_expected.to contain_file(conf_file).with_content(%r{      ssl_key: /etc/ssl/private/client.key}) }
        it { is_expected.to contain_file(conf_file).with_content(%r{      tags:\n        - tag1:key1}) }
        it { is_expected.to contain_file(conf_file).with_content(%r{    - url: https://bar:2424}) }
        it { is_expected.to contain_file(conf_file).with_content(%r{      cluster_stats: false}) }
        it { is_expected.to contain_file(conf_file).with_content(%r{      index_stats: true}) }
        it { is_expected.to contain_file(conf_file).with_content(%r{      pending_task_stats: true}) }
        it { is_expected.to contain_file(conf_file).with_content(%r{      username: username_2}) }
        it { is_expected.to contain_file(conf_file).with_content(%r{      password: password_2}) }
        it { is_expected.to contain_file(conf_file).with_content(%r{      pshard_stats: false}) }
        it { is_expected.to contain_file(conf_file).with_content(%r{      ssl_verify: false}) }
        it { is_expected.to contain_file(conf_file).with_content(%r{      tags:\n        - tag2:key2}) }
      end
    end
  end
end