class CompleterTestCase(unittest.TestCase):
def setUp(self):
self.pushbullet = PushBullet()
self.completer = CommandCompleter(self.pushbullet)
def test_command_list(self):
document = MockDocument('')
words = [i.text for i in self.completer.get_completions(document, None)]
self.assertIn('help', words)
self.assertIn('quit', words)
self.assertIn('note', words)
self.assertIn('link', words)
self.assertIn('devices', words)
def test_device_list(self):
document = MockDocument('note ')
words = [i.text for i in self.completer.get_completions(document, None)]
devices = [i.name.lower() for i in self.pushbullet.devices]
for d in devices:
self.assertIn(d, words)
self.assertIn('all', words)
def test_device_list_partial_match(self):
document = MockDocument('note al')
words = [i.text for i in self.completer.get_completions(document, None)]
self.assertIn('all', words)
self.assertLess(len(words), len(self.pushbullet.devices))
def test_note_message(self):
document = MockDocument('note all dsfewfg')
words = [i.display for i in self.completer.get_completions(document, None)]
self.assertEqual(['Message to push'], words)
def test_link_url_message(self):
document = MockDocument('link all http://xkcd')
words = [i.display for i in self.completer.get_completions(document, None)]
self.assertEqual(['URL to push'], words)
def test_link_url_optional_body(self):
document = MockDocument('link all http://xkcd.com ')
words = [i.display for i in self.completer.get_completions(document, None)]
self.assertEqual(['[Optional] Message to push'], words)
