Revision 5a904197 test/py/ganeti.utils.storage_unittest.py

b/test/py/ganeti.utils.storage_unittest.py
61 61
    (storage_type, storage_key) = \
62 62
        storage._GetDefaultStorageUnitForDiskTemplate(self._cfg,
63 63
                                                      constants.DT_SHARED_FILE)
64
    self.assertEqual(storage_type, constants.ST_FILE)
64
    self.assertEqual(storage_type, constants.ST_SHARED_FILE)
65 65
    self.assertEqual(storage_key, self._cluster.shared_file_storage_dir)
66 66

  
67 67
  def testGetDefaultStorageUnitForDiskTemplateDiskless(self):
......
80 80
    self._cfg = mock.Mock()
81 81

  
82 82
  def testGetStorageUnits(self):
83
    disk_templates = constants.DTS_FILEBASED
83
    disk_templates = constants.DTS_FILEBASED - frozenset(
84
      storage.GetDiskTemplatesOfStorageTypes(constants.ST_SHARED_FILE)
85
    )
84 86
    storage_units = storage.GetStorageUnits(self._cfg, disk_templates)
85 87
    self.assertEqual(len(storage_units), len(disk_templates))
86 88

  

Also available in: Unified diff