module-puppetlabs-mysql/spec/classes/mysql_server_backup_spec.rb

require 'spec_helper'

describe 'mysql::server::backup' do
  on_pe_supported_platforms(PLATFORMS).each do |pe_version,pe_platforms|
    pe_platforms.each do |pe_platform,facts|
      describe "on #{pe_version} #{pe_platform}" do
        let(:facts) { facts }

        let(:default_params) {
          { 'backupuser'         => 'testuser',
            'backuppassword'     => 'testpass',
            'backupdir'          => '/tmp',
            'backuprotate'       => '25',
            'delete_before_dump' => true,
            'execpath'           => '/usr/bin:/usr/sbin:/bin:/sbin:/opt/zimbra/bin',
          }
        }

        context 'standard conditions' do
          let(:params) { default_params }

          it { is_expected.to contain_mysql_user('testuser@localhost').with(
            :require => 'Class[Mysql::Server::Root_password]'
          )}

          it { is_expected.to contain_mysql_grant('testuser@localhost/*.*').with(
            :privileges => ["SELECT", "RELOAD", "LOCK TABLES", "SHOW VIEW", "PROCESS"]
          )}

          it { is_expected.to contain_cron('mysql-backup').with(
            :command => '/usr/local/sbin/mysqlbackup.sh',
            :ensure  => 'present'
          )}

          it { is_expected.to contain_file('mysqlbackup.sh').with(
            :path   => '/usr/local/sbin/mysqlbackup.sh',
            :ensure => 'present'
          ) }

          it { is_expected.to contain_file('mysqlbackupdir').with(
            :path   => '/tmp',
            :ensure => 'directory'
          )}

          it 'should have compression by default' do
            verify_contents(subject, 'mysqlbackup.sh', [
              ' --all-databases | bzcat -zc > ${DIR}/${PREFIX}`date +%Y%m%d-%H%M%S`.sql.bz2',
            ])
          end
          it 'should skip backing up events table by default' do
            verify_contents(subject, 'mysqlbackup.sh', [
              'EVENTS="--ignore-table=mysql.event"',
            ])
          end

          it 'should have 25 days of rotation' do
            # MySQL counts from 0 I guess.
            is_expected.to contain_file('mysqlbackup.sh').with_content(/.*ROTATE=24.*/)
          end

          it 'should have a standard PATH' do
            is_expected.to contain_file('mysqlbackup.sh').with_content(%r{PATH=/usr/bin:/usr/sbin:/bin:/sbin:/opt/zimbra/bin})
          end
        end

        context 'custom ownership and mode for backupdir' do
          let(:params) do
            { :backupdirmode => '0750',
              :backupdirowner => 'testuser',
              :backupdirgroup => 'testgrp',
            }.merge(default_params)
          end

          it { is_expected.to contain_file('mysqlbackupdir').with(
            :path => '/tmp',
            :ensure => 'directory',
            :mode => '0750',
            :owner => 'testuser',
            :group => 'testgrp'
          ) }
        end

        context 'with compression disabled' do
          let(:params) do
            { :backupcompress => false }.merge(default_params)
          end

          it { is_expected.to contain_file('mysqlbackup.sh').with(
            :path   => '/usr/local/sbin/mysqlbackup.sh',
            :ensure => 'present'
          ) }

          it 'should be able to disable compression' do
            verify_contents(subject, 'mysqlbackup.sh', [
              ' --all-databases > ${DIR}/${PREFIX}`date +%Y%m%d-%H%M%S`.sql',
            ])
          end
        end

        context 'with mysql.events backedup' do
          let(:params) do
            { :ignore_events => false }.merge(default_params)
          end

          it { is_expected.to contain_file('mysqlbackup.sh').with(
            :path   => '/usr/local/sbin/mysqlbackup.sh',
            :ensure => 'present'
          ) }

          it 'should be able to backup events table' do
            verify_contents(subject, 'mysqlbackup.sh', [
              'EVENTS="--events"',
            ])
          end
        end


        context 'with database list specified' do
          let(:params) do
            { :backupdatabases => ['mysql'] }.merge(default_params)
          end

          it { is_expected.to contain_file('mysqlbackup.sh').with(
            :path   => '/usr/local/sbin/mysqlbackup.sh',
            :ensure => 'present'
          ) }

          it 'should have a backup file for each database' do
            content = subject.resource('file','mysqlbackup.sh').send(:parameters)[:content]
            expect(content).to match(' mysql | bzcat -zc \${DIR}\\\${PREFIX}mysql_`date')
            #      verify_contents(subject, 'mysqlbackup.sh', [
            #        ' mysql | bzcat -zc ${DIR}/${PREFIX}mysql_`date +%Y%m%d-%H%M%S`.sql',
            #      ])
          end 
        end

        context 'with file per database' do
          let(:params) do
            default_params.merge({ :file_per_database => true })
          end

          it 'should loop through backup all databases' do
            verify_contents(subject, 'mysqlbackup.sh', [
              'mysql -s -r -N -e \'SHOW DATABASES\' | while read dbname',
              'do',
              '  mysqldump -u${USER} -p${PASS} --opt --flush-logs --single-transaction \\',
              '    ${EVENTS} \\',
              '    ${dbname} | bzcat -zc > ${DIR}/${PREFIX}${dbname}_`date +%Y%m%d-%H%M%S`.sql.bz2',
              'done',
            ])
          end

          context 'with compression disabled' do
            let(:params) do
              default_params.merge({ :file_per_database => true, :backupcompress => false })
            end

            it 'should loop through backup all databases without compression' do
              verify_contents(subject, 'mysqlbackup.sh', [
                '    ${dbname} > ${DIR}/${PREFIX}${dbname}_`date +%Y%m%d-%H%M%S`.sql',
              ])
            end
          end
        end

        context 'with postscript' do
          let(:params) do
            default_params.merge({ :postscript => 'rsync -a /tmp backup01.local-lan:' })
          end

          it 'should be add postscript' do
            verify_contents(subject, 'mysqlbackup.sh', [
              'rsync -a /tmp backup01.local-lan:',
            ])
          end
        end

        context 'with postscripts' do
          let(:params) do
            default_params.merge({ :postscript => [
              'rsync -a /tmp backup01.local-lan:',
              'rsync -a /tmp backup02.local-lan:',
            ]})
          end

          it 'should be add postscript' do
            verify_contents(subject, 'mysqlbackup.sh', [
              'rsync -a /tmp backup01.local-lan:',
              'rsync -a /tmp backup02.local-lan:',
            ])
          end
        end
      end
    end
  end
end