+from ganeti import constants
+from ganeti import query
+from ganeti import utils
+
+import qa_config
+import qa_utils
+
+from qa_utils import AssertCommand, AssertEqual, GetCommandOutput
+
+
+def GetDefaultGroup():
+ """Returns the default node group.
+
+ """
+ groups = qa_config.get("groups", {})
+ return groups.get("group-with-nodes", constants.INITIAL_NODE_GROUP_NAME)
+
+
+def TestGroupAddRemoveRename():
+ """gnt-group add/remove/rename"""
+ groups = qa_config.get("groups", {})
+
+ existing_group_with_nodes = GetDefaultGroup()
+
+ group1, group2, group3 = groups.get("inexistent-groups",
+ ["group1", "group2", "group3"])[:3]
+
+ AssertCommand(["gnt-group", "add", group1])
+ AssertCommand(["gnt-group", "add", group2])
+ AssertCommand(["gnt-group", "add", group2], fail=True)
+ AssertCommand(["gnt-group", "add", existing_group_with_nodes], fail=True)
+
+ AssertCommand(["gnt-group", "rename", group1, group2], fail=True)
+ AssertCommand(["gnt-group", "rename", group1, group3])
+
+ try:
+ AssertCommand(["gnt-group", "rename", existing_group_with_nodes, group1])
+
+ AssertCommand(["gnt-group", "remove", group2])
+ AssertCommand(["gnt-group", "remove", group3])
+ AssertCommand(["gnt-group", "remove", group1], fail=True)
+ finally:
+ # Try to ensure idempotency re groups that already existed.
+ AssertCommand(["gnt-group", "rename", group1, existing_group_with_nodes])
+
+
+def TestGroupAddWithOptions():
+ """gnt-group add with options"""
+ groups = qa_config.get("groups", {})
+ group1 = groups.get("inexistent-groups", ["group1"])[0]
+
+ AssertCommand(["gnt-group", "add", "--alloc-policy", "notvalid", group1],
+ fail=True)
+
+ AssertCommand(["gnt-group", "add", "--alloc-policy", "last_resort",
+ "--node-parameters", "oob_program=/bin/true", group1])
+
+ AssertCommand(["gnt-group", "remove", group1])
+
+
+def TestGroupModify():
+ """gnt-group modify"""
+ groups = qa_config.get("groups", {})
+ group1 = groups.get("inexistent-groups", ["group1"])[0]
+
+ AssertCommand(["gnt-group", "add", group1])
+
+ std_defaults = constants.IPOLICY_DEFAULTS[constants.ISPECS_STD]
+ min_v = std_defaults[constants.ISPEC_MEM_SIZE] * 10
+ max_v = min_v * 10
+
+ try:
+ AssertCommand(["gnt-group", "modify", "--alloc-policy", "unallocable",
+ "--node-parameters", "oob_program=/bin/false", group1])
+ AssertCommand(["gnt-group", "modify",
+ "--alloc-policy", "notvalid", group1], fail=True)
+ AssertCommand(["gnt-group", "modify", "--specs-mem-size",
+ "min=%s,max=%s,std=0" % (min_v, max_v), group1], fail=True)
+ AssertCommand(["gnt-group", "modify", "--specs-mem-size",
+ "min=%s,max=%s" % (min_v, max_v), group1])
+ AssertCommand(["gnt-group", "modify",
+ "--node-parameters", "spindle_count=10", group1])
+ if qa_config.TestEnabled("htools"):
+ AssertCommand(["hbal", "-L", "-G", group1])
+ AssertCommand(["gnt-group", "modify",
+ "--node-parameters", "spindle_count=default", group1])
+ finally:
+ AssertCommand(["gnt-group", "remove", group1])
+
+
+def TestGroupList():