|
1 | 1 | import os |
2 | 2 | import pytest |
3 | 3 | from ceph_volume.util import disk |
4 | | -from mock.mock import patch |
| 4 | +from mock.mock import patch, MagicMock |
| 5 | + |
| 6 | + |
| 7 | +class TestFunctions: |
| 8 | + @patch('ceph_volume.util.disk.os.path.exists', MagicMock(return_value=False)) |
| 9 | + def test_is_device_path_does_not_exist(self): |
| 10 | + assert not disk.is_device('/dev/foo') |
| 11 | + |
| 12 | + @patch('ceph_volume.util.disk.os.path.exists', MagicMock(return_value=True)) |
| 13 | + def test_is_device_dev_doesnt_startswith_dev(self): |
| 14 | + assert not disk.is_device('/foo') |
| 15 | + |
| 16 | + @patch('ceph_volume.util.disk.allow_loop_devices', MagicMock(return_value=False)) |
| 17 | + @patch('ceph_volume.util.disk.os.path.exists', MagicMock(return_value=True)) |
| 18 | + def test_is_device_loop_not_allowed(self): |
| 19 | + assert not disk.is_device('/dev/loop123') |
| 20 | + |
| 21 | + @patch('ceph_volume.util.disk.lsblk', MagicMock(return_value={'NAME': 'foo', 'TYPE': 'disk'})) |
| 22 | + @patch('ceph_volume.util.disk.os.path.exists', MagicMock(return_value=True)) |
| 23 | + def test_is_device_type_disk(self): |
| 24 | + assert disk.is_device('/dev/foo') |
| 25 | + |
| 26 | + @patch('ceph_volume.util.disk.lsblk', MagicMock(return_value={'NAME': 'foo', 'TYPE': 'mpath'})) |
| 27 | + @patch('ceph_volume.util.disk.os.path.exists', MagicMock(return_value=True)) |
| 28 | + def test_is_device_type_mpath(self): |
| 29 | + assert disk.is_device('/dev/foo') |
| 30 | + |
| 31 | + @patch('ceph_volume.util.disk.lsblk', MagicMock(return_value={'NAME': 'foo1', 'TYPE': 'part'})) |
| 32 | + @patch('ceph_volume.util.disk.os.path.exists', MagicMock(return_value=True)) |
| 33 | + def test_is_device_type_part(self): |
| 34 | + assert not disk.is_device('/dev/foo1') |
5 | 35 |
|
6 | 36 |
|
7 | 37 | class TestLsblkParser(object): |
|
0 commit comments