Revision e8936ef7 test/py/ganeti.masterd.iallocator_unittest.py

b/test/py/ganeti.masterd.iallocator_unittest.py
217 217
    self.assertEqual(self.free_storage_lvm, free_disk)
218 218
    self.assertEqual(self.total_storage_lvm, total_disk)
219 219

  
220
  def testComputeStorageDataFromSpaceInfoByTemplate(self):
221
    disk_template = constants.DT_FILE
222
    node_name = "mynode"
223
    (total_disk, free_disk, total_spindles, free_spindles) = \
224
        iallocator.IAllocator._ComputeStorageDataFromSpaceInfoByTemplate(
225
            self.space_info, node_name, disk_template)
226
    self.assertEqual(self.free_storage_file, free_disk)
227
    self.assertEqual(self.total_storage_file, total_disk)
228

  
229
  def testComputeStorageDataFromSpaceInfoByTemplateLvm(self):
230
    disk_template = constants.DT_PLAIN
231
    node_name = "mynode"
232
    (total_disk, free_disk, total_spindles, free_spindles) = \
233
        iallocator.IAllocator._ComputeStorageDataFromSpaceInfoByTemplate(
234
            self.space_info, node_name, disk_template)
235
    self.assertEqual(self.free_storage_lvm, free_disk)
236
    self.assertEqual(self.total_storage_lvm, total_disk)
237

  
238
  def testComputeStorageDataFromSpaceInfoByTemplateNoReport(self):
239
    disk_template = constants.DT_DISKLESS
240
    node_name = "mynode"
241
    (total_disk, free_disk, total_spindles, free_spindles) = \
242
        iallocator.IAllocator._ComputeStorageDataFromSpaceInfoByTemplate(
243
            self.space_info, node_name, disk_template)
244
    self.assertEqual(0, free_disk)
245
    self.assertEqual(0, total_disk)
220 246

  
221 247
if __name__ == "__main__":
222 248
  testutils.GanetiTestProgram()

Also available in: Unified diff