Synnefo » snf-ganeti

def RunEnvTests():

  """Run several environment tests.

  if not qa_config.TestEnabled('env'):

    return

  RunTest(qa_env.TestSshConnection)

  RunTest(qa_env.TestIcmpPing)

  RunTest(qa_env.TestGanetiCommands)

def SetupCluster():

  """Initializes the cluster.

  """

  if qa_config.TestEnabled('node-info'):

    RunTest(qa_node.TestNodeInfo)

def RunClusterTests():

  """Runs tests related to gnt-cluster.

  """

def RunOsTests():

  """Runs all tests related to gnt-os.

  """

  if not qa_config.TestEnabled('os'):

    return

  RunTest(qa_os.TestOsList)

  RunTest(qa_os.TestOsDiagnose)

  RunTest(qa_os.TestOsValid)

  RunTest(qa_os.TestOsInvalid)

  RunTest(qa_os.TestOsPartiallyValid)

def RunCommonInstanceTests(instance):

  """Runs a few tests that are common to all disk types.

  """

  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)

  if qa_config.TestEnabled('instance-info'):

    RunTest(qa_instance.TestInstanceInfo, instance)

  if qa_config.TestEnabled('instance-reinstall'):

    RunTest(qa_instance.TestInstanceShutdown, instance)

    RunTest(qa_instance.TestInstanceReinstall, instance)

    RunTest(qa_instance.TestInstanceStartup, instance)

  if qa_config.TestEnabled('node-volumes'):

    RunTest(qa_node.TestNodeVolumes)

def RunExportImportTests(instance, pnode):

  """Tries to export and import the instance.

  """

  if qa_config.TestEnabled('instance-export'):

    expnode = qa_config.AcquireNode(exclude=pnode)

    try:

      name = RunTest(qa_instance.TestInstanceExport, instance, expnode)

      RunTest(qa_instance.TestBackupList, expnode)

      if qa_config.TestEnabled('instance-import'):

        newinst = qa_config.AcquireInstance()

          RunTest(qa_instance.TestInstanceImport, pnode, newinst,

                  expnode, name)

          RunTest(qa_instance.TestInstanceRemove, newinst)

          qa_config.ReleaseInstance(newinst)

    finally:

      qa_config.ReleaseNode(expnode)

def RunDaemonTests(instance, pnode):

  """Test the ganeti-watcher script.

  """

  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()

    if automatic_restart:

      RunTest(qa_daemon.TestInstanceAutomaticRestart, pnode, instance)

    if consecutive_failures:

      RunTest(qa_daemon.TestInstanceConsecutiveFailures, node, instance)

def RunHardwareFailureTests(instance, pnode, snode):

  """Test cluster internal hardware failure recovery.

  """

  if qa_config.TestEnabled('instance-failover'):

    RunTest(qa_instance.TestInstanceFailover, instance)

  if qa_config.TestEnabled('node-evacuate'):

    RunTest(qa_node.TestNodeEvacuate, pnode, snode)

  if qa_config.TestEnabled('node-failover'):

    RunTest(qa_node.TestNodeFailover, pnode, snode)

  if qa_config.TestEnabled('instance-disk-failure'):

    RunTest(qa_instance.TestInstanceMasterDiskFailure,

            instance, pnode, snode)

    RunTest(qa_instance.TestInstanceSecondaryDiskFailure,

            instance, pnode, snode)

def main():

  """Main program.

  """

  parser = 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")

  (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 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)

  qa_config.Load(config_file)

  RunTest(qa_other.UploadKnownHostsFile, known_hosts_file)

  RunEnvTests()

  SetupCluster()

  RunClusterTests()

  RunOsTests()

  pnode = qa_config.AcquireNode()

  try:

    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 qa_config.TestEnabled('instance-add-local-mirror-disk'):

      instance = RunTest(qa_instance.TestInstanceAddWithLocalMirrorDisk, pnode)

      RunCommonInstanceTests(instance)

      RunExportImportTests(instance, pnode)

      RunTest(qa_instance.TestInstanceRemove, instance)

      del instance

    if qa_config.TestEnabled('instance-add-remote-raid-disk'):

      snode = qa_config.AcquireNode(exclude=pnode)

      try:

        instance = RunTest(qa_instance.TestInstanceAddWithRemoteRaidDisk,

                           pnode, snode)

        RunCommonInstanceTests(instance)

        RunExportImportTests(instance, pnode)

        RunHardwareFailureTests(instance, pnode, snode)

        qa_config.ReleaseNode(snode)

    qa_config.ReleaseNode(pnode)