code.grnet.gr Git - ganeti-local/blob - qa/qa_group.py

   1 #
   2 #
   3
   4 # Copyright (C) 2010, 2011 Google Inc.
   5 #
   6 # This program is free software; you can redistribute it and/or modify
   7 # it under the terms of the GNU General Public License as published by
   8 # the Free Software Foundation; either version 2 of the License, or
   9 # (at your option) any later version.
  10 #
  11 # This program is distributed in the hope that it will be useful, but
  12 # WITHOUT ANY WARRANTY; without even the implied warranty of
  13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14 # General Public License for more details.
  15 #
  16 # You should have received a copy of the GNU General Public License
  17 # along with this program; if not, write to the Free Software
  18 # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
  19 # 02110-1301, USA.
  20
  21
  22 """QA tests for node groups.
  23
  24 """
  25
  26 from ganeti import constants
  27 from ganeti import query
  28 from ganeti import utils
  29
  30 import qa_config
  31 import qa_utils
  32
  33 from qa_utils import AssertCommand, AssertEqual, GetCommandOutput
  34
  35
  36 def GetDefaultGroup():
  37   """Returns the default node group.
  38
  39   """
  40   groups = qa_config.get("groups", {})
  41   return groups.get("group-with-nodes", constants.INITIAL_NODE_GROUP_NAME)
  42
  43
  44 def TestGroupAddRemoveRename():
  45   """gnt-group add/remove/rename"""
  46   groups = qa_config.get("groups", {})
  47
  48   existing_group_with_nodes = GetDefaultGroup()
  49
  50   group1, group2, group3 = groups.get("inexistent-groups",
  51                                       ["group1", "group2", "group3"])[:3]
  52
  53   AssertCommand(["gnt-group", "add", group1])
  54   AssertCommand(["gnt-group", "add", group2])
  55   AssertCommand(["gnt-group", "add", group2], fail=True)
  56   AssertCommand(["gnt-group", "add", existing_group_with_nodes], fail=True)
  57
  58   AssertCommand(["gnt-group", "rename", group1, group2], fail=True)
  59   AssertCommand(["gnt-group", "rename", group1, group3])
  60
  61   try:
  62     AssertCommand(["gnt-group", "rename", existing_group_with_nodes, group1])
  63
  64     AssertCommand(["gnt-group", "remove", group2])
  65     AssertCommand(["gnt-group", "remove", group3])
  66     AssertCommand(["gnt-group", "remove", group1], fail=True)
  67   finally:
  68     # Try to ensure idempotency re groups that already existed.
  69     AssertCommand(["gnt-group", "rename", group1, existing_group_with_nodes])
  70
  71
  72 def TestGroupAddWithOptions():
  73   """gnt-group add with options"""
  74   groups = qa_config.get("groups", {})
  75   group1 = groups.get("inexistent-groups", ["group1"])[0]
  76
  77   AssertCommand(["gnt-group", "add", "--alloc-policy", "notvalid", group1],
  78                 fail=True)
  79
  80   AssertCommand(["gnt-group", "add", "--alloc-policy", "last_resort",
  81                  "--node-parameters", "oob_program=/bin/true", group1])
  82
  83   AssertCommand(["gnt-group", "remove", group1])
  84
  85
  86 def TestGroupModify():
  87   """gnt-group modify"""
  88   groups = qa_config.get("groups", {})
  89   group1 = groups.get("inexistent-groups", ["group1"])[0]
  90
  91   AssertCommand(["gnt-group", "add", group1])
  92
  93   try:
  94     AssertCommand(["gnt-group", "modify", "--alloc-policy", "unallocable",
  95                    "--node-parameters", "oob_program=/bin/false", group1])
  96     AssertCommand(["gnt-group", "modify",
  97                    "--alloc-policy", "notvalid", group1], fail=True)
  98   finally:
  99     AssertCommand(["gnt-group", "remove", group1])
 100
 101
 102 def TestGroupList():
 103   """gnt-group list"""
 104   qa_utils.GenericQueryTest("gnt-group", query.GROUP_FIELDS.keys())
 105
 106
 107 def TestGroupListFields():
 108   """gnt-group list-fields"""
 109   qa_utils.GenericQueryFieldsTest("gnt-group", query.GROUP_FIELDS.keys())
 110
 111
 112 def TestAssignNodesIncludingSplit(orig_group, node1, node2):
 113   """gnt-group assign-nodes --force
 114
 115   Expects node1 and node2 to be primary and secondary for a common instance.
 116
 117   """
 118   assert node1 != node2
 119   groups = qa_config.get("groups", {})
 120   other_group = groups.get("inexistent-groups", ["group1"])[0]
 121
 122   master_node = qa_config.GetMasterNode()["primary"]
 123
 124   def AssertInGroup(group, nodes):
 125     real_output = GetCommandOutput(master_node,
 126                                    "gnt-node list --no-headers -o group " +
 127                                    utils.ShellQuoteArgs(nodes))
 128     AssertEqual(real_output.splitlines(), [group] * len(nodes))
 129
 130   AssertInGroup(orig_group, [node1, node2])
 131   AssertCommand(["gnt-group", "add", other_group])
 132
 133   try:
 134     AssertCommand(["gnt-group", "assign-nodes", other_group, node1, node2])
 135     AssertInGroup(other_group, [node1, node2])
 136
 137     # This should fail because moving node1 to orig_group would leave their
 138     # common instance split between orig_group and other_group.
 139     AssertCommand(["gnt-group", "assign-nodes", orig_group, node1], fail=True)
 140     AssertInGroup(other_group, [node1, node2])
 141
 142     AssertCommand(["gnt-group", "assign-nodes", "--force", orig_group, node1])
 143     AssertInGroup(orig_group, [node1])
 144     AssertInGroup(other_group, [node2])
 145
 146     AssertCommand(["gnt-group", "assign-nodes", orig_group, node2])
 147     AssertInGroup(orig_group, [node1, node2])
 148   finally:
 149     AssertCommand(["gnt-group", "remove", other_group])