#!/usr/bin/python
#
-# Copyright (C) 2006, 2007 Google Inc.
+# Copyright (C) 2007 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.
-"""Script for doing Q&A on Ganeti
+"""Script for doing QA on Ganeti.
You can create the required known_hosts file using ssh-keyscan. It's mandatory
to use the full name of a node (FQDN). For security reasons, verify the keys
Example: ssh-keyscan -t rsa node{1,2,3,4}.example.com > known_hosts
"""
-import os
-import re
import sys
-import yaml
-import time
-import tempfile
-from datetime import datetime
-from optparse import OptionParser
-
-# I want more flexibility for testing over SSH, therefore I'm not using
-# Ganeti's ssh module.
-import subprocess
-
-from ganeti import utils
-from ganeti import constants
-
-# {{{ Global variables
-cfg = None
-options = None
-# }}}
-
-# {{{ Errors
-class Error(Exception):
- """An error occurred during Q&A testing.
-
- """
- pass
-
-
-class OutOfNodesError(Error):
- """Out of nodes.
-
- """
- pass
-
-
-class OutOfInstancesError(Error):
- """Out of instances.
-
- """
- pass
-# }}}
-
-# {{{ Utilities
-def TestEnabled(test):
- """Returns True if the given test is enabled."""
- return cfg.get('tests', {}).get(test, False)
-
-
-def RunTest(callable, *args):
+import datetime
+import optparse
+
+import qa_cluster
+import qa_config
+import qa_daemon
+import qa_env
+import qa_instance
+import qa_node
+import qa_os
+import qa_other
+import qa_tags
+import qa_utils
+
+
+def RunTest(fn, *args):
"""Runs a test after printing a header.
"""
- if callable.__doc__:
- desc = callable.__doc__.splitlines()[0].strip()
+ if fn.__doc__:
+ desc = fn.__doc__.splitlines()[0].strip()
else:
- desc = '%r' % callable
+ desc = '%r' % fn
- now = str(datetime.now())
+ now = str(datetime.datetime.now())
print
print '---', now, ('-' * (55 - len(now)))
print desc
print '-' * 60
- return callable(*args)
-
-
-def AssertEqual(first, second, msg=None):
- """Raises an error when values aren't equal.
-
- """
- if not first == second:
- raise Error(msg or '%r == %r' % (first, second))
+ return fn(*args)
-def GetSSHCommand(node, cmd, strict=True):
- """Builds SSH command to be executed.
+def RunEnvTests():
+ """Run several environment tests.
"""
- args = [ 'ssh', '-oEscapeChar=none', '-oBatchMode=yes', '-l', 'root' ]
-
- if strict:
- tmp = 'yes'
- else:
- tmp = 'no'
- args.append('-oStrictHostKeyChecking=%s' % tmp)
- args.append('-oClearAllForwardings=yes')
- args.append('-oForwardAgent=yes')
- args.append(node)
-
- if options.dry_run:
- prefix = 'exit 0; '
- else:
- prefix = ''
-
- args.append(prefix + cmd)
+ if not qa_config.TestEnabled('env'):
+ return
- print 'SSH:', utils.ShellQuoteArgs(args)
+ RunTest(qa_env.TestSshConnection)
+ RunTest(qa_env.TestIcmpPing)
+ RunTest(qa_env.TestGanetiCommands)
- return args
-
-def StartSSH(node, cmd, strict=True):
- """Starts SSH.
+def SetupCluster():
+ """Initializes the cluster.
"""
- args = GetSSHCommand(node, cmd, strict=strict)
- return subprocess.Popen(args, shell=False)
+ RunTest(qa_cluster.TestClusterInit)
+ RunTest(qa_node.TestNodeAddAll)
+ if qa_config.TestEnabled('node-info'):
+ RunTest(qa_node.TestNodeInfo)
-def UploadFile(node, file):
- """Uploads a file to a node and returns the filename.
+def RunClusterTests():
+ """Runs tests related to gnt-cluster.
- Caller needs to remove the returned file on the node when it's not needed
- anymore.
"""
- # Make sure nobody else has access to it while preserving local permissions
- mode = os.stat(file).st_mode & 0700
-
- cmd = ('tmp=$(tempfile --mode %o --prefix gnt) && '
- '[[ -f "${tmp}" ]] && '
- 'cat > "${tmp}" && '
- 'echo "${tmp}"') % mode
-
- f = open(file, 'r')
- try:
- p = subprocess.Popen(GetSSHCommand(node, cmd), shell=False, stdin=f,
- stdout=subprocess.PIPE)
- AssertEqual(p.wait(), 0)
-
- # Return temporary filename
- return p.stdout.read().strip()
- finally:
- f.close()
-# }}}
-
-# {{{ Config helpers
-def GetMasterNode():
- return cfg['nodes'][0]
+ if qa_config.TestEnabled('cluster-verify'):
+ RunTest(qa_cluster.TestClusterVerify)
+ if qa_config.TestEnabled('cluster-rename'):
+ RunTest(qa_cluster.TestClusterRename)
-def AcquireInstance():
- """Returns an instance which isn't in use.
+ if qa_config.TestEnabled('cluster-info'):
+ RunTest(qa_cluster.TestClusterVersion)
+ RunTest(qa_cluster.TestClusterInfo)
+ RunTest(qa_cluster.TestClusterGetmaster)
- """
- # Filter out unwanted instances
- tmp_flt = lambda inst: not inst.get('_used', False)
- instances = filter(tmp_flt, cfg['instances'])
- del tmp_flt
+ if qa_config.TestEnabled('cluster-copyfile'):
+ RunTest(qa_cluster.TestClusterCopyfile)
- if len(instances) == 0:
- raise OutOfInstancesError("No instances left")
+ if qa_config.TestEnabled('cluster-command'):
+ RunTest(qa_cluster.TestClusterCommand)
- inst = instances[0]
- inst['_used'] = True
- return inst
+ if qa_config.TestEnabled('cluster-burnin'):
+ RunTest(qa_cluster.TestClusterBurnin)
+ if qa_config.TestEnabled('cluster-master-failover'):
+ RunTest(qa_cluster.TestClusterMasterFailover)
-def ReleaseInstance(inst):
- inst['_used'] = False
-
-def AcquireNode(exclude=None):
- """Returns the least used node.
+def RunOsTests():
+ """Runs all tests related to gnt-os.
"""
- master = GetMasterNode()
-
- # Filter out unwanted nodes
- # TODO: Maybe combine filters
- if exclude is None:
- nodes = cfg['nodes'][:]
- else:
- nodes = filter(lambda node: node != exclude, cfg['nodes'])
-
- tmp_flt = lambda node: node.get('_added', False) or node == master
- nodes = filter(tmp_flt, nodes)
- del tmp_flt
-
- if len(nodes) == 0:
- raise OutOfNodesError("No nodes left")
-
- # Get node with least number of uses
- def compare(a, b):
- result = cmp(a.get('_count', 0), b.get('_count', 0))
- if result == 0:
- result = cmp(a['primary'], b['primary'])
- return result
-
- nodes.sort(cmp=compare)
+ if not qa_config.TestEnabled('os'):
+ return
- node = nodes[0]
- node['_count'] = node.get('_count', 0) + 1
- return node
+ RunTest(qa_os.TestOsList)
+ RunTest(qa_os.TestOsDiagnose)
+ RunTest(qa_os.TestOsValid)
+ RunTest(qa_os.TestOsInvalid)
+ RunTest(qa_os.TestOsPartiallyValid)
-def ReleaseNode(node):
- node['_count'] = node.get('_count', 0) - 1
-# }}}
-
-# {{{ Environment tests
-def TestConfig():
- """Test configuration for sanity.
+def RunCommonInstanceTests(instance):
+ """Runs a few tests that are common to all disk types.
"""
- if len(cfg['nodes']) < 1:
- raise Error("Need at least one node")
- if len(cfg['instances']) < 1:
- raise Error("Need at least one instance")
- # TODO: Add more checks
+ if qa_config.TestEnabled('instance-shutdown'):
+ RunTest(qa_instance.TestInstanceShutdown, instance)
+ RunTest(qa_instance.TestInstanceStartup, instance)
+ if qa_config.TestEnabled('instance-list'):
+ RunTest(qa_instance.TestInstanceList)
-def TestSshConnection():
- """Test SSH connection.
+ if qa_config.TestEnabled('instance-info'):
+ RunTest(qa_instance.TestInstanceInfo, instance)
- """
- for node in cfg['nodes']:
- AssertEqual(StartSSH(node['primary'], 'exit').wait(), 0)
+ if qa_config.TestEnabled('instance-modify'):
+ RunTest(qa_instance.TestInstanceModify, instance)
+ if qa_config.TestEnabled('instance-console'):
+ RunTest(qa_instance.TestInstanceConsole, instance)
-def TestGanetiCommands():
- """Test availibility of Ganeti commands.
+ if qa_config.TestEnabled('instance-reinstall'):
+ RunTest(qa_instance.TestInstanceShutdown, instance)
+ RunTest(qa_instance.TestInstanceReinstall, instance)
+ RunTest(qa_instance.TestInstanceStartup, instance)
- """
- cmds = ( ['gnt-cluster', '--version'],
- ['gnt-os', '--version'],
- ['gnt-node', '--version'],
- ['gnt-instance', '--version'],
- ['gnt-backup', '--version'],
- ['ganeti-noded', '--version'],
- ['ganeti-watcher', '--version'] )
+ if qa_config.TestEnabled('instance-reboot'):
+ RunTest(qa_instance.TestInstanceReboot, instance)
- cmd = ' && '.join([utils.ShellQuoteArgs(i) for i in cmds])
+ if qa_config.TestEnabled('tags'):
+ RunTest(qa_tags.TestInstanceTags, instance)
- for node in cfg['nodes']:
- AssertEqual(StartSSH(node['primary'], cmd).wait(), 0)
+ if qa_config.TestEnabled('node-volumes'):
+ RunTest(qa_node.TestNodeVolumes)
-def TestIcmpPing():
- """ICMP ping each node.
+def RunExportImportTests(instance, pnode):
+ """Tries to export and import the instance.
"""
- for node in cfg['nodes']:
- check = []
- for i in cfg['nodes']:
- check.append(i['primary'])
- if i.has_key('secondary'):
- check.append(i['secondary'])
-
- ping = lambda ip: utils.ShellQuoteArgs(['ping', '-w', '3', '-c', '1', ip])
- cmd = ' && '.join([ping(i) for i in check])
-
- AssertEqual(StartSSH(node['primary'], cmd).wait(), 0)
-# }}}
-
-# {{{ Cluster tests
-def TestClusterInit():
- """gnt-cluster init"""
- master = GetMasterNode()
-
- cmd = ['gnt-cluster', 'init']
- if master.get('secondary', None):
- cmd.append('--secondary-ip=%s' % master['secondary'])
- if cfg.get('bridge', None):
- cmd.append('--bridge=%s' % cfg['bridge'])
- cmd.append('--master-netdev=%s' % cfg['bridge'])
- cmd.append(cfg['name'])
-
- AssertEqual(StartSSH(master['primary'],
- utils.ShellQuoteArgs(cmd)).wait(), 0)
-
-
-def TestClusterVerify():
- """gnt-cluster verify"""
- cmd = ['gnt-cluster', 'verify']
- AssertEqual(StartSSH(GetMasterNode()['primary'],
- utils.ShellQuoteArgs(cmd)).wait(), 0)
-
-
-def TestClusterInfo():
- """gnt-cluster info"""
- cmd = ['gnt-cluster', 'info']
- AssertEqual(StartSSH(GetMasterNode()['primary'],
- utils.ShellQuoteArgs(cmd)).wait(), 0)
-
-
-def TestClusterBurnin():
- """Burnin"""
- master = GetMasterNode()
-
- # Get as many instances as we need
- instances = []
- try:
- for _ in xrange(0, cfg.get('options', {}).get('burnin-instances', 1)):
- instances.append(AcquireInstance())
- except OutOfInstancesError:
- print "Not enough instances, continuing anyway."
-
- if len(instances) < 1:
- raise Error("Burnin needs at least one instance")
-
- # Run burnin
- try:
- script = UploadFile(master['primary'], '../tools/burnin')
+ if qa_config.TestEnabled('instance-export'):
+ expnode = qa_config.AcquireNode(exclude=pnode)
try:
- cmd = [script,
- '--os=%s' % cfg['os'],
- '--os-size=%s' % cfg['os-size'],
- '--swap-size=%s' % cfg['swap-size']]
- cmd += [inst['name'] for inst in instances]
- AssertEqual(StartSSH(master['primary'],
- utils.ShellQuoteArgs(cmd)).wait(), 0)
+ name = RunTest(qa_instance.TestInstanceExport, instance, expnode)
+
+ RunTest(qa_instance.TestBackupList, expnode)
+
+ if qa_config.TestEnabled('instance-import'):
+ newinst = qa_config.AcquireInstance()
+ try:
+ RunTest(qa_instance.TestInstanceImport, pnode, newinst,
+ expnode, name)
+ RunTest(qa_instance.TestInstanceRemove, newinst)
+ finally:
+ qa_config.ReleaseInstance(newinst)
finally:
- cmd = ['rm', '-f', script]
- AssertEqual(StartSSH(master['primary'],
- utils.ShellQuoteArgs(cmd)).wait(), 0)
- finally:
- for inst in instances:
- ReleaseInstance(inst)
-
-
-def TestClusterMasterFailover():
- """gnt-cluster masterfailover"""
- master = GetMasterNode()
-
- failovermaster = 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)
- finally:
- ReleaseNode(failovermaster)
-
-
-def TestClusterCopyfile():
- """gnt-cluster copyfile"""
- master = GetMasterNode()
-
- # Create temporary file
- f = tempfile.NamedTemporaryFile()
- f.write("I'm a testfile.\n")
- f.flush()
- f.seek(0)
-
- # Upload file to master node
- testname = 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)
- finally:
- # Remove file from all nodes
- for node in cfg['nodes']:
- cmd = ['rm', '-f', testname]
- AssertEqual(StartSSH(node['primary'],
- utils.ShellQuoteArgs(cmd)).wait(), 0)
-
-
-def TestClusterDestroy():
- """gnt-cluster destroy"""
- cmd = ['gnt-cluster', 'destroy', '--yes-do-it']
- AssertEqual(StartSSH(GetMasterNode()['primary'],
- utils.ShellQuoteArgs(cmd)).wait(), 0)
-# }}}
-
-# {{{ Node tests
-def _NodeAdd(node):
- if node.get('_added', False):
- raise Error("Node %s already in cluster" % node['primary'])
-
- cmd = ['gnt-node', 'add']
- if node.get('secondary', None):
- cmd.append('--secondary-ip=%s' % node['secondary'])
- cmd.append(node['primary'])
- AssertEqual(StartSSH(GetMasterNode()['primary'],
- utils.ShellQuoteArgs(cmd)).wait(), 0)
-
- node['_added'] = True
-
-
-def TestNodeAddAll():
- """Adding all nodes to cluster."""
- master = GetMasterNode()
- for node in cfg['nodes']:
- if node != master:
- _NodeAdd(node)
-
+ qa_config.ReleaseNode(expnode)
-def _NodeRemove(node):
- cmd = ['gnt-node', 'remove', node['primary']]
- AssertEqual(StartSSH(GetMasterNode()['primary'],
- utils.ShellQuoteArgs(cmd)).wait(), 0)
- node['_added'] = False
+def RunDaemonTests(instance, pnode):
+ """Test the ganeti-watcher script.
-def TestNodeRemoveAll():
- """Removing all nodes from cluster."""
- master = GetMasterNode()
- for node in cfg['nodes']:
- if node != master:
- _NodeRemove(node)
-
-
-def TestNodeInfo():
- """gnt-node info"""
- cmd = ['gnt-node', 'info']
- AssertEqual(StartSSH(GetMasterNode()['primary'],
- utils.ShellQuoteArgs(cmd)).wait(), 0)
-
-
-def TestNodeVolumes():
- """gnt-node volumes"""
- cmd = ['gnt-node', 'volumes']
- AssertEqual(StartSSH(GetMasterNode()['primary'],
- utils.ShellQuoteArgs(cmd)).wait(), 0)
-# }}}
-
-# {{{ Instance tests
-def _DiskTest(node, instance, args):
- cmd = ['gnt-instance', 'add',
- '--os-type=%s' % cfg['os'],
- '--os-size=%s' % cfg['os-size'],
- '--swap-size=%s' % cfg['swap-size'],
- '--memory=%s' % cfg['mem'],
- '--node=%s' % node['primary']]
- if args:
- cmd += args
- cmd.append(instance['name'])
-
- AssertEqual(StartSSH(GetMasterNode()['primary'],
- utils.ShellQuoteArgs(cmd)).wait(), 0)
- return instance
-
-
-def TestInstanceAddWithPlainDisk(node):
- """gnt-instance add -t plain"""
- return _DiskTest(node, AcquireInstance(), ['--disk-template=plain'])
-
-
-def TestInstanceAddWithLocalMirrorDisk(node):
- """gnt-instance add -t local_raid1"""
- return _DiskTest(node, AcquireInstance(), ['--disk-template=local_raid1'])
-
-
-def TestInstanceAddWithRemoteRaidDisk(node, node2):
- """gnt-instance add -t remote_raid1"""
- return _DiskTest(node, AcquireInstance(),
- ['--disk-template=remote_raid1',
- '--secondary-node=%s' % node2['primary']])
-
-
-def TestInstanceRemove(instance):
- """gnt-instance remove"""
- cmd = ['gnt-instance', 'remove', '-f', instance['name']]
- AssertEqual(StartSSH(GetMasterNode()['primary'],
- utils.ShellQuoteArgs(cmd)).wait(), 0)
-
- ReleaseInstance(instance)
-
-
-def TestInstanceStartup(instance):
- """gnt-instance startup"""
- cmd = ['gnt-instance', 'startup', instance['name']]
- AssertEqual(StartSSH(GetMasterNode()['primary'],
- utils.ShellQuoteArgs(cmd)).wait(), 0)
-
-
-def TestInstanceShutdown(instance):
- """gnt-instance shutdown"""
- cmd = ['gnt-instance', 'shutdown', instance['name']]
- AssertEqual(StartSSH(GetMasterNode()['primary'],
- utils.ShellQuoteArgs(cmd)).wait(), 0)
-
-
-def TestInstanceFailover(instance):
- """gnt-instance failover"""
- cmd = ['gnt-instance', 'failover', '--force', instance['name']]
- AssertEqual(StartSSH(GetMasterNode()['primary'],
- utils.ShellQuoteArgs(cmd)).wait(), 0)
-
-
-def TestInstanceInfo(instance):
- """gnt-instance info"""
- cmd = ['gnt-instance', 'info', instance['name']]
- AssertEqual(StartSSH(GetMasterNode()['primary'],
- utils.ShellQuoteArgs(cmd)).wait(), 0)
-# }}}
-
-# {{{ Daemon tests
-def _ResolveInstanceName(instance):
- """Gets the full Xen name of an instance.
-
- """
- master = GetMasterNode()
-
- info_cmd = utils.ShellQuoteArgs(['gnt-instance', 'info', instance['name']])
- sed_cmd = utils.ShellQuoteArgs(['sed', '-n', '-e', 's/^Instance name: *//p'])
-
- cmd = '%s | %s' % (info_cmd, sed_cmd)
- p = subprocess.Popen(GetSSHCommand(master['primary'], cmd), shell=False,
- stdout=subprocess.PIPE)
- AssertEqual(p.wait(), 0)
-
- return p.stdout.read().strip()
-
-
-def _InstanceRunning(node, name):
- """Checks whether an instance is running.
-
- Args:
- node: Node the instance runs on
- name: Full name of Xen instance
- """
- cmd = utils.ShellQuoteArgs(['xm', 'list', name]) + ' >/dev/null'
- ret = StartSSH(node['primary'], cmd).wait()
- return ret == 0
-
-
-def _XmShutdownInstance(node, name):
- """Shuts down instance using "xm" and waits for completion.
-
- Args:
- node: Node the instance runs on
- name: Full name of Xen instance
"""
- cmd = ['xm', 'shutdown', name]
- AssertEqual(StartSSH(GetMasterNode()['primary'],
- utils.ShellQuoteArgs(cmd)).wait(), 0)
-
- # Wait up to a minute
- end = time.time() + 60
- while time.time() <= end:
- if not _InstanceRunning(node, name):
- break
- time.sleep(5)
- else:
- raise Error("xm shutdown failed")
+ automatic_restart = \
+ qa_config.TestEnabled('instance-automatic-restart')
+ consecutive_failures = \
+ qa_config.TestEnabled('instance-consecutive-failures')
+ if automatic_restart or consecutive_failures:
+ qa_daemon.PrintCronWarning()
-def _ResetWatcherDaemon(node):
- """Removes the watcher daemon's state file.
+ if automatic_restart:
+ RunTest(qa_daemon.TestInstanceAutomaticRestart, pnode, instance)
- Args:
- node: Node to be reset
- """
- cmd = ['rm', '-f', constants.WATCHER_STATEFILE]
- AssertEqual(StartSSH(node['primary'],
- utils.ShellQuoteArgs(cmd)).wait(), 0)
+ if consecutive_failures:
+ RunTest(qa_daemon.TestInstanceConsecutiveFailures, pnode, instance)
-def TestInstanceAutomaticRestart(node, instance):
- """Test automatic restart of instance by ganeti-watcher.
+def RunHardwareFailureTests(instance, pnode, snode):
+ """Test cluster internal hardware failure recovery.
- Note: takes up to 6 minutes to complete.
"""
- master = GetMasterNode()
- inst_name = _ResolveInstanceName(instance)
-
- _ResetWatcherDaemon(node)
- _XmShutdownInstance(node, inst_name)
-
- # Give it a bit more than five minutes to start again
- restart_at = time.time() + 330
+ if qa_config.TestEnabled('instance-failover'):
+ RunTest(qa_instance.TestInstanceFailover, instance)
- # Wait until it's running again
- while time.time() <= restart_at:
- if _InstanceRunning(node, inst_name):
- break
- time.sleep(15)
- else:
- raise Error("Daemon didn't restart instance in time")
-
- cmd = ['gnt-instance', 'info', inst_name]
- AssertEqual(StartSSH(master['primary'],
- utils.ShellQuoteArgs(cmd)).wait(), 0)
-
-
-def TestInstanceConsecutiveFailures(node, instance):
- """Test five consecutive instance failures.
-
- Note: takes at least 35 minutes to complete.
- """
- master = GetMasterNode()
- inst_name = _ResolveInstanceName(instance)
-
- _ResetWatcherDaemon(node)
- _XmShutdownInstance(node, inst_name)
+ if qa_config.TestEnabled('instance-replace-disks'):
+ othernode = qa_config.AcquireNode(exclude=[pnode, snode])
+ try:
+ RunTest(qa_instance.TestReplaceDisks,
+ instance, pnode, snode, othernode)
+ finally:
+ qa_config.ReleaseNode(othernode)
- # Do shutdowns for 30 minutes
- finished_at = time.time() + (35 * 60)
+ if qa_config.TestEnabled('node-evacuate'):
+ RunTest(qa_node.TestNodeEvacuate, pnode, snode)
- while time.time() <= finished_at:
- if _InstanceRunning(node, inst_name):
- _XmShutdownInstance(node, inst_name)
- time.sleep(30)
+ if qa_config.TestEnabled('node-failover'):
+ RunTest(qa_node.TestNodeFailover, pnode, snode)
- # Check for some time whether the instance doesn't start again
- check_until = time.time() + 330
- while time.time() <= check_until:
- if _InstanceRunning(node, inst_name):
- raise Error("Instance started when it shouldn't")
- time.sleep(30)
+ if qa_config.TestEnabled('instance-disk-failure'):
+ RunTest(qa_instance.TestInstanceMasterDiskFailure,
+ instance, pnode, snode)
+ RunTest(qa_instance.TestInstanceSecondaryDiskFailure,
+ instance, pnode, snode)
- cmd = ['gnt-instance', 'info', inst_name]
- AssertEqual(StartSSH(master['primary'],
- utils.ShellQuoteArgs(cmd)).wait(), 0)
-# }}}
-# {{{ Other tests
-def TestUploadKnownHostsFile(localpath):
- """Uploading known_hosts file.
+def main():
+ """Main program.
"""
- master = GetMasterNode()
-
- tmpfile = UploadFile(master['primary'], localpath)
- try:
- cmd = ['mv', tmpfile, constants.SSH_KNOWN_HOSTS_FILE]
- AssertEqual(StartSSH(master['primary'],
- utils.ShellQuoteArgs(cmd)).wait(), 0)
- except:
- cmd = ['rm', '-f', tmpfile]
- AssertEqual(StartSSH(master['primary'],
- utils.ShellQuoteArgs(cmd)).wait(), 0)
- raise
-# }}}
-
-# {{{ Main program
-if __name__ == '__main__':
- # {{{ Option parsing
- parser = OptionParser(usage="%prog [options] <config-file> "
- "<known-hosts-file>")
+ parser = optparse.OptionParser(usage="%prog [options] <config-file>"
+ " <known-hosts-file>")
parser.add_option('--dry-run', dest='dry_run',
action="store_true",
help="Show what would be done")
parser.add_option('--yes-do-it', dest='yes_do_it',
action="store_true",
help="Really execute the tests")
- (options, args) = parser.parse_args()
- # }}}
+ (qa_config.options, args) = parser.parse_args()
if len(args) == 2:
(config_file, known_hosts_file) = args
else:
parser.error("Not enough arguments.")
- if not options.yes_do_it:
+ if not qa_config.options.yes_do_it:
print ("Executing this script irreversibly destroys any Ganeti\n"
"configuration on all nodes involved. If you really want\n"
"to start testing, supply the --yes-do-it option.")
sys.exit(1)
- f = open(config_file, 'r')
- try:
- cfg = yaml.load(f.read())
- finally:
- f.close()
-
- RunTest(TestConfig)
-
- RunTest(TestUploadKnownHostsFile, known_hosts_file)
-
- if TestEnabled('env'):
- RunTest(TestSshConnection)
- RunTest(TestIcmpPing)
- RunTest(TestGanetiCommands)
-
- RunTest(TestClusterInit)
-
- RunTest(TestNodeAddAll)
-
- if TestEnabled('cluster-verify'):
- RunTest(TestClusterVerify)
+ qa_config.Load(config_file)
+ qa_utils.LoadHooks()
- if TestEnabled('cluster-info'):
- RunTest(TestClusterInfo)
+ RunTest(qa_other.UploadKnownHostsFile, known_hosts_file)
- if TestEnabled('cluster-copyfile'):
- RunTest(TestClusterCopyfile)
+ RunEnvTests()
+ SetupCluster()
+ RunClusterTests()
+ RunOsTests()
- if TestEnabled('node-info'):
- RunTest(TestNodeInfo)
+ if qa_config.TestEnabled('tags'):
+ RunTest(qa_tags.TestClusterTags)
- if TestEnabled('cluster-burnin'):
- RunTest(TestClusterBurnin)
-
- if TestEnabled('cluster-master-failover'):
- RunTest(TestClusterMasterFailover)
+ if qa_config.TestEnabled('node-readd'):
+ master = qa_config.GetMasterNode()
+ pnode = qa_config.AcquireNode(exclude=master)
+ try:
+ RunTest(qa_node.TestNodeReadd, pnode)
+ finally:
+ qa_config.ReleaseNode(pnode)
- node = AcquireNode()
+ pnode = qa_config.AcquireNode()
try:
- if TestEnabled('instance-add-plain-disk'):
- instance = RunTest(TestInstanceAddWithPlainDisk, node)
- RunTest(TestInstanceShutdown, instance)
- RunTest(TestInstanceStartup, instance)
-
- if TestEnabled('instance-info'):
- RunTest(TestInstanceInfo, instance)
-
- if TestEnabled('instance-automatic-restart'):
- RunTest(TestInstanceAutomaticRestart, node, instance)
-
- if TestEnabled('instance-consecutive-failures'):
- RunTest(TestInstanceConsecutiveFailures, node, instance)
-
- if TestEnabled('node-volumes'):
- RunTest(TestNodeVolumes)
-
- RunTest(TestInstanceRemove, instance)
+ if qa_config.TestEnabled('tags'):
+ RunTest(qa_tags.TestNodeTags, pnode)
+
+ if qa_config.TestEnabled('instance-add-plain-disk'):
+ instance = RunTest(qa_instance.TestInstanceAddWithPlainDisk, pnode)
+ RunCommonInstanceTests(instance)
+ RunExportImportTests(instance, pnode)
+ RunDaemonTests(instance, pnode)
+ RunTest(qa_instance.TestInstanceRemove, instance)
del instance
- if TestEnabled('instance-add-local-mirror-disk'):
- instance = RunTest(TestInstanceAddWithLocalMirrorDisk, node)
- RunTest(TestInstanceShutdown, instance)
- RunTest(TestInstanceStartup, instance)
-
- if TestEnabled('instance-info'):
- RunTest(TestInstanceInfo, instance)
-
- if TestEnabled('node-volumes'):
- RunTest(TestNodeVolumes)
-
- RunTest(TestInstanceRemove, instance)
- del instance
-
- if TestEnabled('instance-add-remote-raid-disk'):
- node2 = AcquireNode(exclude=node)
- try:
- instance = RunTest(TestInstanceAddWithRemoteRaidDisk, node, node2)
- RunTest(TestInstanceShutdown, instance)
- RunTest(TestInstanceStartup, instance)
-
- if TestEnabled('instance-info'):
- RunTest(TestInstanceInfo, instance)
-
- if TestEnabled('instance-failover'):
- RunTest(TestInstanceFailover, instance)
-
- if TestEnabled('node-volumes'):
- RunTest(TestNodeVolumes)
-
- RunTest(TestInstanceRemove, instance)
- del instance
- finally:
- ReleaseNode(node2)
+ multinode_tests = [
+ ('instance-add-drbd-disk',
+ qa_instance.TestInstanceAddWithDrbdDisk),
+ ]
+
+ for name, func in multinode_tests:
+ if qa_config.TestEnabled(name):
+ snode = qa_config.AcquireNode(exclude=pnode)
+ try:
+ instance = RunTest(func, pnode, snode)
+ RunCommonInstanceTests(instance)
+ RunExportImportTests(instance, pnode)
+ RunHardwareFailureTests(instance, pnode, snode)
+ RunTest(qa_instance.TestInstanceRemove, instance)
+ del instance
+ finally:
+ qa_config.ReleaseNode(snode)
finally:
- ReleaseNode(node)
+ qa_config.ReleaseNode(pnode)
- RunTest(TestNodeRemoveAll)
+ RunTest(qa_node.TestNodeRemoveAll)
- if TestEnabled('cluster-destroy'):
- RunTest(TestClusterDestroy)
-# }}}
+ if qa_config.TestEnabled('cluster-destroy'):
+ RunTest(qa_cluster.TestClusterDestroy)
-# vim: foldmethod=marker :
+
+if __name__ == '__main__':
+ main()
projects / ganeti-local / blobdiff