code.grnet.gr Git - ganeti-local/blob - qa/ganeti-qa.py

   1 #!/usr/bin/python -u
   2 #
   3
   4 # Copyright (C) 2007, 2008, 2009, 2010 Google Inc.
   5 #
   6 # This program is free software; you can redistribute it and/or modify
   7 # it under the terms of the GNU General Public License as published by
   8 # the Free Software Foundation; either version 2 of the License, or
   9 # (at your option) any later version.
  10 #
  11 # This program is distributed in the hope that it will be useful, but
  12 # WITHOUT ANY WARRANTY; without even the implied warranty of
  13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14 # General Public License for more details.
  15 #
  16 # You should have received a copy of the GNU General Public License
  17 # along with this program; if not, write to the Free Software
  18 # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
  19 # 02110-1301, USA.
  20
  21
  22 """Script for doing QA on Ganeti.
  23
  24 """
  25
  26 import sys
  27 import datetime
  28 import optparse
  29
  30 import qa_cluster
  31 import qa_config
  32 import qa_daemon
  33 import qa_env
  34 import qa_instance
  35 import qa_node
  36 import qa_os
  37 import qa_rapi
  38 import qa_tags
  39 import qa_utils
  40
  41 from ganeti import utils
  42 from ganeti import rapi
  43
  44 import ganeti.rapi.client
  45
  46
  47 def _FormatHeader(line, end=72, char="-"):
  48   """Fill a line up to the end column.
  49
  50   """
  51   line = "---- " + line + " "
  52   line += "-" * (end-len(line))
  53   line = line.rstrip()
  54   return line
  55
  56
  57 def RunTest(fn, *args):
  58   """Runs a test after printing a header.
  59
  60   """
  61   if fn.__doc__:
  62     desc = fn.__doc__.splitlines()[0].strip()
  63   else:
  64     desc = "%r" % fn
  65
  66   desc = desc.rstrip(".")
  67
  68   tstart = datetime.datetime.now()
  69
  70   print
  71   print _FormatHeader("%s start %s" % (tstart, desc))
  72
  73   try:
  74     retval = fn(*args)
  75     return retval
  76   finally:
  77     tstop = datetime.datetime.now()
  78     tdelta = tstop - tstart
  79     print _FormatHeader("%s time=%s %s" % (tstop, tdelta, desc))
  80
  81
  82 def RunEnvTests():
  83   """Run several environment tests.
  84
  85   """
  86   if not qa_config.TestEnabled('env'):
  87     return
  88
  89   RunTest(qa_env.TestSshConnection)
  90   RunTest(qa_env.TestIcmpPing)
  91   RunTest(qa_env.TestGanetiCommands)
  92
  93
  94 def SetupCluster(rapi_user, rapi_secret):
  95   """Initializes the cluster.
  96
  97   @param rapi_user: Login user for RAPI
  98   @param rapi_secret: Login secret for RAPI
  99
 100   """
 101   if qa_config.TestEnabled('create-cluster'):
 102     RunTest(qa_cluster.TestClusterInit, rapi_user, rapi_secret)
 103     RunTest(qa_node.TestNodeAddAll)
 104   else:
 105     # consider the nodes are already there
 106     qa_node.MarkNodeAddedAll()
 107
 108   if qa_config.TestEnabled("test-jobqueue"):
 109     RunTest(qa_cluster.TestJobqueue)
 110
 111   # enable the watcher (unconditionally)
 112   RunTest(qa_daemon.TestResumeWatcher)
 113
 114   if qa_config.TestEnabled('node-info'):
 115     RunTest(qa_node.TestNodeInfo)
 116
 117
 118 def RunClusterTests():
 119   """Runs tests related to gnt-cluster.
 120
 121   """
 122   if qa_config.TestEnabled("cluster-renew-crypto"):
 123     RunTest(qa_cluster.TestClusterRenewCrypto)
 124
 125   if qa_config.TestEnabled('cluster-verify'):
 126     RunTest(qa_cluster.TestClusterVerify)
 127
 128   if qa_config.TestEnabled('cluster-reserved-lvs'):
 129     RunTest(qa_cluster.TestClusterReservedLvs)
 130
 131   if qa_config.TestEnabled("cluster-modify"):
 132     RunTest(qa_cluster.TestClusterModifyBe)
 133     # TODO: add more cluster modify tests
 134
 135   if qa_config.TestEnabled('cluster-rename'):
 136     RunTest(qa_cluster.TestClusterRename)
 137
 138   if qa_config.TestEnabled('cluster-info'):
 139     RunTest(qa_cluster.TestClusterVersion)
 140     RunTest(qa_cluster.TestClusterInfo)
 141     RunTest(qa_cluster.TestClusterGetmaster)
 142
 143   if qa_config.TestEnabled('cluster-copyfile'):
 144     RunTest(qa_cluster.TestClusterCopyfile)
 145
 146   if qa_config.TestEnabled('cluster-command'):
 147     RunTest(qa_cluster.TestClusterCommand)
 148
 149   if qa_config.TestEnabled('cluster-burnin'):
 150     RunTest(qa_cluster.TestClusterBurnin)
 151
 152   if qa_config.TestEnabled('cluster-master-failover'):
 153     RunTest(qa_cluster.TestClusterMasterFailover)
 154
 155   if qa_rapi.Enabled():
 156     RunTest(qa_rapi.TestVersion)
 157     RunTest(qa_rapi.TestEmptyCluster)
 158
 159
 160 def RunOsTests():
 161   """Runs all tests related to gnt-os.
 162
 163   """
 164   if not qa_config.TestEnabled('os'):
 165     return
 166
 167   RunTest(qa_os.TestOsList)
 168   RunTest(qa_os.TestOsDiagnose)
 169   RunTest(qa_os.TestOsValid)
 170   RunTest(qa_os.TestOsInvalid)
 171   RunTest(qa_os.TestOsPartiallyValid)
 172   RunTest(qa_os.TestOsModifyValid)
 173   RunTest(qa_os.TestOsModifyInvalid)
 174   RunTest(qa_os.TestOsStates)
 175
 176
 177 def RunCommonInstanceTests(instance):
 178   """Runs a few tests that are common to all disk types.
 179
 180   """
 181   if qa_config.TestEnabled('instance-shutdown'):
 182     RunTest(qa_instance.TestInstanceShutdown, instance)
 183     RunTest(qa_instance.TestInstanceStartup, instance)
 184
 185   if qa_config.TestEnabled('instance-list'):
 186     RunTest(qa_instance.TestInstanceList)
 187
 188   if qa_config.TestEnabled('instance-info'):
 189     RunTest(qa_instance.TestInstanceInfo, instance)
 190
 191   if qa_config.TestEnabled('instance-modify'):
 192     RunTest(qa_instance.TestInstanceModify, instance)
 193     if qa_rapi.Enabled():
 194       RunTest(qa_rapi.TestRapiInstanceModify, instance)
 195
 196   if qa_config.TestEnabled('instance-console'):
 197     RunTest(qa_instance.TestInstanceConsole, instance)
 198
 199   if qa_config.TestEnabled('instance-reinstall'):
 200     RunTest(qa_instance.TestInstanceShutdown, instance)
 201     RunTest(qa_instance.TestInstanceReinstall, instance)
 202     RunTest(qa_instance.TestInstanceStartup, instance)
 203
 204   if qa_config.TestEnabled('instance-reboot'):
 205     RunTest(qa_instance.TestInstanceReboot, instance)
 206
 207   if qa_config.TestEnabled('instance-rename'):
 208     rename_target = qa_config.get("rename", None)
 209     if rename_target is None:
 210       print qa_utils.FormatError("Can rename instance, 'rename' entry is"
 211                                  " missing from configuration")
 212     else:
 213       RunTest(qa_instance.TestInstanceShutdown, instance)
 214       RunTest(qa_instance.TestInstanceRename, instance, rename_target)
 215       if qa_rapi.Enabled():
 216         RunTest(qa_rapi.TestRapiInstanceRename, instance, rename_target)
 217       RunTest(qa_instance.TestInstanceStartup, instance)
 218
 219   if qa_config.TestEnabled('tags'):
 220     RunTest(qa_tags.TestInstanceTags, instance)
 221
 222   if qa_rapi.Enabled():
 223     RunTest(qa_rapi.TestInstance, instance)
 224
 225
 226 def RunCommonNodeTests():
 227   """Run a few common node tests.
 228
 229   """
 230   if qa_config.TestEnabled('node-volumes'):
 231     RunTest(qa_node.TestNodeVolumes)
 232
 233   if qa_config.TestEnabled("node-storage"):
 234     RunTest(qa_node.TestNodeStorage)
 235
 236
 237 def RunExportImportTests(instance, pnode, snode):
 238   """Tries to export and import the instance.
 239
 240   @param pnode: current primary node of the instance
 241   @param snode: current secondary node of the instance, if any,
 242       otherwise None
 243
 244   """
 245   if qa_config.TestEnabled('instance-export'):
 246     RunTest(qa_instance.TestInstanceExportNoTarget, instance)
 247
 248     expnode = qa_config.AcquireNode(exclude=pnode)
 249     try:
 250       name = RunTest(qa_instance.TestInstanceExport, instance, expnode)
 251
 252       RunTest(qa_instance.TestBackupList, expnode)
 253
 254       if qa_config.TestEnabled('instance-import'):
 255         newinst = qa_config.AcquireInstance()
 256         try:
 257           RunTest(qa_instance.TestInstanceImport, pnode, newinst,
 258                   expnode, name)
 259           RunTest(qa_instance.TestInstanceRemove, newinst)
 260         finally:
 261           qa_config.ReleaseInstance(newinst)
 262     finally:
 263       qa_config.ReleaseNode(expnode)
 264
 265   if (qa_rapi.Enabled() and
 266       qa_config.TestEnabled("inter-cluster-instance-move")):
 267     newinst = qa_config.AcquireInstance()
 268     try:
 269       if snode is None:
 270         excl = [pnode]
 271       else:
 272         excl = [pnode, snode]
 273       tnode = qa_config.AcquireNode(exclude=excl)
 274       try:
 275         RunTest(qa_rapi.TestInterClusterInstanceMove, instance, newinst,
 276                 pnode, snode, tnode)
 277       finally:
 278         qa_config.ReleaseNode(tnode)
 279     finally:
 280       qa_config.ReleaseInstance(newinst)
 281
 282
 283 def RunDaemonTests(instance, pnode):
 284   """Test the ganeti-watcher script.
 285
 286   """
 287   automatic_restart = \
 288     qa_config.TestEnabled('instance-automatic-restart')
 289   consecutive_failures = \
 290     qa_config.TestEnabled('instance-consecutive-failures')
 291
 292   RunTest(qa_daemon.TestPauseWatcher)
 293   if automatic_restart or consecutive_failures:
 294
 295     if automatic_restart:
 296       RunTest(qa_daemon.TestInstanceAutomaticRestart, pnode, instance)
 297
 298     if consecutive_failures:
 299       RunTest(qa_daemon.TestInstanceConsecutiveFailures, pnode, instance)
 300
 301   RunTest(qa_daemon.TestResumeWatcher)
 302
 303
 304 def RunHardwareFailureTests(instance, pnode, snode):
 305   """Test cluster internal hardware failure recovery.
 306
 307   """
 308   if qa_config.TestEnabled('instance-failover'):
 309     RunTest(qa_instance.TestInstanceFailover, instance)
 310
 311   if qa_config.TestEnabled("instance-migrate"):
 312     RunTest(qa_instance.TestInstanceMigrate, instance)
 313
 314     if qa_rapi.Enabled():
 315       RunTest(qa_rapi.TestRapiInstanceMigrate, instance)
 316
 317   if qa_config.TestEnabled('instance-replace-disks'):
 318     othernode = qa_config.AcquireNode(exclude=[pnode, snode])
 319     try:
 320       RunTest(qa_instance.TestReplaceDisks,
 321               instance, pnode, snode, othernode)
 322     finally:
 323       qa_config.ReleaseNode(othernode)
 324
 325   if qa_config.TestEnabled('node-evacuate'):
 326     RunTest(qa_node.TestNodeEvacuate, pnode, snode)
 327
 328   if qa_config.TestEnabled('node-failover'):
 329     RunTest(qa_node.TestNodeFailover, pnode, snode)
 330
 331   if qa_config.TestEnabled('instance-disk-failure'):
 332     RunTest(qa_instance.TestInstanceMasterDiskFailure,
 333             instance, pnode, snode)
 334     RunTest(qa_instance.TestInstanceSecondaryDiskFailure,
 335             instance, pnode, snode)
 336
 337
 338 @rapi.client.UsesRapiClient
 339 def main():
 340   """Main program.
 341
 342   """
 343   parser = optparse.OptionParser(usage="%prog [options] <config-file>")
 344   parser.add_option('--yes-do-it', dest='yes_do_it',
 345       action="store_true",
 346       help="Really execute the tests")
 347   (qa_config.options, args) = parser.parse_args()
 348
 349   if len(args) == 1:
 350     (config_file, ) = args
 351   else:
 352     parser.error("Wrong number of arguments.")
 353
 354   if not qa_config.options.yes_do_it:
 355     print ("Executing this script irreversibly destroys any Ganeti\n"
 356            "configuration on all nodes involved. If you really want\n"
 357            "to start testing, supply the --yes-do-it option.")
 358     sys.exit(1)
 359
 360   qa_config.Load(config_file)
 361
 362   rapi_user = "ganeti-qa"
 363   rapi_secret = utils.GenerateSecret()
 364
 365   RunEnvTests()
 366   SetupCluster(rapi_user, rapi_secret)
 367
 368   # Load RAPI certificate
 369   qa_rapi.Setup(rapi_user, rapi_secret)
 370
 371   RunClusterTests()
 372   RunOsTests()
 373
 374   if qa_config.TestEnabled('tags'):
 375     RunTest(qa_tags.TestClusterTags)
 376
 377   RunCommonNodeTests()
 378
 379   pnode = qa_config.AcquireNode(exclude=qa_config.GetMasterNode())
 380   try:
 381     if qa_config.TestEnabled('node-readd'):
 382       RunTest(qa_node.TestNodeReadd, pnode)
 383
 384     if qa_config.TestEnabled("node-modify"):
 385       RunTest(qa_node.TestNodeModify, pnode)
 386   finally:
 387     qa_config.ReleaseNode(pnode)
 388
 389   pnode = qa_config.AcquireNode()
 390   try:
 391     if qa_config.TestEnabled('tags'):
 392       RunTest(qa_tags.TestNodeTags, pnode)
 393
 394     if qa_rapi.Enabled():
 395       RunTest(qa_rapi.TestNode, pnode)
 396
 397       if qa_config.TestEnabled("instance-add-plain-disk"):
 398         for use_client in [True, False]:
 399           rapi_instance = RunTest(qa_rapi.TestRapiInstanceAdd, pnode,
 400                                   use_client)
 401           RunCommonInstanceTests(rapi_instance)
 402           RunTest(qa_rapi.TestRapiInstanceRemove, rapi_instance, use_client)
 403           del rapi_instance
 404
 405     if qa_config.TestEnabled('instance-add-plain-disk'):
 406       instance = RunTest(qa_instance.TestInstanceAddWithPlainDisk, pnode)
 407       RunCommonInstanceTests(instance)
 408       RunExportImportTests(instance, pnode, None)
 409       RunDaemonTests(instance, pnode)
 410       RunTest(qa_instance.TestInstanceRemove, instance)
 411       del instance
 412
 413     multinode_tests = [
 414       ('instance-add-drbd-disk',
 415        qa_instance.TestInstanceAddWithDrbdDisk),
 416     ]
 417
 418     for name, func in multinode_tests:
 419       if qa_config.TestEnabled(name):
 420         snode = qa_config.AcquireNode(exclude=pnode)
 421         try:
 422           instance = RunTest(func, pnode, snode)
 423           if qa_config.TestEnabled("cluster-verify"):
 424             RunTest(qa_cluster.TestClusterVerify)
 425           RunCommonInstanceTests(instance)
 426           if qa_config.TestEnabled('instance-convert-disk'):
 427             RunTest(qa_instance.TestInstanceShutdown, instance)
 428             RunTest(qa_instance.TestInstanceConvertDisk, instance, snode)
 429             RunTest(qa_instance.TestInstanceStartup, instance)
 430           RunExportImportTests(instance, pnode, snode)
 431           RunHardwareFailureTests(instance, pnode, snode)
 432           RunTest(qa_instance.TestInstanceRemove, instance)
 433           del instance
 434         finally:
 435           qa_config.ReleaseNode(snode)
 436
 437     if (qa_config.TestEnabled('instance-add-plain-disk') and
 438         qa_config.TestEnabled("instance-export")):
 439       for shutdown in [False, True]:
 440         instance = RunTest(qa_instance.TestInstanceAddWithPlainDisk, pnode)
 441         expnode = qa_config.AcquireNode(exclude=pnode)
 442         try:
 443           if shutdown:
 444             # Stop instance before exporting and removing it
 445             RunTest(qa_instance.TestInstanceShutdown, instance)
 446           RunTest(qa_instance.TestInstanceExportWithRemove, instance, expnode)
 447           RunTest(qa_instance.TestBackupList, expnode)
 448         finally:
 449           qa_config.ReleaseNode(expnode)
 450         del expnode
 451         del instance
 452
 453   finally:
 454     qa_config.ReleaseNode(pnode)
 455
 456   if qa_config.TestEnabled('create-cluster'):
 457     RunTest(qa_node.TestNodeRemoveAll)
 458
 459   if qa_config.TestEnabled('cluster-destroy'):
 460     RunTest(qa_cluster.TestClusterDestroy)
 461
 462
 463 if __name__ == '__main__':
 464   main()