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

require 'spec_helper'

describe 'mysql::server::backup' do

    let(:default_params) {
        { 'backupuser'         => 'testuser',
            'backuppassword'     => 'testpass',
            'backupdir'          => '/tmp',
            'backuprotate'       => '25',
            'delete_before_dump' => true,
        }
    }
    context 'standard conditions' do
        let(:params) { default_params }

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

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

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

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

        it { should 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.
            should contain_file('mysqlbackup.sh').with_content(/.*ROTATE=24.*/)
        end
    end

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

        it { should 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 { should 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 { should 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 { should 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]
            content.should 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