summaryrefslogtreecommitdiffstats
path: root/puppet-barometer/spec/classes/barometer_db_spec.rb
blob: c60666bb5a176ba12990806274b5eaa65201ad9d (plain)
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 'barometer::db' do

  shared_examples 'barometer::db' do
    context 'with default parameters' do
      it { is_expected.to contain_barometer_config('database/connection').with_value('sqlite:////var/lib/barometer/barometer.sqlite') }
      it { is_expected.to contain_barometer_config('database/idle_timeout').with_value('<SERVICE DEFAULT>') }
      it { is_expected.to contain_barometer_config('database/min_pool_size').with_value('<SERVICE DEFAULT>') }
      it { is_expected.to contain_barometer_config('database/db_max_retries').with_value('<SERVICE DEFAULT>') }
      it { is_expected.to contain_barometer_config('database/max_retries').with_value('<SERVICE DEFAULT>') }
      it { is_expected.to contain_barometer_config('database/retry_interval').with_value('<SERVICE DEFAULT>') }
      it { is_expected.to contain_barometer_config('database/max_pool_size').with_value('<SERVICE DEFAULT>') }
      it { is_expected.to contain_barometer_config('database/max_overflow').with_value('<SERVICE DEFAULT>') }
    end

    context 'with specific parameters' do
      let :params do
        { :database_connection     => 'mysql+pymysql://barometer:barometer@localhost/barometer',
          :database_idle_timeout   => '3601',
          :database_min_pool_size  => '2',
          :database_db_max_retries => '-1',
          :database_max_retries    => '11',
          :database_retry_interval => '11',
          :database_max_pool_size  => '11',
          :database_max_overflow   => '21',
        }
      end

      it { is_expected.to contain_barometer_config('database/connection').with_value('mysql+pymysql://barometer:barometer@localhost/barometer') }
      it { is_expected.to contain_barometer_config('database/idle_timeout').with_value('3601') }
      it { is_expected.to contain_barometer_config('database/min_pool_size').with_value('2') }
      it { is_expected.to contain_barometer_config('database/db_max_retries').with_value('-1') }
      it { is_expected.to contain_barometer_config('database/max_retries').with_value('11') }
      it { is_expected.to contain_barometer_config('database/retry_interval').with_value('11') }
      it { is_expected.to contain_barometer_config('database/max_pool_size').with_value('11') }
      it { is_expected.to contain_barometer_config('database/max_overflow').with_value('21') }
    end

    context 'with postgresql backend' do
      let :params do
        { :database_connection     => 'postgresql://barometer:barometer@localhost/barometer', }
      end

      it 'install the proper backend package' do
        is_expected.to contain_package('python-psycopg2').with(:ensure => 'present')
      end

    end

    context 'with MySQL-python library as backend package' do
      let :params do
        { :database_connection     => 'mysql://barometer:barometer@localhost/barometer', }
      end

      it { is_expected.to contain_package('python-mysqldb').with(:ensure => 'present') }
    end

    context 'with incorrect database_connection string' do
      let :params do
        { :database_connection     => 'foodb://barometer:barometer@localhost/barometer', }
      end

      it_raises 'a Puppet::Error', /validate_re/
    end

    context 'with incorrect pymysql database_connection string' do
      let :params do
        { :database_connection     => 'foo+pymysql://barometer:barometer@localhost/barometer', }
      end

      it_raises 'a Puppet::Error', /validate_re/
    end

  end

  shared_examples_for 'barometer::db on Debian' do
    context 'using pymysql driver' do
      let :params do
        { :database_connection     => 'mysql+pymysql://barometer:barometer@localhost/barometer', }
      end

      it 'install the proper backend package' do
        is_expected.to contain_package('db_backend_package').with(
          :ensure => 'present',
          :name   => 'python-pymysql',
          :tag    => 'openstack'
        )
      end
    end
  end

  shared_examples_for 'barometer::db on RedHat' do
    context 'using pymysql driver' do
      let :params do
        { :database_connection     => 'mysql+pymysql://barometer:barometer@localhost/barometer', }
      end

      it 'install the proper backend package' do
        is_expected.not_to contain_package('db_backend_package')
      end
    end
  end

  on_supported_os({
    :supported_os   => OSDefaults.get_supported_os
  }).each do |os,facts|
    context "on #{os}" do
      let (:facts) do
        facts.merge!(OSDefaults.get_facts())
      end

      it_configures 'barometer::db'
      it_configures "barometer::db on #{facts[:osfamily]}"
    end
  end
end