Synnefo » snf-ganeti

# 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

# the Free Software Foundation; either version 2 of the License, or

# (at your option) any later version.

#

# This program is distributed in the hope that it will be useful, but

# WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU

# General Public License for more details.

#

# You should have received a copy of the GNU General Public License

# along with this program; if not, write to the Free Software

# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA

# 02110-1301, USA.

"""Cluster related QA tests.

"""

import tempfile

from ganeti import utils

import qa_config

import qa_utils

import qa_error

from qa_utils import AssertEqual, StartSSH

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)

def _CheckFileOnAllNodes(filename, content):

  """Verifies the content of the given file on all nodes.

  """

  cmd = utils.ShellQuoteArgs(["cat", filename])

  for node in qa_config.get('nodes'):

    AssertEqual(qa_utils.GetCommandOutput(node['primary'], cmd),

                content)

def TestClusterInit():

  """gnt-cluster init"""

  master = qa_config.GetMasterNode()

  cmd = ['gnt-cluster', 'init']

  if master.get('secondary', None):

    cmd.append('--secondary-ip=%s' % master['secondary'])

  bridge = qa_config.get('bridge', None)

  if bridge:

    cmd.append('--bridge=%s' % bridge)

    cmd.append('--master-netdev=%s' % bridge)

  cmd.append(qa_config.get('name'))

  AssertEqual(StartSSH(master['primary'],

                       utils.ShellQuoteArgs(cmd)).wait(), 0)

def TestClusterVerify():

  """gnt-cluster verify"""

  master = qa_config.GetMasterNode()

  cmd = ['gnt-cluster', 'verify']

  AssertEqual(StartSSH(master['primary'],

                       utils.ShellQuoteArgs(cmd)).wait(), 0)

def TestClusterInfo():

  """gnt-cluster info"""

  master = qa_config.GetMasterNode()

  cmd = ['gnt-cluster', 'info']

  AssertEqual(StartSSH(master['primary'],

                       utils.ShellQuoteArgs(cmd)).wait(), 0)

def TestClusterGetmaster():

  """gnt-cluster getmaster"""

  master = qa_config.GetMasterNode()

  cmd = ['gnt-cluster', 'getmaster']

  AssertEqual(StartSSH(master['primary'],

                       utils.ShellQuoteArgs(cmd)).wait(), 0)

def TestClusterVersion():

  """gnt-cluster version"""

  master = qa_config.GetMasterNode()

  cmd = ['gnt-cluster', 'version']

  AssertEqual(StartSSH(master['primary'],

                       utils.ShellQuoteArgs(cmd)).wait(), 0)

def TestClusterBurnin():

  """Burnin"""

  master = qa_config.GetMasterNode()

  # Get as many instances as we need

  instances = []

  try:

    num = qa_config.get('options', {}).get('burnin-instances', 1)

    for _ in xrange(0, num):

      instances.append(qa_config.AcquireInstance())

  except qa_error.OutOfInstancesError:

    print "Not enough instances, continuing anyway."

  if len(instances) < 1:

    raise qa_error.Error("Burnin needs at least one instance")

  # Run burnin

  try:

    script = qa_utils.UploadFile(master['primary'], '../tools/burnin')

    try:

      cmd = [script,

             '--os=%s' % qa_config.get('os'),

             '--os-size=%s' % qa_config.get('os-size'),

             '--swap-size=%s' % qa_config.get('swap-size')]

      cmd += [inst['name'] for inst in instances]

      AssertEqual(StartSSH(master['primary'],

                           utils.ShellQuoteArgs(cmd)).wait(), 0)

    finally:

      cmd = ['rm', '-f', script]

      AssertEqual(StartSSH(master['primary'],

                           utils.ShellQuoteArgs(cmd)).wait(), 0)

  finally:

    for inst in instances:

      qa_config.ReleaseInstance(inst)

def TestClusterMasterFailover():

  """gnt-cluster masterfailover"""

  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)

  finally:

    qa_config.ReleaseNode(failovermaster)

def TestClusterCopyfile():

  """gnt-cluster copyfile"""

  master = qa_config.GetMasterNode()

  uniqueid = utils.NewUUID()

  # Create temporary file

  f = tempfile.NamedTemporaryFile()

  f.write(uniqueid)

  f.flush()

  f.seek(0)

  # Upload file to master node

  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)

    _CheckFileOnAllNodes(testname, uniqueid)

  finally:

    _RemoveFileFromAllNodes(testname)

def TestClusterCommand():

  """gnt-cluster command"""

  master = qa_config.GetMasterNode()

  uniqueid = utils.NewUUID()

  rfile = "/tmp/gnt%s" % utils.NewUUID()

  rcmd = utils.ShellQuoteArgs(['echo', '-n', uniqueid])

  cmd = utils.ShellQuoteArgs(['gnt-cluster', 'command',

                              "%s >%s" % (rcmd, rfile)])

  try:

    AssertEqual(StartSSH(master['primary'], cmd).wait(), 0)

    _CheckFileOnAllNodes(rfile, uniqueid)

  finally:

    _RemoveFileFromAllNodes(rfile)

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)