spec/classes/datadog_agent_integrations_postgres_spec.rb

require 'spec_helper'

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

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

      context 'with password set' do
        let(:params) {{
          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(/password: abc123/) }

        context 'with default parameters' do
          it { should contain_file(conf_file).with_content(%r{host: localhost}) }
          it { should contain_file(conf_file).with_content(%r{dbname: postgres}) }
          it { should contain_file(conf_file).with_content(%r{port: 5432}) }
          it { should contain_file(conf_file).with_content(%r{username: datadog}) }
          it { should contain_file(conf_file).without_content(%r{^\s*use_psycopg2: }) }
          it { should contain_file(conf_file).without_content(%r{tags: })}
          it { should contain_file(conf_file).without_content(%r{^[^#]*relations: }) }
        end

        context 'with use_psycopg2 enabled' do
          let(:params) {{
            use_psycopg2: true,
            password: 'abc123',
          }}
          it { should contain_file(conf_file).with_content(%r{use_psycopg2: true}) }
        end

        context 'with parameters set' do
          let(:params) {{
            host: 'postgres1',
            dbname: 'cats',
            port: 4142,
            username: 'monitoring',
            password: 'abc123',
            tags: %w{foo bar baz},
            tables: %w{furry fuzzy funky}
          }}
          it { should contain_file(conf_file).with_content(%r{host: postgres1}) }
          it { should contain_file(conf_file).with_content(%r{dbname: cats}) }
          it { should contain_file(conf_file).with_content(%r{port: 4142}) }
          it { should contain_file(conf_file).with_content(%r{username: monitoring}) }
          it { should contain_file(conf_file).with_content(%r{^[^#]*tags:\s+- foo\s+- bar\s+- baz}) }
          it { should contain_file(conf_file).with_content(%r{^[^#]*relations:\s+- furry\s+- fuzzy\s+- funky}) }

          context 'with custom metric query missing %s' do
            let(:params) {{
              host: 'postgres1',
              dbname: 'cats',
              port: 4142,
              username: 'monitoring',
              password: 'abc123',
              custom_metrics: {
                'query_is_missing_%s' => {
                  'query' => 'select * from fuzz',
                  'metrics' => { },
                }
              }
            }}
            it do
              expect {
                is_expected.to compile
              }.to raise_error(/custom_metrics require %s for metric substitution/)
            end
          end

          context 'with custom metric query' do
            let(:params) {{
              host: 'postgres1',
              dbname: 'cats',
              port: 4142,
              username: 'monitoring',
              password: 'abc123',
              custom_metrics: {
                'foo_gooo_bar_query' => {
                  'query' => 'select foo, %s from bar',
                  'metrics' => {
                    "gooo" => ["custom_metric.tag.gooo", "GAUGE"]
                  },
                  'descriptors' => [["foo", "custom_metric.tag.foo"]]
                }
              }
            }}
            it { is_expected.to compile }
            it { should contain_file(conf_file).with_content(%r{^[^#]*custom_metrics:}) }
            it { should contain_file(conf_file).with_content(%r{\s+query:\s*['"]?select foo, %s from bar['"]?}) }
            it { should contain_file(conf_file).with_content(%r{\s+metrics:}) }
            it { should contain_file(conf_file).with_content(%r{\s+"gooo":\s+\[custom_metric.tag.gooo, GAUGE\]}) }
            it { should contain_file(conf_file).with_content(%r{\s+query.*\n\s+relation:\s*false}) }
            it { should contain_file(conf_file).with_content(%r{\s+descriptors.*\n\s+-\s+\[foo, custom_metric.tag.foo\]}) }
          end
        end
      end
    end
  end
end