Statistics
| Branch: | Tag: | Revision:

root / qa / qa_group.py @ 6ea2bb8c

History | View | Annotate | Download (5.6 kB)

1
#
2
#
3

    
4
# Copyright (C) 2010, 2011, 2012 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
  existing_group_with_nodes = GetDefaultGroup()
47

    
48
  (group1, group2, group3) = qa_utils.GetNonexistentGroups(3)
49

    
50
  AssertCommand(["gnt-group", "add", group1])
51
  AssertCommand(["gnt-group", "add", group2])
52
  AssertCommand(["gnt-group", "add", group2], fail=True)
53
  AssertCommand(["gnt-group", "add", existing_group_with_nodes], fail=True)
54

    
55
  AssertCommand(["gnt-group", "rename", group1, group2], fail=True)
56
  AssertCommand(["gnt-group", "rename", group1, group3])
57

    
58
  try:
59
    AssertCommand(["gnt-group", "rename", existing_group_with_nodes, group1])
60

    
61
    AssertCommand(["gnt-group", "remove", group2])
62
    AssertCommand(["gnt-group", "remove", group3])
63
    AssertCommand(["gnt-group", "remove", group1], fail=True)
64
  finally:
65
    # Try to ensure idempotency re groups that already existed.
66
    AssertCommand(["gnt-group", "rename", group1, existing_group_with_nodes])
67

    
68

    
69
def TestGroupAddWithOptions():
70
  """gnt-group add with options"""
71
  (group1, ) = qa_utils.GetNonexistentGroups(1)
72

    
73
  AssertCommand(["gnt-group", "add", "--alloc-policy", "notvalid", group1],
74
                fail=True)
75

    
76
  AssertCommand(["gnt-group", "add", "--alloc-policy", "last_resort",
77
                 "--node-parameters", "oob_program=/bin/true", group1])
78

    
79
  AssertCommand(["gnt-group", "remove", group1])
80

    
81

    
82
def TestGroupModify():
83
  """gnt-group modify"""
84
  (group1, ) = qa_utils.GetNonexistentGroups(1)
85

    
86
  AssertCommand(["gnt-group", "add", group1])
87

    
88
  std_defaults = constants.IPOLICY_DEFAULTS[constants.ISPECS_STD]
89
  min_v = std_defaults[constants.ISPEC_MEM_SIZE] * 10
90
  max_v = min_v * 10
91

    
92
  try:
93
    AssertCommand(["gnt-group", "modify", "--alloc-policy", "unallocable",
94
                   "--node-parameters", "oob_program=/bin/false", group1])
95
    AssertCommand(["gnt-group", "modify",
96
                   "--alloc-policy", "notvalid", group1], fail=True)
97
    AssertCommand(["gnt-group", "modify", "--specs-mem-size",
98
                   "min=%s,max=%s,std=0" % (min_v, max_v), group1], fail=True)
99
    AssertCommand(["gnt-group", "modify", "--specs-mem-size",
100
                   "min=%s,max=%s" % (min_v, max_v), group1])
101
    AssertCommand(["gnt-group", "modify", "--specs-mem-size",
102
                   "min=default,max=default", group1])
103
    AssertCommand(["gnt-group", "modify", "--ipolicy-vcpu-ratio",
104
                   "3.5", group1])
105
    AssertCommand(["gnt-group", "modify", "--ipolicy-vcpu-ratio",
106
                   "default", group1])
107
    AssertCommand(["gnt-group", "modify",
108
                   "--node-parameters", "spindle_count=10", group1])
109
    if qa_config.TestEnabled("htools"):
110
      AssertCommand(["hbal", "-L", "-G", group1])
111
    AssertCommand(["gnt-group", "modify",
112
                   "--node-parameters", "spindle_count=default", group1])
113
  finally:
114
    AssertCommand(["gnt-group", "remove", group1])
115

    
116

    
117
def TestGroupList():
118
  """gnt-group list"""
119
  qa_utils.GenericQueryTest("gnt-group", query.GROUP_FIELDS.keys())
120

    
121

    
122
def TestGroupListFields():
123
  """gnt-group list-fields"""
124
  qa_utils.GenericQueryFieldsTest("gnt-group", query.GROUP_FIELDS.keys())
125

    
126

    
127
def TestAssignNodesIncludingSplit(orig_group, node1, node2):
128
  """gnt-group assign-nodes --force
129

130
  Expects node1 and node2 to be primary and secondary for a common instance.
131

132
  """
133
  assert node1 != node2
134

    
135
  (other_group, ) = qa_utils.GetNonexistentGroups(1)
136

    
137
  master_node = qa_config.GetMasterNode()["primary"]
138

    
139
  def AssertInGroup(group, nodes):
140
    real_output = GetCommandOutput(master_node,
141
                                   "gnt-node list --no-headers -o group " +
142
                                   utils.ShellQuoteArgs(nodes))
143
    AssertEqual(real_output.splitlines(), [group] * len(nodes))
144

    
145
  AssertInGroup(orig_group, [node1, node2])
146
  AssertCommand(["gnt-group", "add", other_group])
147

    
148
  try:
149
    AssertCommand(["gnt-group", "assign-nodes", other_group, node1, node2])
150
    AssertInGroup(other_group, [node1, node2])
151

    
152
    # This should fail because moving node1 to orig_group would leave their
153
    # common instance split between orig_group and other_group.
154
    AssertCommand(["gnt-group", "assign-nodes", orig_group, node1], fail=True)
155
    AssertInGroup(other_group, [node1, node2])
156

    
157
    AssertCommand(["gnt-group", "assign-nodes", "--force", orig_group, node1])
158
    AssertInGroup(orig_group, [node1])
159
    AssertInGroup(other_group, [node2])
160

    
161
    AssertCommand(["gnt-group", "assign-nodes", orig_group, node2])
162
    AssertInGroup(orig_group, [node1, node2])
163
  finally:
164
    AssertCommand(["gnt-group", "remove", other_group])