Synnefo » snf-ganeti

def AcquireManyInstances(num, _cfg=None):

  """Return instances that are not in use.

  @type num: int

  @param num: Number of instances; can be 0.

  @rtype: list of instances

  @return: C{num} different instances

  """

  if _cfg is None:

    cfg = GetConfig()

  else:

    cfg = _cfg

  # Filter out unwanted instances

  instances = filter(lambda inst: not inst.used, cfg["instances"])

  if len(instances) < num:

    raise qa_error.OutOfInstancesError(

      "Not enough instances left (%d needed, %d remaining)" %

      (num, len(instances))

    )

  instances = []

  try:

    for _ in range(0, num):

      n = AcquireInstance(_cfg=cfg)

      instances.append(n)

  except qa_error.OutOfInstancesError:

    ReleaseManyInstances(instances)

    raise

  return instances

def ReleaseManyInstances(instances):

  for instance in instances:

    instance.Release()

from ganeti.rapi.client import GanetiApiError, NODE_EVAC_PRI, NODE_EVAC_SEC

def CreateDRBDInstance(client, node_one, node_two, instance_name):

  """ Creates a DRBD-enabled instance on the given nodes.

  """

  Finish(client, client.CreateInstance,

         "create", instance_name, "drbd", [{"size": "1000"}], [{}],

         os="debian-image", pnode=node_one, snode=node_two)

def TestInstanceMigrations(client, node_one, node_two, node_three,

                           instance_name):

  """ Test various operations related to migrating instances.

  @type node_one string

  @param node_one The name of a node in the cluster.

  @type node_two string

  @param node_two The name of another node in the cluster.

  @type node_three string

  @param node_three The name of yet another node in the cluster.

  @type instance_name string

  @param instance_name An instance name that can be used.

  """

  CreateDRBDInstance(client, node_one, node_two, instance_name)

  Finish(client, client.FailoverInstance, instance_name)

  Finish(client, client.DeleteInstance, instance_name)

  CreateDRBDInstance(client, node_one, node_two, instance_name)

  Finish(client, client.EvacuateNode,

         node_two, early_release=False, mode=NODE_EVAC_SEC,

         remote_node=node_three)

  Finish(client, client.DeleteInstance, instance_name)

  CreateDRBDInstance(client, node_one, node_two, instance_name)

  Finish(client, client.EvacuateNode,

         node_one, early_release=False, mode=NODE_EVAC_PRI, iallocator="hail")

  Finish(client, client.DeleteInstance, instance_name)

  CreateDRBDInstance(client, node_one, node_two, instance_name)

  Finish(client, client.MigrateInstance,

         instance_name, cleanup=True, target_node=node_two)

  Finish(client, client.DeleteInstance, instance_name)

  CreateDRBDInstance(client, node_one, node_two, instance_name)

  Finish(client, client.MigrateNode,

         node_one, iallocator="hail", mode="non-live")

  Finish(client, client.DeleteInstance, instance_name)

  CreateDRBDInstance(client, node_one, node_two, instance_name)

  Finish(client, client.MigrateNode,

         node_one, target_node=node_two, mode="non-live")

  Finish(client, client.DeleteInstance, instance_name)

  instances = qa_config.AcquireManyInstances(2)

  TestSingleInstance(client, instances[0].name, instances[1].name,

  qa_config.ReleaseManyInstances(instances)

  nodes = qa_config.AcquireManyNodes(3)

  instance = qa_config.AcquireInstance()

  TestInstanceMigrations(client, nodes[0].primary, nodes[1].primary,

                         nodes[2].primary, instance.name)

  instance.Release()

  qa_config.ReleaseManyNodes(nodes)