Revision 4e771a95 test/py/ganeti.cmdlib.cluster_unittest.py

b/test/py/ganeti.cmdlib.cluster_unittest.py
32 32
import mock
33 33

  
34 34

  
35
class TestCheckIpolicy(unittest.TestCase):
36

  
37
  def setUp(self):
38
    unittest.TestCase.setUp(self)
39

  
40
  def testAllTemplatesEnabled(self):
41
    allowed_disk_templates = [constants.DT_PLAIN]
42
    ipolicy = {constants.IPOLICY_DTS: allowed_disk_templates}
43
    enabled_disk_templates = [constants.DT_PLAIN, constants.DT_DRBD8]
44
    cluster.LUClusterSetParams._CheckIpolicyVsDiskTemplates(
45
        ipolicy, enabled_disk_templates)
46

  
47
  def testSomeTemplatesUnenabled(self):
48
    allowed_disk_templates = [constants.DT_PLAIN, constants.DT_DISKLESS]
49
    ipolicy = {constants.IPOLICY_DTS: allowed_disk_templates}
50
    enabled_disk_templates = [constants.DT_PLAIN, constants.DT_DRBD8]
51
    self.assertRaises(
52
        errors.OpPrereqError, 
53
        cluster.LUClusterSetParams._CheckIpolicyVsDiskTemplates,
54
        ipolicy, enabled_disk_templates)
55

  
56

  
57 35
class TestCheckFileStoragePath(unittest.TestCase):
58 36

  
59 37
  def setUp(self):

Also available in: Unified diff