Revision 0356a13d test/ganeti.cmdlib_unittest.py

b/test/ganeti.cmdlib_unittest.py
1039 1039
      self.assertEqual(disk.mode, disk_info[idx][constants.IDISK_MODE])
1040 1040
      self.assertTrue(disk.children is None)
1041 1041

  
1042
    self.assertEqual(map(operator.attrgetter("iv_name"), result),
1043
      ["disk/%s" % i for i in range(base_index, base_index + len(disk_info))])
1042
    self._CheckIvNames(result, base_index, base_index + len(disk_info))
1043
    cmdlib._UpdateIvNames(base_index, result)
1044
    self._CheckIvNames(result, base_index, base_index + len(disk_info))
1044 1045

  
1045 1046
    return result
1046 1047

  
1048
  def _CheckIvNames(self, disks, base_index, end_index):
1049
    self.assertEqual(map(operator.attrgetter("iv_name"), disks),
1050
                     ["disk/%s" % i for i in range(base_index, end_index)])
1051

  
1047 1052
  def testPlain(self):
1048 1053
    disk_info = [{
1049 1054
      constants.IDISK_SIZE: 1024,
......
1198 1203
      self.assertEqual(disk.children[0].size, disk.size)
1199 1204
      self.assertEqual(disk.children[1].size, cmdlib.DRBD_META_SIZE)
1200 1205

  
1201
    self.assertEqual(map(operator.attrgetter("iv_name"), result),
1202
                     ["disk/0", "disk/1", "disk/2"])
1206
    self._CheckIvNames(result, 0, len(disk_info))
1207
    cmdlib._UpdateIvNames(0, result)
1208
    self._CheckIvNames(result, 0, len(disk_info))
1203 1209

  
1204 1210
    self.assertEqual(map(operator.attrgetter("logical_id"), result), [
1205 1211
      ("node1334.example.com", "node12272.example.com",

Also available in: Unified diff