puppet-datadog-agent/spec/classes/datadog_agent_integrations_activemq_xml_spec.rb at 7721dc384bdcb54e83d9e93d4d65935af6e4dcb7 · DataDog/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
require 'spec_helper'

describe 'datadog_agent::integrations::activemq_xml' do
  context 'supported agents - v5 and v6' do
    agents = { '5' => true, '6' => false }
    agents.each do |_, is_agent5|
      let(:pre_condition) { "class {'::datadog_agent': agent5_enable => #{is_agent5}}" }
      let(:facts) {{
        operatingsystem: 'Ubuntu',
      }}
      if is_agent5
        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 is_agent5
        let(:conf_file) { "#{conf_dir}/activemq_xml.yaml" }
      else
        let(:conf_file) { "#{conf_dir}/activemq_xml.d/conf.yaml" }
      end

      context 'with default parameters' do
        it { should compile }
      end

      context 'with password set' do
        let(:params) {{
          username: 'foo',
          password: 'abc123',
        }}

        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(/username: foo/) }
        it { should contain_file(conf_file).with_content(/password: abc123/) }

        context 'with default parameters' do
          it {
            should contain_file(conf_file)
                .with_content(%r{http://localhost:8161})
                .with_content(%r{supress_errors: false})
                .without_content(%r{detailed_queues:})
                .without_content(%r{detailed_topics:})
                .without_content(%r{detailed_subscribers:})
          }
        end

        context 'with extra detailed parameters' do
          let(:params) {{
            supress_errors: true,
            detailed_queues: %w(queue1 queue2),
            detailed_topics: %w(topic1 topic2),
            detailed_subscribers: %w(subscriber1 subscriber2),
          }}
          it {
            should contain_file(conf_file)
              .with_content(%r{http://localhost:8161})
              .with_content(%r{supress_errors: true})
              .with_content(%r{detailed_queues:.*\s+- queue1\s+- queue2})
              .with_content(%r{detailed_topics:.*\s+- topic1\s+- topic2})
              .with_content(%r{detailed_subscribers:.*\s+- subscriber1\s+- subscriber2})
          }
        end

        context 'with instances set' do
          let(:params) {{
            instances: [
                {
                    'url'     => 'http://localhost:8161',
                    'username' => 'joe',
                    'password' => 'hunter1',
                    'detailed_queues' => %w(queue1 queue2),
                    'detailed_topics' => %w(topic1 topic2),
                    'detailed_subscribers' => %w(subscriber1 subscriber2),
                },
                {
                    'url'     => 'http://remotehost:8162',
                    'username' => 'moe',
                    'password' => 'hunter2',
                    'detailed_queues' => %w(queue3 queue4),
                    'detailed_topics' => %w(topic3 topic4),
                    'detailed_subscribers' => %w(subscriber3 subscriber4),
                },
            ],
          }}
          it {
            should contain_file(conf_file)
              .with_content(%r{url: http://localhost:8161})
              .without_content(%r{supress_errors:})
              .with_content(%r{username: joe})
              .with_content(%r{password: hunter1})
              .with_content(%r{detailed_queues:.*\s+- queue1\s+- queue2})
              .with_content(%r{detailed_topics:.*\s+- topic1\s+- topic2})
              .with_content(%r{detailed_subscribers:.*\s+- subscriber1\s+- subscriber2})
              .with_content(%r{url: http://remotehost:8162})
              .without_content(%r{supress_errors:})
              .with_content(%r{username: moe})
              .with_content(%r{password: hunter2})
              .with_content(%r{detailed_queues:.*\s+- queue3\s+- queue4})
              .with_content(%r{detailed_topics:.*\s+- topic3\s+- topic4})
              .with_content(%r{detailed_subscribers:.*\s+- subscriber3\s+- subscriber4})
          }
        end
      end
    end
  end
end