Revision 1068639f
| b/lib/rapi/client.py | ||
|---|---|---|
| 45 | 45 |
REPLACE_DISK_SECONDARY = "replace_on_secondary" |
| 46 | 46 |
REPLACE_DISK_CHG = "replace_new_secondary" |
| 47 | 47 |
REPLACE_DISK_AUTO = "replace_auto" |
| 48 |
VALID_NODE_ROLES = frozenset([ |
|
| 49 |
"drained", "master", "master-candidate", "offline", "regular", |
|
| 50 |
]) |
|
| 48 |
|
|
| 49 |
NODE_ROLE_DRAINED = "drained" |
|
| 50 |
NODE_ROLE_MASTER_CANDIATE = "master-candidate" |
|
| 51 |
NODE_ROLE_MASTER = "master" |
|
| 52 |
NODE_ROLE_OFFLINE = "offline" |
|
| 53 |
NODE_ROLE_REGULAR = "regular" |
|
| 51 | 54 |
|
| 52 | 55 |
|
| 53 | 56 |
class Error(Exception): |
| ... | ... | |
| 73 | 76 |
self.code = code |
| 74 | 77 |
|
| 75 | 78 |
|
| 76 |
class InvalidNodeRole(Error): |
|
| 77 |
"""Raised when an invalid node role is used. |
|
| 78 |
|
|
| 79 |
""" |
|
| 80 |
pass |
|
| 81 |
|
|
| 82 |
|
|
| 83 | 79 |
def FormatX509Name(x509_name): |
| 84 | 80 |
"""Formats an X509 name. |
| 85 | 81 |
|
| ... | ... | |
| 932 | 928 |
@rtype: int |
| 933 | 929 |
@return: job id |
| 934 | 930 |
|
| 935 |
@raise InvalidNodeRole: If an invalid node role is specified |
|
| 936 |
|
|
| 937 | 931 |
""" |
| 938 |
if role not in VALID_NODE_ROLES: |
|
| 939 |
raise InvalidNodeRole("%s is not a valid node role" % role)
|
|
| 940 |
|
|
| 941 |
query = [("force", force)]
|
|
| 932 |
query = [ |
|
| 933 |
("force", force),
|
|
| 934 |
] |
|
| 942 | 935 |
|
| 943 | 936 |
return self._SendRequest(HTTP_PUT, |
| 944 | 937 |
("/%s/nodes/%s/role" %
|
| b/test/ganeti.rapi.client_unittest.py | ||
|---|---|---|
| 428 | 428 |
self.assertQuery("force", ["1"])
|
| 429 | 429 |
self.assertEqual("\"master-candidate\"", self.http.last_request.data)
|
| 430 | 430 |
|
| 431 |
self.assertRaises(client.InvalidNodeRole, |
|
| 432 |
self.client.SetNodeRole, "node-bar", "fake-role") |
|
| 433 |
|
|
| 434 | 431 |
def testGetNodeStorageUnits(self): |
| 435 | 432 |
self.rapi.AddResponse("42")
|
| 436 | 433 |
self.assertEqual(42, |
Also available in: Unified diff