Statistics
| Branch: | Tag: | Revision:

root / qa / qa_os.py @ 9b3939ea

History | View | Annotate | Download (3.5 kB)

1
# Copyright (C) 2007 Google Inc.
2
#
3
# This program is free software; you can redistribute it and/or modify
4
# it under the terms of the GNU General Public License as published by
5
# the Free Software Foundation; either version 2 of the License, or
6
# (at your option) any later version.
7
#
8
# This program is distributed in the hope that it will be useful, but
9
# WITHOUT ANY WARRANTY; without even the implied warranty of
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
11
# General Public License for more details.
12
#
13
# You should have received a copy of the GNU General Public License
14
# along with this program; if not, write to the Free Software
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
16
# 02110-1301, USA.
17

    
18

    
19
"""OS related QA tests.
20

21
"""
22

    
23
import os
24
import os.path
25

    
26
from ganeti import utils
27
from ganeti import constants
28

    
29
import qa_config
30
import qa_utils
31

    
32
from qa_utils import AssertEqual, StartSSH
33

    
34

    
35
_TEMP_OS_NAME = "TEMP-Ganeti-QA-OS"
36
_TEMP_OS_PATH = os.path.join(constants.OS_SEARCH_PATH[0], _TEMP_OS_NAME)
37

    
38

    
39
@qa_utils.DefineHook('os-list')
40
def TestOsList():
41
  """gnt-os list"""
42
  master = qa_config.GetMasterNode()
43

    
44
  cmd = ['gnt-os', 'list']
45
  AssertEqual(StartSSH(master['primary'],
46
                       utils.ShellQuoteArgs(cmd)).wait(), 0)
47

    
48

    
49
@qa_utils.DefineHook('os-diagnose')
50
def TestOsDiagnose():
51
  """gnt-os diagnose"""
52
  master = qa_config.GetMasterNode()
53

    
54
  cmd = ['gnt-os', 'diagnose']
55
  AssertEqual(StartSSH(master['primary'],
56
                       utils.ShellQuoteArgs(cmd)).wait(), 0)
57

    
58

    
59
def _SetupTempOs(node, dir, valid):
60
  """Creates a temporary OS definition on the given node.
61

62
  """
63
  sq = utils.ShellQuoteArgs
64
  parts = [sq(["rm", "-rf", dir]),
65
           sq(["mkdir", "-p", dir]),
66
           sq(["cd", dir]),
67
           sq(["ln", "-fs", "/bin/true", "export"]),
68
           sq(["ln", "-fs", "/bin/true", "import"]),
69
           sq(["ln", "-fs", "/bin/true", "rename"])]
70

    
71
  if valid:
72
    parts.append(sq(["ln", "-fs", "/bin/true", "create"]))
73

    
74
  parts.append(sq(["echo", str(constants.OS_API_VERSION)]) +
75
               " >ganeti_api_version")
76

    
77
  cmd = ' && '.join(parts)
78

    
79
  print qa_utils.FormatInfo("Setting up %s with %s OS definition" %
80
                            (node["primary"],
81
                             ["an invalid", "a valid"][int(valid)]))
82

    
83
  AssertEqual(StartSSH(node['primary'], cmd).wait(), 0)
84

    
85

    
86
def _RemoveTempOs(node, dir):
87
  """Removes a temporary OS definition.
88

89
  """
90
  cmd = ['rm', '-rf', dir]
91
  AssertEqual(StartSSH(node['primary'],
92
                       utils.ShellQuoteArgs(cmd)).wait(), 0)
93

    
94

    
95
def _TestOs(mode):
96
  """Generic function for OS definition testing
97

98
  """
99
  master = qa_config.GetMasterNode()
100
  dir = _TEMP_OS_PATH
101

    
102
  nodes = []
103
  try:
104
    i = 0
105
    for node in qa_config.get('nodes'):
106
      nodes.append(node)
107
      if mode == 0:
108
        valid = False
109
      elif mode == 1:
110
        valid = True
111
      else:
112
        valid = bool(i % 2)
113
      _SetupTempOs(node, dir, valid)
114
      i += 1
115

    
116
    cmd = ['gnt-os', 'diagnose']
117
    result = StartSSH(master['primary'],
118
                      utils.ShellQuoteArgs(cmd)).wait()
119
    if mode == 1:
120
      AssertEqual(result, 0)
121
    else:
122
      AssertEqual(result, 1)
123
  finally:
124
    for node in nodes:
125
      _RemoveTempOs(node, dir)
126

    
127

    
128
@qa_utils.DefineHook('os-valid')
129
def TestOsValid():
130
  """Testing valid OS definition"""
131
  return _TestOs(1)
132

    
133

    
134
@qa_utils.DefineHook('os-invalid')
135
def TestOsInvalid():
136
  """Testing invalid OS definition"""
137
  return _TestOs(0)
138

    
139

    
140
@qa_utils.DefineHook('os-partially-valid')
141
def TestOsPartiallyValid():
142
  """Testing partially valid OS definition"""
143
  return _TestOs(2)