def test_parse_disks_values(self):
"""
Checks CloubedArgumentParser.parse_disks() returns the list of given
disk names if values are valid
"""
sys.argv = ["cloubed", "boot", "--domain", "toto", "--overwrite-disks", "disk1", "disk2"]
parser = CloubedArgumentParser(u"test_description")
parser.add_args()
parser.parse_args()
self.assertEqual(parser.parse_disks(), ["disk1", "disk2"])
def test_parse_disks_default(self):
"""
Checks CloubedArgumentParser.parse_disks() returns default value
False
"""
sys.argv = ["cloubed", "boot", "--domain", "toto"]
parser = CloubedArgumentParser(u"test_description")
parser.add_args()
parser.parse_args()
self.assertEqual(parser.parse_disks(), False)
def test_parse_disks_yes_no(self):
"""
Checks CloubedArgumentParser.parse_disks() returns True if parameter
value is yes and False if no
"""
expected_values = {"yes": True, "no": False}
for param, value in expected_values.items():
sys.argv = ["cloubed", "boot", "--domain", "toto", "--overwrite-disks", param]
parser = CloubedArgumentParser(u"test_description")
parser.add_args()
parser.parse_args()
self.assertEqual(parser.parse_disks(), value)
