puppet-datadog-agent/spec/classes/datadog_agent_integrations_disk_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
118
119
120
121
122
require 'spec_helper'

describe 'datadog_agent::integrations::disk' 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}/disk.yaml" }
      else
        let(:conf_file) { "#{conf_dir}/disk.d/conf.yaml" }
      end

      it { is_expected.to compile.with_all_deps }
      it { is_expected.to  contain_file(conf_file).with_content(
        %r{\s+use_mount:\s+no$}
      ).with(
        owner: dd_user,
        group: dd_group,
        mode: '0600',
      )}
      it { is_expected.to contain_file(conf_file).that_requires("Package[#{dd_package}]") }
      it { is_expected.to contain_file(conf_file).that_notifies("Service[#{dd_service}]") }

      context 'compile errors for incorrect values' do
        let(:params) {{ use_mount: 'heaps' }}
        it do
          expect { is_expected.to compile }.to raise_error(/error\s+during\s+compilation/)
        end
      end

      context 'we handle strings and arrays the same' do
        let(:params) {{
          use_mount: 'yes',
          excluded_filesystems: [ 'tmpfs', 'dev' ],
          excluded_disks: '/dev/sda1',
          excluded_disk_re: '/dev/sdb.*',
          excluded_mountpoint_re: '/mnt/other.*',
          all_partitions: 'yes',
          tag_by_filesystem: 'no'
        }}
        let(:yaml_conf) {
           <<-HEREDOC
### MANAGED BY PUPPET

init_config:

instances:
  - use_mount: yes
    excluded_filesystems:
      - tmpfs
      - dev
    excluded_disks:
      - /dev/sda1
    excluded_disk_re: /dev/sdb.*
    excluded_mountpoint_re: /mnt/other.*
    all_partitions: yes
    tag_by_filesystem: no
        HEREDOC
         }
        it { is_expected.to contain_file(conf_file).with_content(yaml_conf) }
      end

      context 'we handle new disk configuration option' do
        let(:params) {{
          use_mount: 'yes',
          filesystem_blacklist: ['tmpfs', 'dev'],
          device_blacklist: ['/dev/sda1'],
          mountpoint_blacklist: ['/mnt/foo'],
          filesystem_whitelist: ['ext4', 'hdfs', 'reiserfs'],
          device_whitelist: ['/dev/sdc1', '/dev/sdc2', '/dev/sdd2'],
          mountpoint_whitelist: ['/mnt/logs', '/mnt/builds'],
          all_partitions: 'yes',
          tag_by_filesystem: 'no'
        }}
        let(:yaml_conf) {
           <<-HEREDOC
### MANAGED BY PUPPET

init_config:

instances:
  - use_mount: yes
    file_system_blacklist:
      - tmpfs
      - dev
    device_blacklist:
      - /dev/sda1
    mount_point_blacklist:
      - /mnt/foo
    file_system_whitelist:
      - ext4
      - hdfs
      - reiserfs
    device_whitelist:
      - /dev/sdc1
      - /dev/sdc2
      - /dev/sdd2
    mount_point_whitelist:
      - /mnt/logs
      - /mnt/builds
    all_partitions: yes
    tag_by_filesystem: no
        HEREDOC
         }
        it { is_expected.to contain_file(conf_file).with_content(yaml_conf) }
      end
    end
  end
end