#
# Copyright (C) 2014 eNovance SAS <licensing@enovance.com>
#
# Licensed under the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License. You may obtain
# a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.
#
# Unit tests for cloud::spof class
#

require 'spec_helper'

describe 'cloud::spof' do

  shared_examples_for 'cloud spof' do

    let :params do
      { :cluster_ip        => '10.0.0.1',
        :multicast_address => '239.1.1.2',
        :cluster_members   => false,
        :cluster_password  => 'verysecrete' }
    end

    context 'with Pacemaker on Debian' do
      before :each do
        facts.merge!( :osfamily => 'Debian' )
      end

      it 'configure pacemaker/corosync' do
        is_expected.to contain_class('corosync').with(
          :enable_secauth    => false,
          :authkey           => '/var/lib/puppet/ssl/certs/ca.pem',
          :bind_address      => '10.0.0.1',
          :multicast_address => '239.1.1.2',
        )
        is_expected.to contain_file('/usr/lib/ocf/resource.d/heartbeat/ceilometer-agent-central').with(
          :source => 'puppet:///modules/cloud/heartbeat/ceilometer-agent-central',
          :mode   => '0755',
          :owner  => 'root',
          :group  => 'root'
        )
        is_expected.to contain_class('cloud::telemetry::centralagent').with(:enabled => false)
        is_expected.to contain_exec('cleanup_ceilometer_agent_central').with(
          :command => 'crm resource cleanup ceilometer-agent-central',
          :path    => ['/usr/sbin', '/bin'],
          :user    => 'root',
          :unless  => 'crm resource show ceilometer-agent-central | grep Started'
        )
      end
    end

    context 'with Pacemaker on Red-Hat' do
      before :each do
        facts.merge!( :osfamily => 'RedHat' )
        params.merge!( :cluster_members => 'srv1 srv2 srv3')
      end

      it 'configure pacemaker/corosync' do
        is_expected.to contain_class('pacemaker').with(:hacluster_pwd => 'verysecrete')
        is_expected.to contain_class('pacemaker::stonith').with(:disable => true)
        is_expected.to contain_class('pacemaker::corosync').with(
          :cluster_name     => 'openstack',
          :settle_timeout   => 10,
          :settle_tries     => 2,
          :settle_try_sleep => 5,
          :manage_fw        => false,
          :cluster_members  => 'srv1 srv2 srv3')
        is_expected.to contain_pcmk_resource('openstack-ceilometer-central')
        is_expected.to contain_class('cloud::telemetry::centralagent').with(:enabled => false)
      end
    end

    context 'with Pacemaker on Red-Hat with missing parameters' do
      before :each do
        facts.merge!( :osfamily => 'RedHat' )
        params.merge!( :cluster_members => false)
      end
      it { is_expected.to compile.and_raise_error(/cluster_members is a required parameter./) }
    end

    context 'with default firewall enabled' do
      let :pre_condition do
        "class { 'cloud': manage_firewall => true }"
      end
      before :each do
        params.merge!( :cluster_members => 'srv1 srv2 srv3')
      end
      it 'configure pacemaker firewall rules' do
        is_expected.to contain_firewall('100 allow vrrp access').with(
          :port   => nil,
          :proto  => 'vrrp',
          :action => 'accept',
        )
        is_expected.to contain_firewall('100 allow corosync tcp access').with(
          :port   => ['2224','3121','21064'],
          :action => 'accept',
        )
        is_expected.to contain_firewall('100 allow corosync udp access').with(
          :port   => ['5404','5405'],
          :proto  => 'udp',
          :action => 'accept',
        )
      end
    end

    context 'with custom firewall enabled' do
      let :pre_condition do
        "class { 'cloud': manage_firewall => true }"
      end
      before :each do
        params.merge!(
          :firewall_settings => { 'limit' => '50/sec' },
          :cluster_members   => 'srv1 srv2 srv3'
        )
      end
      it 'configure pacemaker firewall rules with custom parameter' do
        is_expected.to contain_firewall('100 allow vrrp access').with(
          :port   => nil,
          :proto  => 'vrrp',
          :action => 'accept',
          :limit  => '50/sec',
        )
        is_expected.to contain_firewall('100 allow corosync tcp access').with(
          :port   => ['2224','3121','21064'],
          :action => 'accept',
          :limit  => '50/sec',
        )
        is_expected.to contain_firewall('100 allow corosync udp access').with(
          :port   => ['5404','5405'],
          :proto  => 'udp',
          :action => 'accept',
          :limit  => '50/sec',
        )
      end
    end

  end

  context 'on Debian platforms' do
    let :facts do
      { :osfamily => 'Debian' }
    end

    it_configures 'cloud spof'
  end

  context 'on RedHat platforms' do
    let :facts do
      { :osfamily => 'RedHat' }
    end
    it_configures 'cloud spof'
  end

end