Revision 6c00b2c7 test/py/ganeti.rpc_unittest.py
| b/test/py/ganeti.rpc_unittest.py | ||
|---|---|---|
| 943 | 943 |
KEY_NAME: VAL_VG_NAME, |
| 944 | 944 |
KEY_STORAGE_FREE: VAL_VG_FREE, |
| 945 | 945 |
KEY_STORAGE_TOTAL: VAL_VG_TOTAL, |
| 946 |
KEY_SPINDLES_FREE: VAL_PV_FREE, |
|
| 947 |
KEY_SPINDLES_TOTAL: VAL_PV_TOTAL, |
|
| 946 | 948 |
KEY_CPU_COUNT: VAL_CPU_COUNT, |
| 947 | 949 |
} |
| 948 | 950 |
|
| 949 |
def testStandard(self):
|
|
| 950 |
result = rpc.MakeLegacyNodeInfo(self.STD_LST) |
|
| 951 |
def testWithSpindles(self):
|
|
| 952 |
result = rpc.MakeLegacyNodeInfo(self.STD_LST, constants.DT_PLAIN)
|
|
| 951 | 953 |
self.assertEqual(result, self.STD_DICT) |
| 952 | 954 |
|
| 953 |
def testSpindlesRequired(self): |
|
| 954 |
my_lst = [self.VAL_BOOT, [], [self.DICT_HV]] |
|
| 955 |
self.assertRaises(errors.OpExecError, rpc.MakeLegacyNodeInfo, my_lst, |
|
| 956 |
require_spindles=True) |
|
| 957 |
|
|
| 958 |
def testNoSpindlesRequired(self): |
|
| 959 |
my_lst = [self.VAL_BOOT, [], [self.DICT_HV]] |
|
| 960 |
result = rpc.MakeLegacyNodeInfo(my_lst, require_spindles = False) |
|
| 961 |
self.assertEqual(result, {self.KEY_BOOT: self.VAL_BOOT,
|
|
| 962 |
self.KEY_CPU_COUNT: self.VAL_CPU_COUNT}) |
|
| 963 |
result = rpc.MakeLegacyNodeInfo(self.STD_LST, require_spindles = False) |
|
| 964 |
self.assertEqual(result, self.STD_DICT) |
|
| 965 |
|
|
| 966 |
|
|
| 967 |
class TestAddDefaultStorageInfoToLegacyNodeInfo(unittest.TestCase): |
|
| 968 |
|
|
| 969 |
def setUp(self): |
|
| 970 |
self.free_storage_file = 23 |
|
| 971 |
self.total_storage_file = 42 |
|
| 972 |
self.free_storage_lvm = 69 |
|
| 973 |
self.total_storage_lvm = 666 |
|
| 974 |
self.node_info = [{"name": "myfile",
|
|
| 975 |
"type": constants.ST_FILE, |
|
| 976 |
"storage_free": self.free_storage_file, |
|
| 977 |
"storage_size": self.total_storage_file}, |
|
| 978 |
{"name": "myvg",
|
|
| 979 |
"type": constants.ST_LVM_VG, |
|
| 980 |
"storage_free": self.free_storage_lvm, |
|
| 981 |
"storage_size": self.total_storage_lvm}, |
|
| 982 |
{"name": "myspindle",
|
|
| 983 |
"type": constants.ST_LVM_PV, |
|
| 984 |
"storage_free": 33, |
|
| 985 |
"storage_size": 44}] |
|
| 986 |
|
|
| 987 |
def testAddDefaultStorageInfoToLegacyNodeInfo(self): |
|
| 988 |
result = {}
|
|
| 989 |
rpc._AddDefaultStorageInfoToLegacyNodeInfo(result, self.node_info) |
|
| 990 |
self.assertEqual(self.free_storage_file, result["storage_free"]) |
|
| 991 |
self.assertEqual(self.total_storage_file, result["storage_size"]) |
|
| 992 |
|
|
| 993 |
def testAddDefaultStorageInfoToLegacyNodeInfoNoDefaults(self): |
|
| 994 |
result = {}
|
|
| 995 |
rpc._AddDefaultStorageInfoToLegacyNodeInfo(result, self.node_info[-1:]) |
|
| 996 |
self.assertFalse("storage_free" in result)
|
|
| 997 |
self.assertFalse("storage_size" in result)
|
|
| 955 |
def testNoSpindles(self): |
|
| 956 |
my_lst = [self.VAL_BOOT, [self.DICT_VG], [self.DICT_HV]] |
|
| 957 |
result = rpc.MakeLegacyNodeInfo(my_lst, constants.DT_PLAIN) |
|
| 958 |
expected_dict = dict((k,v) for k, v in self.STD_DICT.iteritems()) |
|
| 959 |
expected_dict[self.KEY_SPINDLES_FREE] = 0 |
|
| 960 |
expected_dict[self.KEY_SPINDLES_TOTAL] = 0 |
|
| 961 |
self.assertEqual(result, expected_dict) |
|
| 998 | 962 |
|
| 999 | 963 |
|
| 1000 | 964 |
if __name__ == "__main__": |
Also available in: Unified diff