299 |
299 |
class TestGetNodeRole(unittest.TestCase):
|
300 |
300 |
def test(self):
|
301 |
301 |
tested_role = set()
|
302 |
|
|
|
302 |
master_uuid = "969502b9-f632-4d3d-83a5-a78b0ca8cdf6"
|
|
303 |
node_uuid = "d75499b5-83e3-4b80-b6fe-3e1aee7e5a35"
|
303 |
304 |
checks = [
|
304 |
|
(constants.NR_MASTER, "node1", objects.Node(name="node1")),
|
305 |
|
(constants.NR_MCANDIDATE, "master",
|
306 |
|
objects.Node(name="node1", master_candidate=True)),
|
307 |
|
(constants.NR_REGULAR, "master", objects.Node(name="node1")),
|
308 |
|
(constants.NR_DRAINED, "master",
|
309 |
|
objects.Node(name="node1", drained=True)),
|
|
305 |
(constants.NR_MASTER,
|
|
306 |
objects.Node(name="node1", uuid=master_uuid)),
|
|
307 |
(constants.NR_MCANDIDATE,
|
|
308 |
objects.Node(name="node1", uuid=node_uuid, master_candidate=True)),
|
|
309 |
(constants.NR_REGULAR,
|
|
310 |
objects.Node(name="node1", uuid=node_uuid)),
|
|
311 |
(constants.NR_DRAINED,
|
|
312 |
objects.Node(name="node1", uuid=node_uuid, drained=True)),
|
310 |
313 |
(constants.NR_OFFLINE,
|
311 |
|
"master", objects.Node(name="node1", offline=True)),
|
|
314 |
objects.Node(name="node1", uuid=node_uuid, offline=True)),
|
312 |
315 |
]
|
313 |
316 |
|
314 |
|
for (role, master_name, node) in checks:
|
315 |
|
result = query._GetNodeRole(node, master_name)
|
|
317 |
for (role, node) in checks:
|
|
318 |
result = query._GetNodeRole(node, master_uuid)
|
316 |
319 |
self.assertEqual(result, role)
|
317 |
320 |
tested_role.add(result)
|
318 |
321 |
|