Revision 91c17910 test/py/ganeti.rpc_unittest.py
| b/test/py/ganeti.rpc_unittest.py | ||
|---|---|---|
| 1 | 1 |
#!/usr/bin/python |
| 2 | 2 |
# |
| 3 | 3 |
|
| 4 |
# Copyright (C) 2010, 2011, 2012 Google Inc. |
|
| 4 |
# Copyright (C) 2010, 2011, 2012, 2013 Google Inc.
|
|
| 5 | 5 |
# |
| 6 | 6 |
# This program is free software; you can redistribute it and/or modify |
| 7 | 7 |
# it under the terms of the GNU General Public License as published by |
| ... | ... | |
| 889 | 889 |
msg="Configuration objects were modified") |
| 890 | 890 |
|
| 891 | 891 |
|
| 892 |
class TestLegacyNodeInfo(unittest.TestCase): |
|
| 893 |
KEY_BOOT = "bootid" |
|
| 894 |
KEY_VG = "disk_free" |
|
| 895 |
KEY_HV = "cpu_count" |
|
| 896 |
VAL_BOOT = 0 |
|
| 897 |
VAL_VG = 1 |
|
| 898 |
VAL_HV = 2 |
|
| 899 |
DICT_VG = {KEY_VG: VAL_VG}
|
|
| 900 |
DICT_HV = {KEY_HV: VAL_HV}
|
|
| 901 |
STD_LST = [VAL_BOOT, [DICT_VG], [DICT_HV]] |
|
| 902 |
STD_DICT = {
|
|
| 903 |
KEY_BOOT: VAL_BOOT, |
|
| 904 |
KEY_VG: VAL_VG, |
|
| 905 |
KEY_HV: VAL_HV |
|
| 906 |
} |
|
| 907 |
|
|
| 908 |
def testStandard(self): |
|
| 909 |
result = rpc.MakeLegacyNodeInfo(self.STD_LST) |
|
| 910 |
self.assertEqual(result, self.STD_DICT) |
|
| 911 |
|
|
| 912 |
def testReqVg(self): |
|
| 913 |
my_lst = [self.VAL_BOOT, [], [self.DICT_HV]] |
|
| 914 |
self.assertRaises(ValueError, rpc.MakeLegacyNodeInfo, my_lst) |
|
| 915 |
|
|
| 916 |
def testNoReqVg(self): |
|
| 917 |
my_lst = [self.VAL_BOOT, [], [self.DICT_HV]] |
|
| 918 |
result = rpc.MakeLegacyNodeInfo(my_lst, require_vg_info = False) |
|
| 919 |
self.assertEqual(result, {self.KEY_BOOT: self.VAL_BOOT,
|
|
| 920 |
self.KEY_HV: self.VAL_HV}) |
|
| 921 |
result = rpc.MakeLegacyNodeInfo(self.STD_LST, require_vg_info = False) |
|
| 922 |
self.assertEqual(result, self.STD_DICT) |
|
| 923 |
|
|
| 924 |
|
|
| 892 | 925 |
if __name__ == "__main__": |
| 893 | 926 |
testutils.GanetiTestProgram() |
Also available in: Unified diff