Synnefo » snf-ganeti

  def MigrateInstance(self, instance, mode=None, cleanup=None):

    """Starts up an instance.

    @type instance: string

    @param instance: Instance name

    @type mode: string

    @param mode: Migration mode

    @type cleanup: bool

    @param cleanup: Whether to clean up a previously failed migration

    """

    body = {}

    if mode is not None:

      body["mode"] = mode

    if cleanup is not None:

      body["cleanup"] = cleanup

    return self._SendRequest(HTTP_PUT,

                             ("/%s/instances/%s/migrate" %

                              (GANETI_RAPI_VERSION, instance)), None, body)

  def testMigrateInstanceDefaults(self):

    self.rapi.AddResponse("24873")

    job_id = self.client.MigrateInstance("inst91")

    self.assertEqual(job_id, 24873)

    self.assertHandler(rlib2.R_2_instances_name_migrate)

    self.assertItems(["inst91"])

    data = serializer.LoadJson(self.rapi.GetLastRequestData())

    self.assertFalse(data)

  def testMigrateInstance(self):

    for mode in constants.HT_MIGRATION_MODES:

      for cleanup in [False, True]:

        self.rapi.AddResponse("31910")

        job_id = self.client.MigrateInstance("inst289", mode=mode,

                                             cleanup=cleanup)

        self.assertEqual(job_id, 31910)

        self.assertHandler(rlib2.R_2_instances_name_migrate)

        self.assertItems(["inst289"])

        data = serializer.LoadJson(self.rapi.GetLastRequestData())

        self.assertEqual(len(data), 2)

        self.assertEqual(data["mode"], mode)

        self.assertEqual(data["cleanup"], cleanup)