diff --git a/dropbot/bot.py b/dropbot/bot.py index c0c7014..9f240f4 100644 --- a/dropbot/bot.py +++ b/dropbot/bot.py @@ -174,7 +174,7 @@ class DropBot(ClientXMPP): if item.strip() == '': return 'Usage: !price ' 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()) diff --git a/tests/test_bot.py b/tests/test_bot.py index a8af6cb..da14ec1 100644 --- a/tests/test_bot.py +++ b/tests/test_bot.py @@ -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)"))