root / qa / qa_node.py @ 1abbbbe2
History | View | Annotate | Download (4.2 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 | cec9845c | Michael Hanselmann | def _NodeAdd(node): |
33 | cec9845c | Michael Hanselmann | master = qa_config.GetMasterNode() |
34 | cec9845c | Michael Hanselmann | |
35 | cec9845c | Michael Hanselmann | if node.get('_added', False): |
36 | cec9845c | Michael Hanselmann | raise qa_error.Error("Node %s already in cluster" % node['primary']) |
37 | cec9845c | Michael Hanselmann | |
38 | cec9845c | Michael Hanselmann | cmd = ['gnt-node', 'add'] |
39 | cec9845c | Michael Hanselmann | if node.get('secondary', None): |
40 | cec9845c | Michael Hanselmann | cmd.append('--secondary-ip=%s' % node['secondary']) |
41 | cec9845c | Michael Hanselmann | cmd.append(node['primary'])
|
42 | cec9845c | Michael Hanselmann | AssertEqual(StartSSH(master['primary'],
|
43 | cec9845c | Michael Hanselmann | utils.ShellQuoteArgs(cmd)).wait(), 0)
|
44 | cec9845c | Michael Hanselmann | |
45 | cec9845c | Michael Hanselmann | node['_added'] = True |
46 | cec9845c | Michael Hanselmann | |
47 | cec9845c | Michael Hanselmann | |
48 | e42b5307 | Michael Hanselmann | @qa_utils.DefineHook('node-remove') |
49 | cec9845c | Michael Hanselmann | def _NodeRemove(node): |
50 | cec9845c | Michael Hanselmann | master = qa_config.GetMasterNode() |
51 | cec9845c | Michael Hanselmann | |
52 | cec9845c | Michael Hanselmann | cmd = ['gnt-node', 'remove', node['primary']] |
53 | cec9845c | Michael Hanselmann | AssertEqual(StartSSH(master['primary'],
|
54 | cec9845c | Michael Hanselmann | utils.ShellQuoteArgs(cmd)).wait(), 0)
|
55 | cec9845c | Michael Hanselmann | node['_added'] = False |
56 | cec9845c | Michael Hanselmann | |
57 | cec9845c | Michael Hanselmann | |
58 | cec9845c | Michael Hanselmann | def TestNodeAddAll(): |
59 | cec9845c | Michael Hanselmann | """Adding all nodes to cluster."""
|
60 | cec9845c | Michael Hanselmann | master = qa_config.GetMasterNode() |
61 | cec9845c | Michael Hanselmann | for node in qa_config.get('nodes'): |
62 | cec9845c | Michael Hanselmann | if node != master:
|
63 | cec9845c | Michael Hanselmann | _NodeAdd(node) |
64 | cec9845c | Michael Hanselmann | |
65 | cec9845c | Michael Hanselmann | |
66 | cec9845c | Michael Hanselmann | def TestNodeRemoveAll(): |
67 | cec9845c | Michael Hanselmann | """Removing all nodes from cluster."""
|
68 | cec9845c | Michael Hanselmann | master = qa_config.GetMasterNode() |
69 | cec9845c | Michael Hanselmann | for node in qa_config.get('nodes'): |
70 | cec9845c | Michael Hanselmann | if node != master:
|
71 | cec9845c | Michael Hanselmann | _NodeRemove(node) |
72 | cec9845c | Michael Hanselmann | |
73 | cec9845c | Michael Hanselmann | |
74 | e42b5307 | Michael Hanselmann | @qa_utils.DefineHook('node-info') |
75 | cec9845c | Michael Hanselmann | def TestNodeInfo(): |
76 | cec9845c | Michael Hanselmann | """gnt-node info"""
|
77 | cec9845c | Michael Hanselmann | master = qa_config.GetMasterNode() |
78 | cec9845c | Michael Hanselmann | |
79 | cec9845c | Michael Hanselmann | cmd = ['gnt-node', 'info'] |
80 | cec9845c | Michael Hanselmann | AssertEqual(StartSSH(master['primary'],
|
81 | cec9845c | Michael Hanselmann | utils.ShellQuoteArgs(cmd)).wait(), 0)
|
82 | cec9845c | Michael Hanselmann | |
83 | cec9845c | Michael Hanselmann | |
84 | e42b5307 | Michael Hanselmann | @qa_utils.DefineHook('node-volumes') |
85 | cec9845c | Michael Hanselmann | def TestNodeVolumes(): |
86 | cec9845c | Michael Hanselmann | """gnt-node volumes"""
|
87 | cec9845c | Michael Hanselmann | master = qa_config.GetMasterNode() |
88 | cec9845c | Michael Hanselmann | |
89 | cec9845c | Michael Hanselmann | cmd = ['gnt-node', 'volumes'] |
90 | cec9845c | Michael Hanselmann | AssertEqual(StartSSH(master['primary'],
|
91 | cec9845c | Michael Hanselmann | utils.ShellQuoteArgs(cmd)).wait(), 0)
|
92 | 4b62db14 | Michael Hanselmann | |
93 | 4b62db14 | Michael Hanselmann | |
94 | e42b5307 | Michael Hanselmann | @qa_utils.DefineHook('node-failover') |
95 | 4b62db14 | Michael Hanselmann | def TestNodeFailover(node, node2): |
96 | 4b62db14 | Michael Hanselmann | """gnt-node failover"""
|
97 | 4b62db14 | Michael Hanselmann | master = qa_config.GetMasterNode() |
98 | 4b62db14 | Michael Hanselmann | |
99 | d7e49c13 | Michael Hanselmann | if qa_utils.GetNodeInstances(node2, secondaries=False): |
100 | f4bc1f2c | Michael Hanselmann | raise qa_error.UnusableNodeError("Secondary node has at least one" |
101 | f4bc1f2c | Michael Hanselmann | " primary instance. This test requires"
|
102 | f4bc1f2c | Michael Hanselmann | " it to have no primary instances.")
|
103 | 4b62db14 | Michael Hanselmann | |
104 | 4b62db14 | Michael Hanselmann | # Fail over to secondary node
|
105 | 4b62db14 | Michael Hanselmann | cmd = ['gnt-node', 'failover', '-f', node['primary']] |
106 | 4b62db14 | Michael Hanselmann | AssertEqual(StartSSH(master['primary'],
|
107 | 4b62db14 | Michael Hanselmann | utils.ShellQuoteArgs(cmd)).wait(), 0)
|
108 | 4b62db14 | Michael Hanselmann | |
109 | 4b62db14 | Michael Hanselmann | # ... and back again.
|
110 | 4b62db14 | Michael Hanselmann | cmd = ['gnt-node', 'failover', '-f', node2['primary']] |
111 | 4b62db14 | Michael Hanselmann | AssertEqual(StartSSH(master['primary'],
|
112 | 4b62db14 | Michael Hanselmann | utils.ShellQuoteArgs(cmd)).wait(), 0)
|
113 | 4b62db14 | Michael Hanselmann | |
114 | 4b62db14 | Michael Hanselmann | |
115 | e42b5307 | Michael Hanselmann | @qa_utils.DefineHook('node-evacuate') |
116 | 4b62db14 | Michael Hanselmann | def TestNodeEvacuate(node, node2): |
117 | 4b62db14 | Michael Hanselmann | """gnt-node evacuate"""
|
118 | 4b62db14 | Michael Hanselmann | master = qa_config.GetMasterNode() |
119 | 4b62db14 | Michael Hanselmann | |
120 | 4b62db14 | Michael Hanselmann | node3 = qa_config.AcquireNode(exclude=[node, node2]) |
121 | 4b62db14 | Michael Hanselmann | try:
|
122 | d7e49c13 | Michael Hanselmann | if qa_utils.GetNodeInstances(node3, secondaries=True): |
123 | f4bc1f2c | Michael Hanselmann | raise qa_error.UnusableNodeError("Evacuation node has at least one" |
124 | f4bc1f2c | Michael Hanselmann | " secondary instance. This test requires"
|
125 | f4bc1f2c | Michael Hanselmann | " it to have no secondary instances.")
|
126 | 4b62db14 | Michael Hanselmann | |
127 | 4b62db14 | Michael Hanselmann | # Evacuate all secondary instances
|
128 | 4b62db14 | Michael Hanselmann | cmd = ['gnt-node', 'evacuate', '-f', node2['primary'], node3['primary']] |
129 | 4b62db14 | Michael Hanselmann | AssertEqual(StartSSH(master['primary'],
|
130 | 4b62db14 | Michael Hanselmann | utils.ShellQuoteArgs(cmd)).wait(), 0)
|
131 | 4b62db14 | Michael Hanselmann | |
132 | 4b62db14 | Michael Hanselmann | # ... and back again.
|
133 | 4b62db14 | Michael Hanselmann | cmd = ['gnt-node', 'evacuate', '-f', node3['primary'], node2['primary']] |
134 | 4b62db14 | Michael Hanselmann | AssertEqual(StartSSH(master['primary'],
|
135 | 4b62db14 | Michael Hanselmann | utils.ShellQuoteArgs(cmd)).wait(), 0)
|
136 | 4b62db14 | Michael Hanselmann | finally:
|
137 | 4b62db14 | Michael Hanselmann | qa_config.ReleaseNode(node3) |