Fix up looking for plex.

This commit is contained in:
Rob
2015-06-04 10:45:12 +01:00
parent 35da7264c7
commit 034fb3a2dd
2 changed files with 6 additions and 2 deletions

View File

@@ -174,7 +174,7 @@ class DropBot(ClientXMPP):
if item.strip() == '':
return 'Usage: !price <item>'
if item.lower() == 'plex':
item = '30 Day'
return (u"29668", u"30 Day Pilot's License Extension (PLEX)")
types = dict([(i, v) for i, v in self.types.iteritems() if item.lower() in v.lower()])
if len(types) == 0:
return "No items named {} found".format(item)
@@ -184,6 +184,7 @@ class DropBot(ClientXMPP):
return (i, v)
else:
if len(types) > 10:
import pdb; pdb.set_trace()
return "More than 10 items found, please narrow down what you want."
return "Did you mean: {}?".format(
', '.join(types.itervalues())

View File

@@ -150,8 +150,11 @@ class DropBotTestCase(TestCase):
def test_cmd_rageping(self):
pass
def test_carnyx_items(self):
def test_jackdaw(self):
"""
The items in the Carnyx release can be found.
"""
self.assertEqual(self.bot._item_picker("Jackdaw"), (u'34828', u'Jackdaw'))
def test_carnyx_plex(self):
self.assertEqual(self.bot._item_picker("plex"), (u"29668", "30 Day Pilot's License Extension (PLEX)"))