#
#
-# Copyright (C) 2007 Google Inc.
+# Copyright (C) 2007, 2010 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
import tempfile
+from ganeti import constants
from ganeti import utils
import qa_config
import qa_utils
import qa_error
-from qa_utils import AssertEqual, StartSSH
+from qa_utils import AssertEqual, AssertCommand
def _RemoveFileFromAllNodes(filename):
"""Removes a file from all nodes.
"""
- for node in qa_config.get('nodes'):
- cmd = ['rm', '-f', filename]
- AssertEqual(StartSSH(node['primary'],
- utils.ShellQuoteArgs(cmd)).wait(), 0)
+ for node in qa_config.get("nodes"):
+ AssertCommand(["rm", "-f", filename], node=node)
def _CheckFileOnAllNodes(filename, content):
"""
cmd = utils.ShellQuoteArgs(["cat", filename])
- for node in qa_config.get('nodes'):
- AssertEqual(qa_utils.GetCommandOutput(node['primary'], cmd),
- content)
+ for node in qa_config.get("nodes"):
+ AssertEqual(qa_utils.GetCommandOutput(node["primary"], cmd), content)
-@qa_utils.DefineHook('cluster-init')
-def TestClusterInit():
+def TestClusterInit(rapi_user, rapi_secret):
"""gnt-cluster init"""
master = qa_config.GetMasterNode()
+ # First create the RAPI credentials
+ fh = tempfile.NamedTemporaryFile()
+ try:
+ fh.write("%s %s write\n" % (rapi_user, rapi_secret))
+ fh.flush()
+
+ tmpru = qa_utils.UploadFile(master["primary"], fh.name)
+ try:
+ AssertCommand(["mv", tmpru, constants.RAPI_USERS_FILE])
+ finally:
+ AssertCommand(["rm", "-f", tmpru])
+ finally:
+ fh.close()
+
+ # Initialize cluster
cmd = ['gnt-cluster', 'init']
+ cmd.append("--primary-ip-version=%d" %
+ qa_config.get("primary_ip_version", 4))
+
if master.get('secondary', None):
cmd.append('--secondary-ip=%s' % master['secondary'])
cmd.append('--bridge=%s' % bridge)
cmd.append('--master-netdev=%s' % bridge)
- htype = qa_config.get('hypervisor-type', None)
+ htype = qa_config.get('enabled-hypervisors', None)
if htype:
- cmd.append('--hypervisor-type=%s' % htype)
+ cmd.append('--enabled-hypervisors=%s' % htype)
cmd.append(qa_config.get('name'))
- AssertEqual(StartSSH(master['primary'],
- utils.ShellQuoteArgs(cmd)).wait(), 0)
+ AssertCommand(cmd)
-@qa_utils.DefineHook('cluster-rename')
def TestClusterRename():
"""gnt-cluster rename"""
- master = qa_config.GetMasterNode()
-
cmd = ['gnt-cluster', 'rename', '-f']
original_name = qa_config.get('name')
print qa_utils.FormatError('"rename" entry is missing')
return
- cmd_1 = cmd + [rename_target]
- cmd_2 = cmd + [original_name]
-
cmd_verify = ['gnt-cluster', 'verify']
- AssertEqual(StartSSH(master['primary'],
- utils.ShellQuoteArgs(cmd_1)).wait(), 0)
-
- AssertEqual(StartSSH(master['primary'],
- utils.ShellQuoteArgs(cmd_verify)).wait(), 0)
+ for data in [
+ cmd + [rename_target],
+ cmd_verify,
+ cmd + [original_name],
+ cmd_verify,
+ ]:
+ AssertCommand(data)
- AssertEqual(StartSSH(master['primary'],
- utils.ShellQuoteArgs(cmd_2)).wait(), 0)
- AssertEqual(StartSSH(master['primary'],
- utils.ShellQuoteArgs(cmd_verify)).wait(), 0)
-
-
-@qa_utils.DefineHook('cluster-verify')
def TestClusterVerify():
"""gnt-cluster verify"""
- master = qa_config.GetMasterNode()
+ AssertCommand(["gnt-cluster", "verify"])
+
+
+def TestJobqueue():
+ """gnt-debug test-jobqueue"""
+ AssertCommand(["gnt-debug", "test-jobqueue"])
+
+
+def TestClusterReservedLvs():
+ """gnt-cluster reserved lvs"""
+ CVERIFY = ["gnt-cluster", "verify"]
+ for fail, cmd in [
+ (False, CVERIFY),
+ (False, ["gnt-cluster", "modify", "--reserved-lvs", ""]),
+ (False, ["lvcreate", "-L1G", "-nqa-test", "xenvg"]),
+ (True, CVERIFY),
+ (False, ["gnt-cluster", "modify", "--reserved-lvs", "qa-test,other-test"]),
+ (False, CVERIFY),
+ (False, ["gnt-cluster", "modify", "--reserved-lvs", "qa-.*"]),
+ (False, CVERIFY),
+ (False, ["gnt-cluster", "modify", "--reserved-lvs", ""]),
+ (True, CVERIFY),
+ (False, ["lvremove", "-f", "xenvg/qa-test"]),
+ (False, CVERIFY),
+ ]:
+ AssertCommand(cmd, fail=fail)
+
+
+def TestClusterModifyBe():
+ """gnt-cluster modify -B"""
+ for fail, cmd in [
+ # mem
+ (False, ["gnt-cluster", "modify", "-B", "memory=256"]),
+ (False, ["sh", "-c", "gnt-cluster info|grep '^ *memory: 256$'"]),
+ (True, ["gnt-cluster", "modify", "-B", "memory=a"]),
+ (False, ["gnt-cluster", "modify", "-B", "memory=128"]),
+ (False, ["sh", "-c", "gnt-cluster info|grep '^ *memory: 128$'"]),
+ # vcpus
+ (False, ["gnt-cluster", "modify", "-B", "vcpus=4"]),
+ (False, ["sh", "-c", "gnt-cluster info|grep '^ *vcpus: 4$'"]),
+ (True, ["gnt-cluster", "modify", "-B", "vcpus=a"]),
+ (False, ["gnt-cluster", "modify", "-B", "vcpus=1"]),
+ (False, ["sh", "-c", "gnt-cluster info|grep '^ *vcpus: 1$'"]),
+ # auto_balance
+ (False, ["gnt-cluster", "modify", "-B", "auto_balance=False"]),
+ (False, ["sh", "-c", "gnt-cluster info|grep '^ *auto_balance: False$'"]),
+ (True, ["gnt-cluster", "modify", "-B", "auto_balance=1"]),
+ (False, ["gnt-cluster", "modify", "-B", "auto_balance=True"]),
+ (False, ["sh", "-c", "gnt-cluster info|grep '^ *auto_balance: True$'"]),
+ ]:
+ AssertCommand(cmd, fail=fail)
- cmd = ['gnt-cluster', 'verify']
- AssertEqual(StartSSH(master['primary'],
- utils.ShellQuoteArgs(cmd)).wait(), 0)
-
-@qa_utils.DefineHook('cluster-info')
def TestClusterInfo():
"""gnt-cluster info"""
- master = qa_config.GetMasterNode()
+ AssertCommand(["gnt-cluster", "info"])
- cmd = ['gnt-cluster', 'info']
- AssertEqual(StartSSH(master['primary'],
- utils.ShellQuoteArgs(cmd)).wait(), 0)
-
-@qa_utils.DefineHook('cluster-getmaster')
def TestClusterGetmaster():
"""gnt-cluster getmaster"""
- master = qa_config.GetMasterNode()
-
- cmd = ['gnt-cluster', 'getmaster']
- AssertEqual(StartSSH(master['primary'],
- utils.ShellQuoteArgs(cmd)).wait(), 0)
+ AssertCommand(["gnt-cluster", "getmaster"])
-@qa_utils.DefineHook('cluster-version')
def TestClusterVersion():
"""gnt-cluster version"""
+ AssertCommand(["gnt-cluster", "version"])
+
+
+def TestClusterRenewCrypto():
+ """gnt-cluster renew-crypto"""
master = qa_config.GetMasterNode()
- cmd = ['gnt-cluster', 'version']
- AssertEqual(StartSSH(master['primary'],
- utils.ShellQuoteArgs(cmd)).wait(), 0)
+ # Conflicting options
+ cmd = ["gnt-cluster", "renew-crypto", "--force",
+ "--new-cluster-certificate", "--new-confd-hmac-key"]
+ conflicting = [
+ ["--new-rapi-certificate", "--rapi-certificate=/dev/null"],
+ ["--new-cluster-domain-secret", "--cluster-domain-secret=/dev/null"],
+ ]
+ for i in conflicting:
+ AssertCommand(cmd+i, fail=True)
+
+ # Invalid RAPI certificate
+ cmd = ["gnt-cluster", "renew-crypto", "--force",
+ "--rapi-certificate=/dev/null"]
+ AssertCommand(cmd, fail=True)
+
+ rapi_cert_backup = qa_utils.BackupFile(master["primary"],
+ constants.RAPI_CERT_FILE)
+ try:
+ # Custom RAPI certificate
+ fh = tempfile.NamedTemporaryFile()
+
+ # Ensure certificate doesn't cause "gnt-cluster verify" to complain
+ validity = constants.SSL_CERT_EXPIRATION_WARN * 3
+
+ utils.GenerateSelfSignedSslCert(fh.name, validity=validity)
+
+ tmpcert = qa_utils.UploadFile(master["primary"], fh.name)
+ try:
+ AssertCommand(["gnt-cluster", "renew-crypto", "--force",
+ "--rapi-certificate=%s" % tmpcert])
+ finally:
+ AssertCommand(["rm", "-f", tmpcert])
+
+ # Custom cluster domain secret
+ cds_fh = tempfile.NamedTemporaryFile()
+ cds_fh.write(utils.GenerateSecret())
+ cds_fh.write("\n")
+ cds_fh.flush()
+
+ tmpcds = qa_utils.UploadFile(master["primary"], cds_fh.name)
+ try:
+ AssertCommand(["gnt-cluster", "renew-crypto", "--force",
+ "--cluster-domain-secret=%s" % tmpcds])
+ finally:
+ AssertCommand(["rm", "-f", tmpcds])
+
+ # Normal case
+ AssertCommand(["gnt-cluster", "renew-crypto", "--force",
+ "--new-cluster-certificate", "--new-confd-hmac-key",
+ "--new-rapi-certificate", "--new-cluster-domain-secret"])
+
+ # Restore RAPI certificate
+ AssertCommand(["gnt-cluster", "renew-crypto", "--force",
+ "--rapi-certificate=%s" % rapi_cert_backup])
+ finally:
+ AssertCommand(["rm", "-f", rapi_cert_backup])
-@qa_utils.DefineHook('cluster-burnin')
def TestClusterBurnin():
"""Burnin"""
master = qa_config.GetMasterNode()
- disk_template = (qa_config.get('options', {}).
- get('burnin-disk-template', 'remote_raid1'))
+ options = qa_config.get('options', {})
+ disk_template = options.get('burnin-disk-template', 'drbd')
+ parallel = options.get('burnin-in-parallel', False)
+ check_inst = options.get('burnin-check-instances', False)
+ do_rename = options.get('burnin-rename', '')
+ do_reboot = options.get('burnin-reboot', True)
+ reboot_types = options.get("reboot-types", constants.REBOOT_TYPES)
# Get as many instances as we need
instances = []
try:
try:
num = qa_config.get('options', {}).get('burnin-instances', 1)
- for _ in xrange(0, num):
+ for _ in range(0, num):
instances.append(qa_config.AcquireInstance())
except qa_error.OutOfInstancesError:
print "Not enough instances, continuing anyway."
# Run burnin
cmd = [script,
'--os=%s' % qa_config.get('os'),
- '--os-size=%s' % qa_config.get('os-size'),
- '--swap-size=%s' % qa_config.get('swap-size'),
+ '--disk-size=%s' % ",".join(qa_config.get('disk')),
+ '--disk-growth=%s' % ",".join(qa_config.get('disk-growth')),
'--disk-template=%s' % disk_template]
+ if parallel:
+ cmd.append('--parallel')
+ cmd.append('--early-release')
+ if check_inst:
+ cmd.append('--http-check')
+ if do_rename:
+ cmd.append('--rename=%s' % do_rename)
+ if not do_reboot:
+ cmd.append('--no-reboot')
+ else:
+ cmd.append('--reboot-types=%s' % ",".join(reboot_types))
cmd += [inst['name'] for inst in instances]
- AssertEqual(StartSSH(master['primary'],
- utils.ShellQuoteArgs(cmd)).wait(), 0)
+ AssertCommand(cmd)
finally:
- cmd = ['rm', '-f', script]
- AssertEqual(StartSSH(master['primary'],
- utils.ShellQuoteArgs(cmd)).wait(), 0)
+ AssertCommand(["rm", "-f", script])
+
finally:
for inst in instances:
qa_config.ReleaseInstance(inst)
-@qa_utils.DefineHook('cluster-master-failover')
def TestClusterMasterFailover():
- """gnt-cluster masterfailover"""
+ """gnt-cluster master-failover"""
master = qa_config.GetMasterNode()
-
failovermaster = qa_config.AcquireNode(exclude=master)
- try:
- cmd = ['gnt-cluster', 'masterfailover']
- AssertEqual(StartSSH(failovermaster['primary'],
- utils.ShellQuoteArgs(cmd)).wait(), 0)
- cmd = ['gnt-cluster', 'masterfailover']
- AssertEqual(StartSSH(master['primary'],
- utils.ShellQuoteArgs(cmd)).wait(), 0)
+ cmd = ["gnt-cluster", "master-failover"]
+ try:
+ AssertCommand(cmd, node=failovermaster)
+ AssertCommand(cmd, node=master)
finally:
qa_config.ReleaseNode(failovermaster)
-@qa_utils.DefineHook('cluster-copyfile')
def TestClusterCopyfile():
"""gnt-cluster copyfile"""
master = qa_config.GetMasterNode()
testname = qa_utils.UploadFile(master['primary'], f.name)
try:
# Copy file to all nodes
- cmd = ['gnt-cluster', 'copyfile', testname]
- AssertEqual(StartSSH(master['primary'],
- utils.ShellQuoteArgs(cmd)).wait(), 0)
+ AssertCommand(["gnt-cluster", "copyfile", testname])
_CheckFileOnAllNodes(testname, uniqueid)
finally:
_RemoveFileFromAllNodes(testname)
-@qa_utils.DefineHook('cluster-command')
def TestClusterCommand():
"""gnt-cluster command"""
- master = qa_config.GetMasterNode()
-
uniqueid = utils.NewUUID()
rfile = "/tmp/gnt%s" % utils.NewUUID()
rcmd = utils.ShellQuoteArgs(['echo', '-n', uniqueid])
"%s >%s" % (rcmd, rfile)])
try:
- AssertEqual(StartSSH(master['primary'], cmd).wait(), 0)
+ AssertCommand(cmd)
_CheckFileOnAllNodes(rfile, uniqueid)
finally:
_RemoveFileFromAllNodes(rfile)
-@qa_utils.DefineHook('cluster-destroy')
def TestClusterDestroy():
"""gnt-cluster destroy"""
- master = qa_config.GetMasterNode()
-
- cmd = ['gnt-cluster', 'destroy', '--yes-do-it']
- AssertEqual(StartSSH(master['primary'],
- utils.ShellQuoteArgs(cmd)).wait(), 0)
+ AssertCommand(["gnt-cluster", "destroy", "--yes-do-it"])