gnt-instance: Adding instance policy to failover
[ganeti-local] / qa / qa_os.py
index 740c8c5..6426f94 100644 (file)
@@ -31,8 +31,9 @@ from ganeti import constants
 
 import qa_config
 import qa_utils
+import qa_error
 
-from qa_utils import AssertCommand
+from qa_utils import AssertCommand, AssertIn, AssertNotIn
 
 
 _TEMP_OS_NAME = "TEMP-Ganeti-QA-OS"
@@ -55,50 +56,56 @@ def TestOsDiagnose():
 
 def _TestOsModify(hvp_dict, fail=False):
   """gnt-os modify"""
-  cmd = ['gnt-os', 'modify']
+  cmd = ["gnt-os", "modify"]
 
   for hv_name, hv_params in hvp_dict.items():
-    cmd.append('-H')
+    cmd.append("-H")
     options = []
     for key, value in hv_params.items():
       options.append("%s=%s" % (key, value))
-    cmd.append('%s:%s' % (hv_name, ','.join(options)))
+    cmd.append("%s:%s" % (hv_name, ",".join(options)))
 
   cmd.append(_TEMP_OS_NAME)
   AssertCommand(cmd, fail=fail)
 
 
-def _TestOsStates():
+def _TestOsStates(os_name):
   """gnt-os modify, more stuff"""
   cmd = ["gnt-os", "modify"]
 
   for param in ["hidden", "blacklisted"]:
     for val in ["yes", "no"]:
-      new_cmd = cmd + ["--%s" % param, val, _TEMP_OS_NAME]
+      new_cmd = cmd + ["--%s" % param, val, os_name]
       AssertCommand(new_cmd)
       # check that double-running the command is OK
       AssertCommand(new_cmd)
 
 
-def _SetupTempOs(node, dirname, valid):
+def _SetupTempOs(node, dirname, variant, valid):
   """Creates a temporary OS definition on the given node.
 
   """
   sq = utils.ShellQuoteArgs
-  parts = [sq(["rm", "-rf", dirname]),
-           sq(["mkdir", "-p", dirname]),
-           sq(["cd", dirname]),
-           sq(["ln", "-fs", "/bin/true", "export"]),
-           sq(["ln", "-fs", "/bin/true", "import"]),
-           sq(["ln", "-fs", "/bin/true", "rename"])]
+  parts = [
+    sq(["rm", "-rf", dirname]),
+    sq(["mkdir", "-p", dirname]),
+    sq(["cd", dirname]),
+    sq(["ln", "-fs", "/bin/true", "export"]),
+    sq(["ln", "-fs", "/bin/true", "import"]),
+    sq(["ln", "-fs", "/bin/true", "rename"]),
+    sq(["ln", "-fs", "/bin/true", "verify"]),
+    ]
 
   if valid:
     parts.append(sq(["ln", "-fs", "/bin/true", "create"]))
 
