require 'spec_helper'

describe 'collectd::plugin::dbi', type: :class do
  on_supported_os(baseline_os_hash).each do |os, facts|
    context "on #{os} " do
      let :facts do
        facts
      end

      options = os_specific_options(facts)
      context ':ensure => present and default parameters' do
        it "Will create #{options[:plugin_conf_dir]}/10-dbi.conf to load the plugin" do
          is_expected.to contain_file('dbi.load').with(
            ensure: 'present',
            path: "#{options[:plugin_conf_dir]}/10-dbi.conf",
            content: %r{LoadPlugin dbi}
          )
        end
        it "Will create #{options[:plugin_conf_dir]}/dbi-config.conf" do
          is_expected.to contain_concat("#{options[:plugin_conf_dir]}/dbi-config.conf").
            that_requires('File[collectd.d]')
          is_expected.to contain_concat__fragment('collectd_plugin_dbi_conf_header').with(content: %r{<Plugin dbi>})
        end
      end

      context ':ensure => present and create a db with a query' do
        let(:params) do
          {
            databases: {
              'mydatabase' => {
                'host'         => 'localhost',
                'driver'       => 'mysql',
                'driveroption' => {
                  'host'     => 'db.example.com',
                  'username' => 'dbuser',
                  'password' => 'dbpasswd'
                },
                'selectdb' => 'db',
                'query'    => %w[disk_io log_delay]
              }
            },
            queries: {
              'log_delay' => {
                'statement' => 'SELECT * FROM log_delay_repli;',
                'results'   => [{
                  'type'           => 'gauge',
                  'instanceprefix' => 'log_delay',
                  'instancesfrom'  => %w[inet_server_port inet_server_host],
                  'valuesfrom'     => 'log_delay'
                }]
              }
            }
          }
        end

        it "Will create #{options[:plugin_conf_dir]}/dbi-config.conf" do
          is_expected.to contain_concat__fragment('collectd_plugin_dbi_conf_db_mydatabase').with(content: %r{Driver \"mysql\"\n})
          is_expected.to contain_concat__fragment('collectd_plugin_dbi_conf_db_mydatabase').with(content: %r{Host \"localhost\"\n})
          is_expected.not_to contain_concat__fragment('collectd_plugin_dbi_conf_db_mydatabase').with(content: %r{Interval})
          is_expected.to contain_concat__fragment('collectd_plugin_dbi_conf_db_mydatabase').with(content: %r{Query \"disk_io\"\n})
          is_expected.to contain_concat__fragment('collectd_plugin_dbi_conf_db_mydatabase').with(content: %r{DriverOption \"host\" \"db\.example\.com\"\n})
          is_expected.to contain_concat__fragment('collectd_plugin_dbi_conf_db_mydatabase').with(content: %r{DriverOption \"username\" \"dbuser\"\n})
          is_expected.to contain_concat__fragment('collectd_plugin_dbi_conf_db_mydatabase').with(content: %r{DriverOption \"password\" \"dbpasswd\"\n})

          is_expected.to contain_concat__fragment('collectd_plugin_dbi_conf_query_log_delay').with(content: %r{Statement \"SELECT \* FROM log_delay_repli;\"\n})
          is_expected.to contain_concat__fragment('collectd_plugin_dbi_conf_query_log_delay').with(content: %r{<Result>\n})
          is_expected.to contain_concat__fragment('collectd_plugin_dbi_conf_query_log_delay').with(content: %r{InstancesFrom \"inet_server_port\" \"inet_server_host\"\n})
          is_expected.to contain_concat__fragment('collectd_plugin_dbi_conf_query_log_delay').with(content: %r{ValuesFrom \"log_delay\"\n})
        end
      end

      context ':ensure => present and create a db with a query and certain interval' do
        let(:params) do
          {
            databases: {
              'mydatabase' => {
                'host'              => 'localhost',
                'driver'            => 'mysql',
                'db_query_interval' => 60,
                'driveroption'      => {
                  'host'     => 'db.example.com',
                  'username' => 'dbuser',
                  'password' => 'dbpasswd'
                },
                'selectdb'          => 'db',
                'query'             => %w[disk_io log_delay]
              }
            },
            queries: {
              'log_delay' => {
                'statement' => 'SELECT * FROM log_delay_repli;',
                'results'   => [{
                  'type'           => 'gauge',
                  'instanceprefix' => 'log_delay',
                  'instancesfrom'  => %w[inet_server_port inet_server_host],
                  'valuesfrom'     => 'log_delay'
                }]
              }
            }
          }
        end

        it "Will create #{options[:plugin_conf_dir]}/dbi-config.conf with Interval" do
          is_expected.to contain_concat__fragment('collectd_plugin_dbi_conf_db_mydatabase').with(content: %r{Driver \"mysql\"\n})
          is_expected.to contain_concat__fragment('collectd_plugin_dbi_conf_db_mydatabase').with(content: %r{Host \"localhost\"\n})
          is_expected.to contain_concat__fragment('collectd_plugin_dbi_conf_db_mydatabase').with(content: %r{Interval 60\n})
          is_expected.to contain_concat__fragment('collectd_plugin_dbi_conf_db_mydatabase').with(content: %r{Query \"disk_io\"\n})
          is_expected.to contain_concat__fragment('collectd_plugin_dbi_conf_db_mydatabase').with(content: %r{DriverOption \"host\" \"db\.example\.com\"\n})
          is_expected.to contain_concat__fragment('collectd_plugin_dbi_conf_db_mydatabase').with(content: %r{DriverOption \"username\" \"dbuser\"\n})
          is_expected.to contain_concat__fragment('collectd_plugin_dbi_conf_db_mydatabase').with(content: %r{DriverOption \"password\" \"dbpasswd\"\n})

          is_expected.to contain_concat__fragment('collectd_plugin_dbi_conf_query_log_delay').with(content: %r{Statement \"SELECT \* FROM log_delay_repli;\"\n})
          is_expected.to contain_concat__fragment('collectd_plugin_dbi_conf_query_log_delay').with(content: %r{<Result>\n})
          is_expected.to contain_concat__fragment('collectd_plugin_dbi_conf_query_log_delay').with(content: %r{InstancesFrom \"inet_server_port\" \"inet_server_host\"\n})
          is_expected.to contain_concat__fragment('collectd_plugin_dbi_conf_query_log_delay').with(content: %r{ValuesFrom \"log_delay\"\n})
        end
      end
    end
  end
end
