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()