root / qa / qa_node.py @ c54784d9
History | View | Annotate | Download (4.5 kB)
1 | c68d1f43 | Michael Hanselmann | #
|
---|---|---|---|
2 | c68d1f43 | Michael Hanselmann | #
|
3 | c68d1f43 | Michael Hanselmann | |
4 | cec9845c | Michael Hanselmann | # Copyright (C) 2007 Google Inc.
|
5 | cec9845c | Michael Hanselmann | #
|
6 | cec9845c | Michael Hanselmann | # This program is free software; you can redistribute it and/or modify
|
7 | cec9845c | Michael Hanselmann | # it under the terms of the GNU General Public License as published by
|
8 | cec9845c | Michael Hanselmann | # the Free Software Foundation; either version 2 of the License, or
|
9 | cec9845c | Michael Hanselmann | # (at your option) any later version.
|
10 | cec9845c | Michael Hanselmann | #
|
11 | cec9845c | Michael Hanselmann | # This program is distributed in the hope that it will be useful, but
|
12 | cec9845c | Michael Hanselmann | # WITHOUT ANY WARRANTY; without even the implied warranty of
|
13 | cec9845c | Michael Hanselmann | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
14 | cec9845c | Michael Hanselmann | # General Public License for more details.
|
15 | cec9845c | Michael Hanselmann | #
|
16 | cec9845c | Michael Hanselmann | # You should have received a copy of the GNU General Public License
|
17 | cec9845c | Michael Hanselmann | # along with this program; if not, write to the Free Software
|
18 | cec9845c | Michael Hanselmann | # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
|
19 | cec9845c | Michael Hanselmann | # 02110-1301, USA.
|
20 | cec9845c | Michael Hanselmann | |
21 | cec9845c | Michael Hanselmann | |
22 | cec9845c | Michael Hanselmann | from ganeti import utils |
23 | cec9845c | Michael Hanselmann | |
24 | cec9845c | Michael Hanselmann | import qa_config |
25 | cec9845c | Michael Hanselmann | import qa_error |
26 | 4b62db14 | Michael Hanselmann | import qa_utils |
27 | cec9845c | Michael Hanselmann | |
28 | cec9845c | Michael Hanselmann | from qa_utils import AssertEqual, StartSSH |
29 | cec9845c | Michael Hanselmann | |
30 | cec9845c | Michael Hanselmann | |
31 | e42b5307 | Michael Hanselmann | @qa_utils.DefineHook('node-add') |
32 | e7c6e02b | Michael Hanselmann | def _NodeAdd(node, readd=False): |
33 | cec9845c | Michael Hanselmann | master = qa_config.GetMasterNode() |
34 | cec9845c | Michael Hanselmann | |
35 | e7c6e02b | Michael Hanselmann | if not readd and node.get('_added', False): |
36 | cec9845c | Michael Hanselmann | raise qa_error.Error("Node %s already in cluster" % node['primary']) |
37 | e7c6e02b | Michael Hanselmann | elif readd and not node.get('_added', False): |
38 | 102b115b | Michael Hanselmann | raise qa_error.Error("Node %s not yet in cluster" % node['primary']) |
39 | cec9845c | Michael Hanselmann | |
40 | cec9845c | Michael Hanselmann | cmd = ['gnt-node', 'add'] |
41 | cec9845c | Michael Hanselmann | if node.get('secondary', None): |
42 | cec9845c | Michael Hanselmann | cmd.append('--secondary-ip=%s' % node['secondary']) |
43 | e7c6e02b | Michael Hanselmann | if readd:
|
44 | e7c6e02b | Michael Hanselmann | cmd.append('--readd')
|
45 | cec9845c | Michael Hanselmann | cmd.append(node['primary'])
|
46 | cec9845c | Michael Hanselmann | AssertEqual(StartSSH(master['primary'],
|
47 | cec9845c | Michael Hanselmann | utils.ShellQuoteArgs(cmd)).wait(), 0)
|
48 | cec9845c | Michael Hanselmann | |
49 | cec9845c | Michael Hanselmann | node['_added'] = True |
50 | cec9845c | Michael Hanselmann | |
51 | cec9845c | Michael Hanselmann | |
52 | e42b5307 | Michael Hanselmann | @qa_utils.DefineHook('node-remove') |
53 | cec9845c | Michael Hanselmann | def _NodeRemove(node): |
54 | cec9845c | Michael Hanselmann | master = qa_config.GetMasterNode() |
55 | cec9845c | Michael Hanselmann | |
56 | cec9845c | Michael Hanselmann | cmd = ['gnt-node', 'remove', node['primary']] |
57 | cec9845c | Michael Hanselmann | AssertEqual(StartSSH(master['primary'],
|
58 | cec9845c | Michael Hanselmann | utils.ShellQuoteArgs(cmd)).wait(), 0)
|
59 | cec9845c | Michael Hanselmann | node['_added'] = False |
60 | cec9845c | Michael Hanselmann | |
61 | cec9845c | Michael Hanselmann | |
62 | cec9845c | Michael Hanselmann | def TestNodeAddAll(): |
63 | cec9845c | Michael Hanselmann | """Adding all nodes to cluster."""
|
64 | cec9845c | Michael Hanselmann | master = qa_config.GetMasterNode() |
65 | cec9845c | Michael Hanselmann | for node in qa_config.get('nodes'): |
66 | cec9845c | Michael Hanselmann | if node != master:
|
67 | e7c6e02b | Michael Hanselmann | _NodeAdd(node, readd=False)
|
68 | cec9845c | Michael Hanselmann | |
69 | cec9845c | Michael Hanselmann | |
70 | cec9845c | Michael Hanselmann | def TestNodeRemoveAll(): |
71 | cec9845c | Michael Hanselmann | """Removing all nodes from cluster."""
|
72 | cec9845c | Michael Hanselmann | master = qa_config.GetMasterNode() |
73 | cec9845c | Michael Hanselmann | for node in qa_config.get('nodes'): |
74 | cec9845c | Michael Hanselmann | if node != master:
|
75 | cec9845c | Michael Hanselmann | _NodeRemove(node) |
76 | cec9845c | Michael Hanselmann | |
77 | cec9845c | Michael Hanselmann | |
78 | e7c6e02b | Michael Hanselmann | @qa_utils.DefineHook('node-readd') |
79 | e7c6e02b | Michael Hanselmann | def TestNodeReadd(node): |
80 | e7c6e02b | Michael Hanselmann | """gnt-node add --readd"""
|
81 | e7c6e02b | Michael Hanselmann | _NodeAdd(node, readd=True)
|
82 | e7c6e02b | Michael Hanselmann | |
83 | e7c6e02b | Michael Hanselmann | |
84 | e42b5307 | Michael Hanselmann | @qa_utils.DefineHook('node-info') |
85 | cec9845c | Michael Hanselmann | def TestNodeInfo(): |
86 | cec9845c | Michael Hanselmann | """gnt-node info"""
|
87 | cec9845c | Michael Hanselmann | master = qa_config.GetMasterNode() |
88 | cec9845c | Michael Hanselmann | |
89 | cec9845c | Michael Hanselmann | cmd = ['gnt-node', 'info'] |
90 | cec9845c | Michael Hanselmann | AssertEqual(StartSSH(master['primary'],
|
91 | cec9845c | Michael Hanselmann | utils.ShellQuoteArgs(cmd)).wait(), 0)
|
92 | cec9845c | Michael Hanselmann | |
93 | cec9845c | Michael Hanselmann | |
94 | e42b5307 | Michael Hanselmann | @qa_utils.DefineHook('node-volumes') |
95 | cec9845c | Michael Hanselmann | def TestNodeVolumes(): |
96 | cec9845c | Michael Hanselmann | """gnt-node volumes"""
|
97 | cec9845c | Michael Hanselmann | master = qa_config.GetMasterNode() |
98 | cec9845c | Michael Hanselmann | |
99 | cec9845c | Michael Hanselmann | cmd = ['gnt-node', 'volumes'] |
100 | cec9845c | Michael Hanselmann | AssertEqual(StartSSH(master['primary'],
|
101 | cec9845c | Michael Hanselmann | utils.ShellQuoteArgs(cmd)).wait(), 0)
|
102 | 4b62db14 | Michael Hanselmann | |
103 | 4b62db14 | Michael Hanselmann | |
104 | e42b5307 | Michael Hanselmann | @qa_utils.DefineHook('node-failover') |
105 | 4b62db14 | Michael Hanselmann | def TestNodeFailover(node, node2): |
106 | 4b62db14 | Michael Hanselmann | """gnt-node failover"""
|
107 | 4b62db14 | Michael Hanselmann | master = qa_config.GetMasterNode() |
108 | 4b62db14 | Michael Hanselmann | |
109 | d7e49c13 | Michael Hanselmann | if qa_utils.GetNodeInstances(node2, secondaries=False): |
110 | f4bc1f2c | Michael Hanselmann | raise qa_error.UnusableNodeError("Secondary node has at least one" |
111 | f4bc1f2c | Michael Hanselmann | " primary instance. This test requires"
|
112 | f4bc1f2c | Michael Hanselmann | " it to have no primary instances.")
|
113 | 4b62db14 | Michael Hanselmann | |
114 | 4b62db14 | Michael Hanselmann | # Fail over to secondary node
|
115 | 4b62db14 | Michael Hanselmann | cmd = ['gnt-node', 'failover', '-f', node['primary']] |
116 | 4b62db14 | Michael Hanselmann | AssertEqual(StartSSH(master['primary'],
|
117 | 4b62db14 | Michael Hanselmann | utils.ShellQuoteArgs(cmd)).wait(), 0)
|
118 | 4b62db14 | Michael Hanselmann | |
119 | 4b62db14 | Michael Hanselmann | # ... and back again.
|
120 | 4b62db14 | Michael Hanselmann | cmd = ['gnt-node', 'failover', '-f', node2['primary']] |
121 | 4b62db14 | Michael Hanselmann | AssertEqual(StartSSH(master['primary'],
|
122 | 4b62db14 | Michael Hanselmann | utils.ShellQuoteArgs(cmd)).wait(), 0)
|
123 | 4b62db14 | Michael Hanselmann | |
124 | 4b62db14 | Michael Hanselmann | |
125 | e42b5307 | Michael Hanselmann | @qa_utils.DefineHook('node-evacuate') |
126 | 4b62db14 | Michael Hanselmann | def TestNodeEvacuate(node, node2): |
127 | 4b62db14 | Michael Hanselmann | """gnt-node evacuate"""
|
128 | 4b62db14 | Michael Hanselmann | master = qa_config.GetMasterNode() |
129 | 4b62db14 | Michael Hanselmann | |
130 | 4b62db14 | Michael Hanselmann | node3 = qa_config.AcquireNode(exclude=[node, node2]) |
131 | 4b62db14 | Michael Hanselmann | try:
|
132 | d7e49c13 | Michael Hanselmann | if qa_utils.GetNodeInstances(node3, secondaries=True): |
133 | f4bc1f2c | Michael Hanselmann | raise qa_error.UnusableNodeError("Evacuation node has at least one" |
134 | f4bc1f2c | Michael Hanselmann | " secondary instance. This test requires"
|
135 | f4bc1f2c | Michael Hanselmann | " it to have no secondary instances.")
|
136 | 4b62db14 | Michael Hanselmann | |
137 | 4b62db14 | Michael Hanselmann | # Evacuate all secondary instances
|
138 | 4b62db14 | Michael Hanselmann | cmd = ['gnt-node', 'evacuate', '-f', node2['primary'], node3['primary']] |
139 | 4b62db14 | Michael Hanselmann | AssertEqual(StartSSH(master['primary'],
|
140 | 4b62db14 | Michael Hanselmann | utils.ShellQuoteArgs(cmd)).wait(), 0)
|
141 | 4b62db14 | Michael Hanselmann | |
142 | 4b62db14 | Michael Hanselmann | # ... and back again.
|
143 | 4b62db14 | Michael Hanselmann | cmd = ['gnt-node', 'evacuate', '-f', node3['primary'], node2['primary']] |
144 | 4b62db14 | Michael Hanselmann | AssertEqual(StartSSH(master['primary'],
|
145 | 4b62db14 | Michael Hanselmann | utils.ShellQuoteArgs(cmd)).wait(), 0)
|
146 | 4b62db14 | Michael Hanselmann | finally:
|
147 | 4b62db14 | Michael Hanselmann | qa_config.ReleaseNode(node3) |