-  parts.append(sq(["echo", str(constants.OS_API_V10)]) +
+  parts.append(sq(["echo", str(constants.OS_API_V20)]) +
                " >ganeti_api_version")
 
-  cmd = ' && '.join(parts)
+  parts.append(sq(["echo", variant]) + " >variants.list")
+  parts.append(sq(["echo", "funny this is funny"]) + " >parameters.list")
+
+  cmd = " && ".join(parts)
 
   print qa_utils.FormatInfo("Setting up %s with %s OS definition" %
                             (node["primary"],
@@ -114,12 +121,21 @@ def _RemoveTempOs(node, dirname):
   AssertCommand(["rm", "-rf", dirname], node=node)
 
 
-def _TestOs(mode):
+def _TestOs(mode, rapi_cb):
   """Generic function for OS definition testing
 
   """
+  master = qa_config.GetMasterNode()
+
+  name = _TEMP_OS_NAME
+  variant = "default"
+  fullname = "%s+%s" % (name, variant)
   dirname = _TEMP_OS_PATH
 
+  # Ensure OS is usable
+  cmd = ["gnt-os", "modify", "--hidden=no", "--blacklisted=no", name]
+  AssertCommand(cmd)
+
   nodes = []
   try:
     for i, node in enumerate(qa_config.get("nodes")):
@@ -132,27 +148,75 @@ def _TestOs(mode):
         valid = bool(i % 2)
       else:
         raise AssertionError("Unknown mode %s" % mode)
-      _SetupTempOs(node, dirname, valid)
+      _SetupTempOs(node, dirname, variant, valid)
+
+    # TODO: Use Python 2.6's itertools.permutations
+    for (hidden, blacklisted) in [(False, False), (True, False),
+                                  (False, True), (True, True)]:
+      # Change OS' visibility
+      cmd = ["gnt-os", "modify", "--hidden", ["no", "yes"][int(hidden)],
+             "--blacklisted", ["no", "yes"][int(blacklisted)], name]
+      AssertCommand(cmd)
+
+      # Diagnose, checking exit status
+      AssertCommand(["gnt-os", "diagnose"], fail=(mode != _ALL_VALID))
+
+      # Diagnose again, ignoring exit status
+      output = qa_utils.GetCommandOutput(master["primary"],
+                                         "gnt-os diagnose || :")
+      for line in output.splitlines():
+        if line.startswith("OS: %s [global status:" % name):
+          break
+      else:
+        raise qa_error.Error("Didn't find OS '%s' in 'gnt-os diagnose'" % name)
+
+      # Check info for all
+      cmd = ["gnt-os", "info"]
+      output = qa_utils.GetCommandOutput(master["primary"],
+                                         utils.ShellQuoteArgs(cmd))
+      AssertIn("%s:" % name, output.splitlines())
+
+      # Check info for OS
+      cmd = ["gnt-os", "info", name]
+      output = qa_utils.GetCommandOutput(master["primary"],
+                                         utils.ShellQuoteArgs(cmd)).splitlines()
+      AssertIn("%s:" % name, output)
+      for (field, value) in [("valid", mode == _ALL_VALID),
+                             ("hidden", hidden),
+                             ("blacklisted", blacklisted)]:
+        AssertIn("  - %s: %s" % (field, value), output)
+
+      # Only valid OSes should be listed
+      cmd = ["gnt-os", "list", "--no-headers"]
+      output = qa_utils.GetCommandOutput(master["primary"],
+                                         utils.ShellQuoteArgs(cmd))
+      if mode == _ALL_VALID and not (hidden or blacklisted):
+        assert_fn = AssertIn
+      else:
+        assert_fn = AssertNotIn
+      assert_fn(fullname, output.splitlines())
 
-    AssertCommand(["gnt-os", "diagnose"], fail=(mode != _ALL_VALID))
+      # Check via RAPI
+      if rapi_cb:
+        assert_fn(fullname, rapi_cb())
   finally:
     for node in nodes:
       _RemoveTempOs(node, dirname)
 
 
-def TestOsValid():
+def TestOsValid(rapi_cb):
   """Testing valid OS definition"""
-  return _TestOs(_ALL_VALID)
+  return _TestOs(_ALL_VALID, rapi_cb)
 
 
-def TestOsInvalid():
+def TestOsInvalid(rapi_cb):
   """Testing invalid OS definition"""
-  return _TestOs(_ALL_INVALID)
+  return _TestOs(_ALL_INVALID, rapi_cb)
 
 
-def TestOsPartiallyValid():
+def TestOsPartiallyValid(rapi_cb):
   """Testing partially valid OS definition"""
-  return _TestOs(_PARTIALLY_VALID)
+  return _TestOs(_PARTIALLY_VALID, rapi_cb)
 
 
 def TestOsModifyValid():
@@ -179,7 +243,7 @@ def TestOsModifyInvalid():
   return _TestOsModify(hv_dict, fail=True)
 
 
-def TestOsStates():
-  """Testing OS states"""
-
-  return _TestOsStates()
+def TestOsStatesNonExisting():
+  """Testing OS states with non-existing OS"""
+  AssertCommand(["test", "-e", _TEMP_OS_PATH], fail=True)
+  return _TestOsStates(_TEMP_OS_NAME)