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