47 lines
1.4 KiB
47 lines
1.4 KiB
from unittest import TestCase
|
|
|
|
from disco.client import ClientConfig, Client
|
|
from disco.bot.bot import Bot
|
|
from disco.bot.command import Command
|
|
|
|
|
|
class MockBot(Bot):
|
|
@property
|
|
def commands(self):
|
|
return getattr(self, '_commands', [])
|
|
|
|
|
|
class TestBot(TestCase):
|
|
def setUp(self):
|
|
self.client = Client(ClientConfig(
|
|
{'config': 'TEST_TOKEN'}
|
|
))
|
|
self.bot = MockBot(self.client)
|
|
|
|
def test_command_abbreviation(self):
|
|
groups = ['config', 'copy', 'copez', 'copypasta']
|
|
result = self.bot.compute_group_abbrev(groups)
|
|
self.assertDictEqual(result, {
|
|
'config': 'con',
|
|
'copypasta': 'copy',
|
|
'copez': 'cope',
|
|
})
|
|
|
|
def test_command_abbreivation_conflicting(self):
|
|
groups = ['cat', 'cap', 'caz', 'cas']
|
|
result = self.bot.compute_group_abbrev(groups)
|
|
self.assertDictEqual(result, {})
|
|
|
|
def test_many_commands(self):
|
|
self.bot._commands = [
|
|
Command(None, None, 'test{}'.format(i), '<test:str>')
|
|
for i in range(1000)
|
|
]
|
|
|
|
self.bot.compute_command_matches_re()
|
|
match = self.bot.command_matches_re.match('test5 123')
|
|
self.assertNotEqual(match, None)
|
|
|
|
match = self.bot._commands[0].compiled_regex.match('test0 123 456')
|
|
self.assertEqual(match.group(1).strip(), 'test0')
|
|
self.assertEqual(match.group(2).strip(), '123 456')
|
|
|