23 lines
638 B
Python
23 lines
638 B
Python
import unittest
|
|
|
|
from core.commands import CommandRegistry
|
|
|
|
|
|
class CommandRegistryTests(unittest.TestCase):
|
|
def test_parse(self):
|
|
registry = CommandRegistry(prefix=".")
|
|
parsed = registry.parse(".ping 1 2")
|
|
self.assertEqual(parsed.name, "ping")
|
|
self.assertEqual(parsed.args, ["1", "2"])
|
|
|
|
def test_flags(self):
|
|
registry = CommandRegistry(prefix=".")
|
|
parsed = registry.parse(".cmd --foo=bar -ab")
|
|
self.assertEqual(parsed.flags["foo"], "bar")
|
|
self.assertTrue(parsed.flags["a"])
|
|
self.assertTrue(parsed.flags["b"])
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|