def setUp(self): super(MarkFlagsAsMutualExclusiveTest, self).setUp() self.flag_values = _flagvalues.FlagValues() _defines.DEFINE_string('flag_one', None, 'flag one', flag_values=self.flag_values) _defines.DEFINE_string('flag_two', None, 'flag two', flag_values=self.flag_values) _defines.DEFINE_string('flag_three', None, 'flag three', flag_values=self.flag_values) _defines.DEFINE_integer('int_flag_one', None, 'int flag one', flag_values=self.flag_values) _defines.DEFINE_integer('int_flag_two', None, 'int flag two', flag_values=self.flag_values) _defines.DEFINE_multi_string('multi_flag_one', None, 'multi flag one', flag_values=self.flag_values) _defines.DEFINE_multi_string('multi_flag_two', None, 'multi flag two', flag_values=self.flag_values)
def test_multi_string_default_none(self): fv = _flagvalues.FlagValues() _defines.DEFINE_multi_string('foo', None, 'help', flag_values=fv) fv.mark_as_parsed() self.assertEqual(None, fv.foo) fv(['', '--foo=aa']) self.assertEqual(['aa'], fv.foo) fv.unparse_flags() self.assertEqual(None, fv['foo'].value) fv(['', '--foo=bb', '--foo=cc']) self.assertEqual(['bb', 'cc'], fv.foo) fv.unparse_flags() self.assertEqual(None, fv['foo'].value)
def test_multi_string_default_list(self): fv = _flagvalues.FlagValues() _defines.DEFINE_multi_string( 'foo', ['xx', 'yy', 'zz'], 'help', flag_values=fv) expected_default = ['xx', 'yy', 'zz'] fv.mark_as_parsed() self.assertEqual(expected_default, fv.foo) fv(['', '--foo=aa']) self.assertEqual(['aa'], fv.foo) fv.unparse_flags() self.assertEqual(expected_default, fv['foo'].value) fv(['', '--foo=bb', '--foo=cc']) self.assertEqual(['bb', 'cc'], fv.foo) fv.unparse_flags() self.assertEqual(expected_default, fv['foo'].value)