# vim: tabstop=4 shiftwidth=4 softtabstop=4

# Copyright (c) 2012 NetApp, Inc.
# All Rights Reserved.
#
#    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 the NFS driver module"""

import __builtin__
import errno
import os

import mox as mox_lib
from mox import IgnoreArg
from mox import IsA
from mox import stubout

from nova import context
from nova import exception
from nova.exception import ProcessExecutionError
from nova import test

from nova.volume import nfs


class DumbVolume(object):
    fields = {}

    def __setitem__(self, key, value):
        self.fields[key] = value

    def __getitem__(self, item):
        return self.fields[item]


class NfsDriverTestCase(test.TestCase):
    """Test case for NFS driver"""

    TEST_NFS_EXPORT1 = 'nfs-host1:/export'
    TEST_NFS_EXPORT2 = 'nfs-host2:/export'
    TEST_SIZE_IN_GB = 1
    TEST_MNT_POINT = '/mnt/nfs'
    TEST_MNT_POINT_BASE = '/mnt/test'
    TEST_LOCAL_PATH = '/mnt/nfs/volume-123'
    TEST_FILE_NAME = 'test.txt'
    TEST_SHARES_CONFIG_FILE = '/etc/cinder/test-shares.conf'
    ONE_GB_IN_BYTES = 1024 * 1024 * 1024

    def setUp(self):
        self._driver = nfs.NfsDriver()
        super(NfsDriverTestCase, self).setUp()

    def stub_out_not_replaying(self, obj, attr_name):
        attr_to_replace = getattr(obj, attr_name)
        stub = mox_lib.MockObject(attr_to_replace)
        self.stubs.Set(obj, attr_name, stub)

    def test_path_exists_should_return_true(self):
        """_path_exists should return True if stat returns 0"""
        mox = self.mox
        drv = self._driver

        mox.StubOutWithMock(drv, '_execute')
        drv._execute('stat', self.TEST_FILE_NAME, run_as_root=True)

        mox.ReplayAll()

        self.assertTrue(drv._path_exists(self.TEST_FILE_NAME))

    def test_path_exists_should_return_false(self):
        """_path_exists should return True if stat doesn't return 0"""
        mox = self.mox
        drv = self._driver

        mox.StubOutWithMock(drv, '_execute')
        drv._execute('stat', self.TEST_FILE_NAME, run_as_root=True).\
            AndRaise(ProcessExecutionError(
            stderr="stat: cannot stat `test.txt': No such file or directory"))

        mox.ReplayAll()

        self.assertFalse(drv._path_exists(self.TEST_FILE_NAME))

    def test_local_path(self):
        """local_path common use case"""
        self.flags(nfs_mount_point_base=self.TEST_MNT_POINT_BASE)
        drv = self._driver

        volume = DumbVolume()
        volume['provider_location'] = self.TEST_NFS_EXPORT1
        volume['name'] = 'volume-123'

        self.assertEqual(
            '/mnt/test/2f4f60214cf43c595666dd815f0360a4/volume-123',
            drv.local_path(volume))

    def test_mount_nfs_should_mount_correctly(self):
        """_mount_nfs common case usage"""
        mox = self.mox
        drv = self._driver

        mox.StubOutWithMock(drv, '_path_exists')
        drv._path_exists(self.TEST_MNT_POINT).AndReturn(True)

        mox.StubOutWithMock(drv, '_execute')
        drv._execute('mount', '-t', 'nfs', self.TEST_NFS_EXPORT1,
                     self.TEST_MNT_POINT, run_as_root=True)

        mox.ReplayAll()

        drv._mount_nfs(self.TEST_NFS_EXPORT1, self.TEST_MNT_POINT)

    def test_mount_nfs_should_suppress_already_mounted_error(self):
        """_mount_nfs should suppress already mounted error if ensure=True
        """
        mox = self.mox
        drv = self._driver

        mox.StubOutWithMock(drv, '_path_exists')
        drv._path_exists(self.TEST_MNT_POINT).AndReturn(True)

        mox.StubOutWithMock(drv, '_execute')
        drv._execute('mount', '-t', 'nfs', self.TEST_NFS_EXPORT1,
                     self.TEST_MNT_POINT, run_as_root=True).\
            AndRaise(ProcessExecutionError(
                        stderr='is busy or already mounted'))

        mox.ReplayAll()

        drv._mount_nfs(self.TEST_NFS_EXPORT1, self.TEST_MNT_POINT, ensure=True)

    def test_mount_nfs_should_reraise_already_mounted_error(self):
        """_mount_nfs should not suppress already mounted error if ensure=False
        """
        mox = self.mox
        drv = self._driver

        mox.StubOutWithMock(drv, '_path_exists')
        drv._path_exists(self.TEST_MNT_POINT).AndReturn(True)

        mox.StubOutWithMock(drv, '_execute')
        drv._execute('mount', '-t', 'nfs', self.TEST_NFS_EXPORT1,
                     self.TEST_MNT_POINT, run_as_root=True).\
        AndRaise(ProcessExecutionError(stderr='is busy or already mounted'))

        mox.ReplayAll()

        self.assertRaises(ProcessExecutionError, drv._mount_nfs,
                          self.TEST_NFS_EXPORT1, self.TEST_MNT_POINT,
                          ensure=False)

    def test_mount_nfs_should_create_mountpoint_if_not_yet(self):
        """_mount_nfs should create mountpoint if it doesn't exist"""
        mox = self.mox
        drv = self._driver

        mox.StubOutWithMock(drv, '_path_exists')
        drv._path_exists(self.TEST_MNT_POINT).AndReturn(False)

        mox.StubOutWithMock(drv, '_execute')
        drv._execute('mkdir', '-p', self.TEST_MNT_POINT)
        drv._execute(*([IgnoreArg()] * 5), run_as_root=IgnoreArg())

        mox.ReplayAll()

        drv._mount_nfs(self.TEST_NFS_EXPORT1, self.TEST_MNT_POINT)

    def test_mount_nfs_should_not_create_mountpoint_if_already(self):
        """_mount_nfs should not create mountpoint if it already exists"""
        mox = self.mox
        drv = self._driver

        mox.StubOutWithMock(drv, '_path_exists')
        drv._path_exists(self.TEST_MNT_POINT).AndReturn(True)

        mox.StubOutWithMock(drv, '_execute')
        drv._execute(*([IgnoreArg()] * 5), run_as_root=IgnoreArg())

        mox.ReplayAll()

        drv._mount_nfs(self.TEST_NFS_EXPORT1, self.TEST_MNT_POINT)

    def test_get_hash_str(self):
        """_get_hash_str should calculation correct value"""
        drv = self._driver

        self.assertEqual('2f4f60214cf43c595666dd815f0360a4',
                         drv._get_hash_str(self.TEST_NFS_EXPORT1))

    def test_get_mount_point_for_share(self):
        """_get_mount_point_for_share should calculate correct value"""
        drv = self._driver

        self.flags(nfs_mount_point_base=self.TEST_MNT_POINT_BASE)

        self.assertEqual('/mnt/test/2f4f60214cf43c595666dd815f0360a4',
                         drv._get_mount_point_for_share(self.TEST_NFS_EXPORT1))

    def test_get_available_capacity_with_df(self):
        """_get_available_capacity should calculate correct value"""
        mox = self.mox
        drv = self._driver

        df_avail = 1490560
        df_head = 'Filesystem 1K-blocks Used Available Use% Mounted on\n'
        df_data = 'nfs-host:/export 2620544 996864 %d 41%% /mnt' % df_avail
        df_output = df_head + df_data

        self.flags(nfs_disk_util='df')

        mox.StubOutWithMock(drv, '_get_mount_point_for_share')
        drv._get_mount_point_for_share(self.TEST_NFS_EXPORT1).\
            AndReturn(self.TEST_MNT_POINT)

        mox.StubOutWithMock(drv, '_execute')
        drv._execute('df', '-P', '-B', '1', self.TEST_MNT_POINT,
                     run_as_root=True).AndReturn((df_output, None))

        mox.ReplayAll()

        self.assertEquals(df_avail,
                          drv._get_available_capacity(self.TEST_NFS_EXPORT1))

    def test_get_available_capacity_with_du(self):
        """_get_available_capacity should calculate correct value"""
        mox = self.mox
        drv = self._driver

        self.flags(nfs_disk_util='du')

        df_total_size = 2620544
        df_used_size = 996864
        df_avail_size = 1490560
        df_title = 'Filesystem 1-blocks Used Available Use% Mounted on\n'
        df_mnt_data = 'nfs-host:/export %d %d %d 41%% /mnt' % (df_total_size,
                                                               df_used_size,
                                                               df_avail_size)
        df_output = df_title + df_mnt_data

        du_used = 490560
        du_output = '%d /mnt' % du_used

        mox.StubOutWithMock(drv, '_get_mount_point_for_share')
        drv._get_mount_point_for_share(self.TEST_NFS_EXPORT1).\
            AndReturn(self.TEST_MNT_POINT)

        mox.StubOutWithMock(drv, '_execute')
        drv._execute('df', '-P', '-B', '1', self.TEST_MNT_POINT,
                     run_as_root=True).\
            AndReturn((df_output, None))
        drv._execute('du', '-sb', '--apparent-size',
                     '--exclude', '*snapshot*',
                     self.TEST_MNT_POINT,
                     run_as_root=True).AndReturn((du_output, None))

        mox.ReplayAll()

        self.assertEquals(df_total_size - du_used,
                          drv._get_available_capacity(self.TEST_NFS_EXPORT1))

    def test_load_shares_config(self):
        mox = self.mox
        drv = self._driver

        self.flags(nfs_shares_config=self.TEST_SHARES_CONFIG_FILE)

        mox.StubOutWithMock(__builtin__, 'open')
        config_data = []
        config_data.append(self.TEST_NFS_EXPORT1)
        config_data.append('#' + self.TEST_NFS_EXPORT2)
        config_data.append('')
        __builtin__.open(self.TEST_SHARES_CONFIG_FILE).AndReturn(config_data)
        mox.ReplayAll()

        shares = drv._load_shares_config()

        self.assertEqual([self.TEST_NFS_EXPORT1], shares)

    def test_ensure_share_mounted(self):
        """_ensure_share_mounted simple use case"""
        mox = self.mox
        drv = self._driver

        mox.StubOutWithMock(drv, '_get_mount_point_for_share')
        drv._get_mount_point_for_share(self.TEST_NFS_EXPORT1).\
            AndReturn(self.TEST_MNT_POINT)

        mox.StubOutWithMock(drv, '_mount_nfs')
        drv._mount_nfs(self.TEST_NFS_EXPORT1, self.TEST_MNT_POINT, ensure=True)

        mox.ReplayAll()

        drv._ensure_share_mounted(self.TEST_NFS_EXPORT1)

    def test_ensure_shares_mounted_should_save_mounting_successfully(self):
        """_ensure_shares_mounted should save share if mounted with success"""
        mox = self.mox
        drv = self._driver

        mox.StubOutWithMock(drv, '_load_shares_config')
        drv._load_shares_config().AndReturn([self.TEST_NFS_EXPORT1])
        mox.StubOutWithMock(drv, '_ensure_share_mounted')
        drv._ensure_share_mounted(self.TEST_NFS_EXPORT1)

        mox.ReplayAll()

        drv._ensure_shares_mounted()

        self.assertEqual(1, len(drv._mounted_shares))
        self.assertEqual(self.TEST_NFS_EXPORT1, drv._mounted_shares[0])

    def test_ensure_shares_mounted_should_not_save_mounting_with_error(self):
        """_ensure_shares_mounted should not save share if failed to mount"""
        mox = self.mox
        drv = self._driver

        mox.StubOutWithMock(drv, '_load_shares_config')
        drv._load_shares_config().AndReturn([self.TEST_NFS_EXPORT1])
        mox.StubOutWithMock(drv, '_ensure_share_mounted')
        drv._ensure_share_mounted(self.TEST_NFS_EXPORT1).AndRaise(Exception())

        mox.ReplayAll()

        drv._ensure_shares_mounted()

        self.assertEqual(0, len(drv._mounted_shares))

    def test_setup_should_throw_error_if_shares_config_not_configured(self):
        """do_setup should throw error if shares config is not configured """
        drv = self._driver

        self.flags(nfs_shares_config=self.TEST_SHARES_CONFIG_FILE)

        self.assertRaises(exception.NfsException,
                          drv.do_setup, IsA(context.RequestContext))

    def test_setup_should_throw_exception_if_nfs_client_is_not_installed(self):
        """do_setup should throw error if nfs client is not installed """
        mox = self.mox
        drv = self._driver

        self.flags(nfs_shares_config=self.TEST_SHARES_CONFIG_FILE)

        mox.StubOutWithMock(os.path, 'exists')
        os.path.exists(self.TEST_SHARES_CONFIG_FILE).AndReturn(True)
        mox.StubOutWithMock(drv, '_execute')
        drv._execute('mount.nfs', check_exit_code=False).\
            AndRaise(OSError(errno.ENOENT, 'No such file or directory'))

        mox.ReplayAll()

        self.assertRaises(exception.NfsException,
                          drv.do_setup, IsA(context.RequestContext))

    def test_find_share_should_throw_error_if_there_is_no_mounted_shares(self):
        """_find_share should throw error if there is no mounted shares"""
        drv = self._driver

        drv._mounted_shares = []

        self.assertRaises(exception.NotFound, drv._find_share,
                          self.TEST_SIZE_IN_GB)

    def test_find_share(self):
        """_find_share simple use case"""
        mox = self.mox
        drv = self._driver

        drv._mounted_shares = [self.TEST_NFS_EXPORT1, self.TEST_NFS_EXPORT2]

        mox.StubOutWithMock(drv, '_get_available_capacity')
        drv._get_available_capacity(self.TEST_NFS_EXPORT1).\
            AndReturn(2 * self.ONE_GB_IN_BYTES)
        drv._get_available_capacity(self.TEST_NFS_EXPORT2).\
            AndReturn(3 * self.ONE_GB_IN_BYTES)

        mox.ReplayAll()

        self.assertEqual(self.TEST_NFS_EXPORT2,
                         drv._find_share(self.TEST_SIZE_IN_GB))

    def test_find_share_should_throw_error_if_there_is_no_enough_place(self):
        """_find_share should throw error if there is no share to host vol"""
        mox = self.mox
        drv = self._driver

        drv._mounted_shares = [self.TEST_NFS_EXPORT1, self.TEST_NFS_EXPORT2]

        mox.StubOutWithMock(drv, '_get_available_capacity')
        drv._get_available_capacity(self.TEST_NFS_EXPORT1).\
            AndReturn(0)
        drv._get_available_capacity(self.TEST_NFS_EXPORT2).\
            AndReturn(0)

        mox.ReplayAll()

        self.assertRaises(exception.NfsNoSuitableShareFound, drv._find_share,
                          self.TEST_SIZE_IN_GB)

    def _simple_volume(self):
        volume = DumbVolume()
        volume['provider_location'] = '127.0.0.1:/mnt'
        volume['name'] = 'volume_name'
        volume['size'] = 10

        return volume

    def test_create_sparsed_volume(self):
        mox = self.mox
        drv = self._driver
        volume = self._simple_volume()

        self.flags(nfs_sparsed_volumes=True)

        mox.StubOutWithMock(drv, '_create_sparsed_file')
        mox.StubOutWithMock(drv, '_set_rw_permissions_for_all')

        drv._create_sparsed_file(IgnoreArg(), IgnoreArg())
        drv._set_rw_permissions_for_all(IgnoreArg())

        mox.ReplayAll()

        drv._do_create_volume(volume)

    def test_create_nonsparsed_volume(self):
        mox = self.mox
        drv = self._driver
        volume = self._simple_volume()

        self.flags(nfs_sparsed_volumes=False)

        mox.StubOutWithMock(drv, '_create_regular_file')
        mox.StubOutWithMock(drv, '_set_rw_permissions_for_all')

        drv._create_regular_file(IgnoreArg(), IgnoreArg())
        drv._set_rw_permissions_for_all(IgnoreArg())

        mox.ReplayAll()

        drv._do_create_volume(volume)

    def test_create_volume_should_ensure_nfs_mounted(self):
        """create_volume should ensure shares provided in config are mounted"""
        mox = self.mox
        drv = self._driver

        self.stub_out_not_replaying(nfs, 'LOG')
        self.stub_out_not_replaying(drv, '_find_share')
        self.stub_out_not_replaying(drv, '_do_create_volume')

        mox.StubOutWithMock(drv, '_ensure_shares_mounted')
        drv._ensure_shares_mounted()

        mox.ReplayAll()

        volume = DumbVolume()
        volume['size'] = self.TEST_SIZE_IN_GB
        drv.create_volume(volume)

    def test_create_volume_should_return_provider_location(self):
        """create_volume should return provider_location with found share """
        mox = self.mox
        drv = self._driver

        self.stub_out_not_replaying(nfs, 'LOG')
        self.stub_out_not_replaying(drv, '_ensure_shares_mounted')
        self.stub_out_not_replaying(drv, '_do_create_volume')

        mox.StubOutWithMock(drv, '_find_share')
        drv._find_share(self.TEST_SIZE_IN_GB).AndReturn(self.TEST_NFS_EXPORT1)

        mox.ReplayAll()

        volume = DumbVolume()
        volume['size'] = self.TEST_SIZE_IN_GB
        result = drv.create_volume(volume)
        self.assertEqual(self.TEST_NFS_EXPORT1, result['provider_location'])

    def test_delete_volume(self):
        """delete_volume simple test case"""
        mox = self.mox
        drv = self._driver

        self.stub_out_not_replaying(drv, '_ensure_share_mounted')

        volume = DumbVolume()
        volume['name'] = 'volume-123'
        volume['provider_location'] = self.TEST_NFS_EXPORT1

        mox.StubOutWithMock(drv, 'local_path')
        drv.local_path(volume).AndReturn(self.TEST_LOCAL_PATH)

        mox.StubOutWithMock(drv, '_path_exists')
        drv._path_exists(self.TEST_LOCAL_PATH).AndReturn(True)

        mox.StubOutWithMock(drv, '_execute')
        drv._execute('rm', '-f', self.TEST_LOCAL_PATH, run_as_root=True)

        mox.ReplayAll()

        drv.delete_volume(volume)

    def test_delete_should_ensure_share_mounted(self):
        """delete_volume should ensure that corresponding share is mounted"""
        mox = self.mox
        drv = self._driver

        self.stub_out_not_replaying(drv, '_execute')

        volume = DumbVolume()
        volume['name'] = 'volume-123'
        volume['provider_location'] = self.TEST_NFS_EXPORT1

        mox.StubOutWithMock(drv, '_ensure_share_mounted')
        drv._ensure_share_mounted(self.TEST_NFS_EXPORT1)

        mox.ReplayAll()

        drv.delete_volume(volume)

    def test_delete_should_not_delete_if_provider_location_not_provided(self):
        """delete_volume shouldn't try to delete if provider_location missed"""
        mox = self.mox
        drv = self._driver

        self.stub_out_not_replaying(drv, '_ensure_share_mounted')

        volume = DumbVolume()
        volume['name'] = 'volume-123'
        volume['provider_location'] = None

        mox.StubOutWithMock(drv, '_execute')

        mox.ReplayAll()

        drv.delete_volume(volume)

    def test_delete_should_not_delete_if_there_is_no_file(self):
        """delete_volume should not try to delete if file missed"""
        mox = self.mox
        drv = self._driver

        self.stub_out_not_replaying(drv, '_ensure_share_mounted')

        volume = DumbVolume()
        volume['name'] = 'volume-123'
        volume['provider_location'] = self.TEST_NFS_EXPORT1

        mox.StubOutWithMock(drv, 'local_path')
        drv.local_path(volume).AndReturn(self.TEST_LOCAL_PATH)

        mox.StubOutWithMock(drv, '_path_exists')
        drv._path_exists(self.TEST_LOCAL_PATH).AndReturn(False)

        mox.StubOutWithMock(drv, '_execute')

        mox.ReplayAll()

        drv.delete_volume(volume)