Revision 6c00b2c7 test/py/ganeti.utils.storage_unittest.py
b/test/py/ganeti.utils.storage_unittest.py | ||
---|---|---|
26 | 26 |
import unittest |
27 | 27 |
|
28 | 28 |
from ganeti import constants |
29 |
from ganeti import objects |
|
30 | 29 |
from ganeti.utils import storage |
31 | 30 |
|
32 | 31 |
import testutils |
... | ... | |
72 | 71 |
self.assertEqual(storage_type, constants.ST_DISKLESS) |
73 | 72 |
self.assertEqual(storage_key, None) |
74 | 73 |
|
75 |
def testGetDefaultStorageUnitForSpindles(self): |
|
76 |
(storage_type, storage_key) = \ |
|
77 |
storage._GetDefaultStorageUnitForSpindles(self._cfg) |
|
78 |
self.assertEqual(storage_type, constants.ST_LVM_PV) |
|
79 |
self.assertEqual(storage_key, self._default_vg_name) |
|
80 |
|
|
81 |
|
|
82 |
class TestGetStorageUnitsOfCluster(unittest.TestCase): |
|
83 |
|
|
84 |
def setUp(self): |
|
85 |
storage._GetDefaultStorageUnitForDiskTemplate = \ |
|
86 |
mock.Mock(return_value=("foo", "bar")) |
|
87 |
|
|
88 |
self._cluster_cfg = objects.Cluster() |
|
89 |
self._enabled_disk_templates = \ |
|
90 |
[constants.DT_DRBD8, constants.DT_PLAIN, constants.DT_FILE, |
|
91 |
constants.DT_SHARED_FILE] |
|
92 |
self._cluster_cfg.enabled_disk_templates = \ |
|
93 |
self._enabled_disk_templates |
|
94 |
self._cfg = mock.Mock() |
|
95 |
self._cfg.GetClusterInfo = mock.Mock(return_value=self._cluster_cfg) |
|
96 |
self._cfg.GetVGName = mock.Mock(return_value="some_vg_name") |
|
97 |
|
|
98 |
def testGetStorageUnitsOfCluster(self): |
|
99 |
storage_units = storage.GetStorageUnitsOfCluster(self._cfg) |
|
100 |
self.assertEqual(len(storage_units), len(self._enabled_disk_templates)) |
|
101 |
|
|
102 |
def testGetStorageUnitsOfClusterWithSpindles(self): |
|
103 |
storage_units = storage.GetStorageUnitsOfCluster( |
|
104 |
self._cfg, include_spindles=True) |
|
105 |
self.assertEqual(len(storage_units), len(self._enabled_disk_templates) + 1) |
|
106 |
self.assertTrue(constants.ST_LVM_PV in [st for (st, sk) in storage_units]) |
|
107 |
|
|
108 | 74 |
|
109 | 75 |
class TestGetStorageUnits(unittest.TestCase): |
110 | 76 |
|
... | ... | |
121 | 87 |
def testGetStorageUnitsLvm(self): |
122 | 88 |
disk_templates = [constants.DT_PLAIN, constants.DT_DRBD8] |
123 | 89 |
storage_units = storage.GetStorageUnits(self._cfg, disk_templates) |
124 |
self.assertEqual(len(storage_units), len(disk_templates) + 1)
|
|
90 |
self.assertEqual(len(storage_units), len(disk_templates)) |
|
125 | 91 |
|
126 | 92 |
|
127 | 93 |
class TestLookupSpaceInfoByStorageType(unittest.TestCase): |
Also available in: Unified diff