#
# 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::volume::storage class
#

require 'spec_helper'

describe 'cloud::volume::storage' do

  shared_examples_for 'openstack volume storage' do

    let :pre_condition do
      "class { 'cloud::volume':
        cinder_db_host             => '10.0.0.1',
        cinder_db_user             => 'cinder',
        cinder_db_password         => 'secret',
        cinder_db_idle_timeout     => 5000,
        rabbit_hosts               => ['10.0.0.1'],
        rabbit_password            => 'secret',
        verbose                    => true,
        debug                      => true,
        log_facility               => 'LOG_LOCAL0',
        storage_availability_zone  => 'nova',
        use_syslog                 => true }"
    end

    let :params do
      { :cinder_rbd_pool            => 'ceph_cinder',
        :cinder_rbd_user            => 'cinder',
        :cinder_rbd_secret_uuid     => 'secret',
        :cinder_rbd_max_clone_depth => '10',
        :cinder_backends            => {
          'rbd' => {
            'lowcost' => {
              'rbd_pool'               => 'ceph_cinder',
              'rbd_user'               => 'cinder',
              'rbd_secret_uuid'        => 'secret'
            }
          },
          'netapp' => {
            'premium' => {
              'netapp_server_hostname' => 'netapp-server.host',
              'netapp_login'           => 'joe',
              'netapp_password'        => 'secret',
	      'nfs_mount_options'      => 'rw,proto=tcp,sec=sys',
            }
          },
          'iscsi' => {
            'fast' => {
              'iscsi_ip_address' => '10.0.0.1',
              'volume_group'     => 'fast-vol'
            }
          },
          'emc_vnx' => {
            'very-fast' => {
              'iscsi_ip_address'      => '10.0.0.1',
              'san_ip'                => '10.0.0.2',
              'san_password'          => 'secrete',
              'storage_vnx_pool_name' => 'emc-volumes',
            }
          },
          'eqlx' => {
            'dell' => {
              'san_ip'          => '10.0.0.1',
              'san_login'       => 'admin',
              'san_password'    => 'secrete',
              'eqlx_group_name' => 'dell-volumes',
            }
          },
          'glusterfs' => {
            'gluster' => {
              'glusterfs_shares'        => ['/mnt/share'],
              'glusterfs_shares_config' => '/etc/cinder/shares-gluster.conf',
            }
          },
          'nfs' => {
            'freenas' => {
              'nfs_servers'          => ['10.0.0.1:/myshare'],
              'nfs_mount_options'    => 'defaults',
              'nfs_disk_util'        => 'df',
              'nfs_mount_point_base' => '/mnt/shares',
              'nfs_shares_config'    => '/etc/cinder/shares.conf',
              'nfs_used_ratio'       => '0.6',
              'nfs_oversub_ratio'    => '1.0'
            }
          }
        },
        :ks_keystone_internal_proto => 'http',
        :ks_keystone_internal_port  => '5000',
        :ks_keystone_internal_host  => 'keystone.host',
        :ks_cinder_password         => 'secret' }
    end

    it 'configure cinder common' do
      is_expected.to contain_class('cinder').with(
          :database_connection       => 'mysql://cinder:secret@10.0.0.1/cinder?charset=utf8',
          :database_idle_timeout     => '5000',
          :rabbit_userid             => 'cinder',
          :rabbit_hosts              => ['10.0.0.1'],
          :rabbit_password           => 'secret',
          :rabbit_virtual_host       => '/',
          :verbose                   => true,
          :debug                     => true,
          :log_facility              => 'LOG_LOCAL0',
          :log_dir                   => false,
          :use_syslog                => true,
          :storage_availability_zone => 'nova'
        )

      is_expected.to contain_cinder_config('DEFAULT/notification_driver').with('value' => 'cinder.openstack.common.notifier.rpc_notifier')

    end

    it 'checks if Cinder DB is populated' do
      is_expected.to contain_exec('cinder_db_sync').with(
        :command => 'cinder-manage db sync',
        :user    => 'cinder',
        :path    => '/usr/bin',
        :unless  => '/usr/bin/mysql cinder -h 10.0.0.1 -u cinder -psecret -e "show tables" | /bin/grep Tables'
      )
    end

    it 'configure cinder volume service' do
      is_expected.to contain_class('cinder::volume')
    end

    context 'with RBD backend' do
      it 'configures rbd volume driver' do
        is_expected.to contain_cinder_config('lowcost/volume_backend_name').with_value('lowcost')
        is_expected.to contain_cinder_config('lowcost/rbd_pool').with_value('ceph_cinder')
        is_expected.to contain_cinder_config('lowcost/rbd_user').with_value('cinder')
        is_expected.to contain_cinder_config('lowcost/rbd_secret_uuid').with_value('secret')
        is_expected.to contain_cinder_config('lowcost/volume_tmp_dir').with_value('/tmp')
        is_expected.to contain_cinder__type('lowcost').with(
          :set_key        => 'volume_backend_name',
          :set_value      => 'lowcost',
          :os_tenant_name => 'services',
          :os_username    => 'cinder',
          :os_password    => 'secret',
          :os_auth_url    => 'http://keystone.host:5000/v2.0'
        )
        is_expected.to contain_group('cephkeyring').with(:ensure => 'present')
        is_expected.to contain_exec('add-cinder-to-group').with(
          :command => 'usermod -a -G cephkeyring cinder',
          :path    => ['/usr/sbin', '/usr/bin', '/bin', '/sbin'],
          :unless  => 'groups cinder | grep cephkeyring'
        )
        is_expected.to contain_file('/etc/ceph/ceph.client.cinder.keyring').with({
          'owner'  => 'root',
          'group'  => 'cephkeyring',
          'mode'   => '0440',
        })
      end
    end

    context 'with NetApp backend' do
      it 'configures netapp volume driver' do
        is_expected.to contain_cinder_config('premium/volume_backend_name').with_value('premium')
        is_expected.to contain_cinder_config('premium/netapp_login').with_value('joe')
        is_expected.to contain_cinder_config('premium/netapp_password').with_value('secret')
        is_expected.to contain_cinder_config('premium/netapp_server_hostname').with_value('netapp-server.host')
        is_expected.to contain_cinder_config('premium/nfs_mount_options').with_value('rw,proto=tcp,sec=sys')
        is_expected.to contain_cinder__type('premium').with(
          :set_key   => 'volume_backend_name',
          :set_value => 'premium',
          :notify    => 'Service[cinder-volume]'
        )
      end
    end

    context 'with iSCSI backend' do
      it 'configures iSCSI volume driver' do
        is_expected.to contain_cinder_config('fast/volume_backend_name').with_value('fast')
        is_expected.to contain_cinder_config('fast/iscsi_ip_address').with_value('10.0.0.1')
        is_expected.to contain_cinder_config('fast/volume_group').with_value('fast-vol')
        is_expected.to contain_cinder__type('fast').with(
          :set_key   => 'volume_backend_name',
          :set_value => 'fast',
          :notify    => 'Service[cinder-volume]'
        )
      end
    end

    context 'with EMC VNX backend' do
      it 'configures EMC VNX volume driver' do
        should contain_cinder_config('very-fast/volume_backend_name').with_value('very-fast')
        should contain_cinder_config('very-fast/iscsi_ip_address').with_value('10.0.0.1')
        should contain_cinder_config('very-fast/san_ip').with_value('10.0.0.2')
        should contain_cinder_config('very-fast/san_password').with_value('secrete')
        should contain_cinder_config('very-fast/storage_vnx_pool_name').with_value('emc-volumes')
        should contain_cinder__type('very-fast').with(
          :set_key   => 'volume_backend_name',
          :set_value => 'very-fast',
          :notify    => 'Service[cinder-volume]'
        )
      end
    end

    context 'with EQLX backend' do
      it 'configures EQLX volume driver' do
        should contain_cinder_config('dell/volume_backend_name').with_value('dell')
        should contain_cinder_config('dell/san_ip').with_value('10.0.0.1')
        should contain_cinder_config('dell/san_login').with_value('admin')
        should contain_cinder_config('dell/san_password').with_value('secrete')
        should contain_cinder_config('dell/eqlx_group_name').with_value('dell-volumes')
        should contain_cinder__type('dell').with(
          :set_key   => 'volume_backend_name',
          :set_value => 'dell',
          :notify    => 'Service[cinder-volume]'
        )
      end
    end

    context 'with GlusterFS backend' do
      it 'configures GlusterFS volume driver' do
        should contain_cinder_config('gluster/volume_backend_name').with_value('gluster')
        should contain_cinder_config('gluster/glusterfs_shares_config').with_value('/etc/cinder/shares-gluster.conf')
        should contain_cinder__type('gluster').with(
          :set_key   => 'volume_backend_name',
          :set_value => 'gluster',
          :notify    => 'Service[cinder-volume]'
        )
      end
    end

    context 'with NFS backend' do
      it 'configures NFS volume driver' do
        is_expected.to contain_cinder_config('freenas/volume_backend_name').with_value('freenas')
        is_expected.to contain_cinder_config('freenas/nfs_mount_options').with_value('defaults')
        is_expected.to contain_cinder_config('freenas/nfs_mount_point_base').with_value('/mnt/shares')
        is_expected.to contain_cinder_config('freenas/nfs_disk_util').with_value('df')
        is_expected.to contain_cinder_config('freenas/nfs_shares_config').with_value('/etc/cinder/shares.conf')
        is_expected.to contain_cinder_config('freenas/nfs_used_ratio').with_value('0.6')
        is_expected.to contain_cinder_config('freenas/nfs_oversub_ratio').with_value('1.0')
        is_expected.to contain_cinder__type('freenas').with(
          :set_key   => 'volume_backend_name',
          :set_value => 'freenas',
          :notify    => 'Service[cinder-volume]'
        )
        should contain_file('/etc/cinder/shares.conf').with_content(/^10.0.0.1:\/myshare$/)
      end
    end

    context 'with two RBD backends' do
      before :each do
        params.merge!(
          :cinder_backends => {
            'rbd' => {
              'lowcost' => {
                'rbd_pool'        => 'low',
                'rbd_user'        => 'cinder',
                'rbd_secret_uuid' => 'secret',
              },
              'normal' => {
                'rbd_pool'         => 'normal',
                'rbd_user'         => 'cinder',
                'rbd_secret_uuid'  => 'secret',
              }
            }
          }
        )
      end

      it 'configures two rbd volume backends' do
        is_expected.to contain_cinder_config('lowcost/volume_backend_name').with_value('lowcost')
        is_expected.to contain_cinder_config('lowcost/rbd_pool').with_value('low')
        is_expected.to contain_cinder_config('lowcost/rbd_user').with_value('cinder')
        is_expected.to contain_cinder_config('lowcost/rbd_secret_uuid').with_value('secret')
        is_expected.to contain_cinder_config('lowcost/volume_tmp_dir').with_value('/tmp')
        is_expected.to contain_cinder__type('lowcost').with(
          :set_key        => 'volume_backend_name',
          :set_value      => 'lowcost',
          :os_tenant_name => 'services',
          :os_username    => 'cinder',
          :os_password    => 'secret',
          :os_auth_url    => 'http://keystone.host:5000/v2.0'
        )
        is_expected.to contain_cinder_config('normal/volume_backend_name').with_value('normal')
        is_expected.to contain_cinder_config('normal/rbd_pool').with_value('normal')
        is_expected.to contain_cinder_config('normal/rbd_user').with_value('cinder')
        is_expected.to contain_cinder_config('normal/rbd_secret_uuid').with_value('secret')
        is_expected.to contain_cinder_config('normal/volume_tmp_dir').with_value('/tmp')
        is_expected.to contain_cinder__type('normal').with(
          :set_key        => 'volume_backend_name',
          :set_value      => 'normal',
          :os_tenant_name => 'services',
          :os_username    => 'cinder',
          :os_password    => 'secret',
          :os_auth_url    => 'http://keystone.host:5000/v2.0'
        )
      end
    end

    context 'with all backends enabled' do
      it 'configure all cinder backends' do
        is_expected.to contain_class('cinder::backends').with(
          :enabled_backends => ['lowcost', 'premium', 'fast', 'very-fast', 'dell', 'freenas', 'gluster']
        )
      end
    end

  end

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

    it_configures 'openstack volume storage'
  end

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

    it_configures 'openstack volume storage'
  end

end