Fix tests and raise exception if a non parameter modifier is supplied with an argument.

This commit is contained in:
2014-08-04 18:07:05 +01:00
parent feadde428e
commit c9d0569de1
2 changed files with 5 additions and 3 deletions

View File

@@ -80,6 +80,8 @@ class ZKillboard(object):
raise InvalidModifier('%s is a unknown modifier' % mod_name)
if args is None and mod_name.lower() in ZKillboard._MODIFIERS_ARGUMENT:
raise InvalidModifier('%s requires a argument' % mod_name)
if args and mod_name.lower() not in ZKillboard._MODIFIERS_ARGUMENT:
raise InvalidModifier('%s does not require an argument' % mod_name)
if args and isinstance(args, list) and len(args) > 10:
raise ValueError('No more than 10 IDs are allowed at a time')
x = copy.deepcopy(self)

View File

@@ -57,9 +57,9 @@ class TestZKillboard(unittest.TestCase):
self.assertTrue(len(x._modifiers) == 1)
def test_xml_format_set(self):
self.kb.xml()
self.assertTrue(len(self.kb._modifiers), 1)
self.assertTrue(self.kb._xml_format)
x = self.kb.xml()
self.assertTrue(len(x._modifiers), 1)
self.assertTrue(x._xml_format)
def tearDown(self):
pass