#
#
-# Copyright (C) 2007 Google Inc.
+# Copyright (C) 2007, 2011 Google Inc.
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# 02110-1301, USA.
+"""Node-related QA tests.
+
+"""
+
from ganeti import utils
from ganeti import constants
from ganeti import query
+from ganeti import serializer
import qa_config
import qa_error
import qa_utils
-from qa_utils import AssertCommand
+from qa_utils import AssertCommand, AssertEqual
def _NodeAdd(node, readd=False):
AssertCommand(["gnt-node", "modify-storage", "--allocatable", i,
node_name, storage_type, st_name], fail=fail)
+ # Verify list output
+ cmd = ["gnt-node", "list-storage", "--storage-type", storage_type,
+ "--output=name,allocatable", "--separator=|",
+ "--no-headers", node_name]
+ listout = qa_utils.GetCommandOutput(master["primary"],
+ utils.ShellQuoteArgs(cmd))
+ for line in listout.splitlines():
+ (vfy_name, vfy_allocatable) = line.split("|")
+ if vfy_name == st_name and not fail:
+ AssertEqual(vfy_allocatable, i[0].upper())
+ else:
+ AssertEqual(vfy_allocatable, st_allocatable)
+
# Test repair functionality
fail = (constants.SO_FIX_CONSISTENCY not in
constants.VALID_STORAGE_OPERATIONS.get(storage_type, []))
"--auto-promote", node["primary"]])
+def _CreateOobScriptStructure():
+ """Create a simple OOB handling script and its structure."""
+ master = qa_config.GetMasterNode()
+
+ data_path = qa_utils.UploadData(master["primary"], "")
+ verify_path = qa_utils.UploadData(master["primary"], "")
+ exit_code_path = qa_utils.UploadData(master["primary"], "")
+
+ oob_script = (("#!/bin/bash\n"
+ "echo \"$@\" > %s\n"
+ "cat %s\n"
+ "exit $(< %s)\n") %
+ (utils.ShellQuote(verify_path), utils.ShellQuote(data_path),
+ utils.ShellQuote(exit_code_path)))
+ oob_path = qa_utils.UploadData(master["primary"], oob_script, mode=0700)
+
+ return [oob_path, verify_path, data_path, exit_code_path]
+
+
+def _UpdateOobFile(path, data):
+ """Updates the data file with data."""
+ master = qa_config.GetMasterNode()
+ qa_utils.UploadData(master["primary"], data, filename=path)
+
+
+def _AssertOobCall(verify_path, expected_args):
+ """Assert the OOB call was performed with expetected args."""
+ master = qa_config.GetMasterNode()
+
+ verify_output_cmd = utils.ShellQuoteArgs(["cat", verify_path])
+ output = qa_utils.GetCommandOutput(master["primary"], verify_output_cmd,
+ tty=False)
+
+ AssertEqual(expected_args, output.strip())
+
+
+def TestOutOfBand():
+ """gnt-node power"""
+ master = qa_config.GetMasterNode()
+
+ node = qa_config.AcquireNode(exclude=master)
+
+ master_name = master["primary"]
+ node_name = node["primary"]
+ full_node_name = qa_utils.ResolveNodeName(node)
+
+ (oob_path, verify_path,
+ data_path, exit_code_path) = _CreateOobScriptStructure()
+
+ try:
+ AssertCommand(["gnt-cluster", "modify", "--node-parameters",
+ "oob_program=%s" % oob_path])
+
+ # No data, exit 0
+ _UpdateOobFile(exit_code_path, "0")
+
+ AssertCommand(["gnt-node", "power", "on", node_name])
+ _AssertOobCall(verify_path, "power-on %s" % full_node_name)
+
+ AssertCommand(["gnt-node", "power", "-f", "off", node_name])
+ _AssertOobCall(verify_path, "power-off %s" % full_node_name)
+
+ # Power off on master without options should fail
+ AssertCommand(["gnt-node", "power", "-f", "off", master_name], fail=True)
+ # With force master it should still fail
+ AssertCommand(["gnt-node", "power", "-f", "--ignore-status", "off",
+ master_name],
+ fail=True)
+
+ # Verify we can't transform back to online when not yet powered on
+ AssertCommand(["gnt-node", "modify", "-O", "no", node_name],
+ fail=True)
+ # Now reset state
+ AssertCommand(["gnt-node", "modify", "-O", "no", "--node-powered", "yes",
+ node_name])
+
+ AssertCommand(["gnt-node", "power", "-f", "cycle", node_name])
+ _AssertOobCall(verify_path, "power-cycle %s" % full_node_name)
+
+ # Those commands should fail as they expect output which isn't provided yet
+ # But they should have called the oob helper nevermind
+ AssertCommand(["gnt-node", "power", "status", node_name],
+ fail=True)
+ _AssertOobCall(verify_path, "power-status %s" % full_node_name)
+
+ AssertCommand(["gnt-node", "health", node_name],
+ fail=True)
+ _AssertOobCall(verify_path, "health %s" % full_node_name)
+
+ AssertCommand(["gnt-node", "health"], fail=True)
+
+ # Correct Data, exit 0
+ _UpdateOobFile(data_path, serializer.DumpJson({ "powered": True }))
+
+ AssertCommand(["gnt-node", "power", "status", node_name])
+ _AssertOobCall(verify_path, "power-status %s" % full_node_name)
+
+ _UpdateOobFile(data_path, serializer.DumpJson([["temp", "OK"],
+ ["disk0", "CRITICAL"]]))
+
+ AssertCommand(["gnt-node", "health", node_name])
+ _AssertOobCall(verify_path, "health %s" % full_node_name)
+
+ AssertCommand(["gnt-node", "health"])
+
+
+ # Those commands should fail as they expect no data regardless of exit 0
+ AssertCommand(["gnt-node", "power", "on", node_name], fail=True)
+ _AssertOobCall(verify_path, "power-on %s" % full_node_name)
+
+ try:
+ AssertCommand(["gnt-node", "power", "-f", "off", node_name], fail=True)
+ _AssertOobCall(verify_path, "power-off %s" % full_node_name)
+ finally:
+ AssertCommand(["gnt-node", "modify", "-O", "no", node_name])
+
+ AssertCommand(["gnt-node", "power", "-f", "cycle", node_name], fail=True)
+ _AssertOobCall(verify_path, "power-cycle %s" % full_node_name)
+
+ # Data, exit 1 (all should fail)
+ _UpdateOobFile(exit_code_path, "1")
+
+ AssertCommand(["gnt-node", "power", "on", node_name], fail=True)
+ _AssertOobCall(verify_path, "power-on %s" % full_node_name)
+
+ try:
+ AssertCommand(["gnt-node", "power", "-f", "off", node_name], fail=True)
+ _AssertOobCall(verify_path, "power-off %s" % full_node_name)
+ finally:
+ AssertCommand(["gnt-node", "modify", "-O", "no", node_name])
+
+ AssertCommand(["gnt-node", "power", "-f", "cycle", node_name], fail=True)
+ _AssertOobCall(verify_path, "power-cycle %s" % full_node_name)
+
+ AssertCommand(["gnt-node", "power", "status", node_name],
+ fail=True)
+ _AssertOobCall(verify_path, "power-status %s" % full_node_name)
+
+ AssertCommand(["gnt-node", "health", node_name],
+ fail=True)
+ _AssertOobCall(verify_path, "health %s" % full_node_name)
+
+ AssertCommand(["gnt-node", "health"], fail=True)
+
+ # No data, exit 1 (all should fail)
+ _UpdateOobFile(data_path, "")
+ AssertCommand(["gnt-node", "power", "on", node_name], fail=True)
+ _AssertOobCall(verify_path, "power-on %s" % full_node_name)
+
+ try:
+ AssertCommand(["gnt-node", "power", "-f", "off", node_name], fail=True)
+ _AssertOobCall(verify_path, "power-off %s" % full_node_name)
+ finally:
+ AssertCommand(["gnt-node", "modify", "-O", "no", node_name])
+
+ AssertCommand(["gnt-node", "power", "-f", "cycle", node_name], fail=True)
+ _AssertOobCall(verify_path, "power-cycle %s" % full_node_name)
+
+ AssertCommand(["gnt-node", "power", "status", node_name],
+ fail=True)
+ _AssertOobCall(verify_path, "power-status %s" % full_node_name)
+
+ AssertCommand(["gnt-node", "health", node_name],
+ fail=True)
+ _AssertOobCall(verify_path, "health %s" % full_node_name)
+
+ AssertCommand(["gnt-node", "health"], fail=True)
+
+ # Different OOB script for node
+ verify_path2 = qa_utils.UploadData(master["primary"], "")
+ oob_script = ("#!/bin/sh\n"
+ "echo \"$@\" > %s\n") % verify_path2
+ oob_path2 = qa_utils.UploadData(master["primary"], oob_script, mode=0700)
+
+ try:
+ AssertCommand(["gnt-node", "modify", "--node-parameters",
+ "oob_program=%s" % oob_path2, node_name])
+ AssertCommand(["gnt-node", "power", "on", node_name])
+ _AssertOobCall(verify_path2, "power-on %s" % full_node_name)
+ finally:
+ AssertCommand(["gnt-node", "modify", "--node-parameters",
+ "oob_program=default", node_name])
+ AssertCommand(["rm", "-f", oob_path2, verify_path2])
+ finally:
+ AssertCommand(["gnt-cluster", "modify", "--node-parameters",
+ "oob_program="])
+ AssertCommand(["rm", "-f", oob_path, verify_path, data_path,
+ exit_code_path])
+
+
def TestNodeList():
"""gnt-node list"""
qa_utils.GenericQueryTest("gnt-node", query.NODE_FIELDS.keys())