Revision 75f53ffe test/ganeti.rapi.client_unittest.py

b/test/ganeti.rapi.client_unittest.py
418 418
    self.assertQuery("storage_type", ["lvm-pv"])
419 419
    self.assertQuery("output_fields", ["fields"])
420 420

  
421
    self.assertRaises(client.InvalidStorageType,
422
                      self.client.GetNodeStorageUnits,
423
                      "node-y", "floppy-disk", "fields")
424

  
425 421
  def testModifyNodeStorageUnits(self):
426 422
    self.rapi.AddResponse("14")
427 423
    self.assertEqual(14,
......
431 427
    self.assertQuery("storage_type", ["lvm-pv"])
432 428
    self.assertQuery("name", ["hda"])
433 429

  
434
    self.assertRaises(client.InvalidStorageType,
435
                      self.client.ModifyNodeStorageUnits,
436
                      "node-n", "floppy-disk", "hdc")
437

  
438 430
  def testRepairNodeStorageUnits(self):
439 431
    self.rapi.AddResponse("99")
440 432
    self.assertEqual(99, self.client.RepairNodeStorageUnits("node-z", "lvm-pv",
......
444 436
    self.assertQuery("storage_type", ["lvm-pv"])
445 437
    self.assertQuery("name", ["hda"])
446 438

  
447
    self.assertRaises(client.InvalidStorageType,
448
                      self.client.RepairNodeStorageUnits,
449
                      "node-n", "floppy-disk", "hdc")
450

  
451 439
  def testGetNodeTags(self):
452 440
    self.rapi.AddResponse("[\"fry\", \"bender\"]")
453 441
    self.assertEqual(["fry", "bender"], self.client.GetNodeTags("node-k"))

Also available in: Unified diff