test_disks.py

#
# Project Kimchi
#
# Copyright IBM Corp, 2017
#
# Code derived from Ginger Base
#
# This library is free software; you can redistribute it and/or
# modify it under the terms of the GNU Lesser General Public
# License as published by the Free Software Foundation; either
# version 2.1 of the License, or (at your option) any later version.
#
# This library is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
# Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public
# License along with this library; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301 USA
import unittest

import mock
from wok.exception import NotFoundError
from wok.exception import OperationFailed
from wok.plugins.kimchi.disks import _get_lsblk_devs


class DiskTests(unittest.TestCase):

    @mock.patch('wok.plugins.kimchi.disks.run_command')
    def test_lsblk_returns_404_when_device_not_found(self, mock_run_command):
        mock_run_command.return_value = ['', 'not a block device', 32]
        fake_dev = '/not/a/true/block/dev'
        keys = ['MOUNTPOINT']

        with self.assertRaises(NotFoundError):
            _get_lsblk_devs(keys, [fake_dev])
            cmd = ['lsblk', '-Pbo', 'MOUNTPOINT', fake_dev]
            mock_run_command.assert_called_once_with(cmd)

    @mock.patch('wok.plugins.kimchi.disks.run_command')
    def test_lsblk_returns_500_when_unknown_error_occurs(
            self, mock_run_command):

        mock_run_command.return_value = ['', '', 1]
        valid_dev = '/valid/block/dev'
        keys = ['MOUNTPOINT']

        with self.assertRaises(OperationFailed):
            _get_lsblk_devs(keys, [valid_dev])
            cmd = ['lsblk', '-Pbo', 'MOUNTPOINT', valid_dev]
            mock_run_command.assert_called_once_with(cmd)