From ba872d1e48ad229903316fc30e61cebe9c115258 Mon Sep 17 00:00:00 2001
From: "Danilo M." <danix@danix.xyz>
Date: Tue, 30 Jun 2026 10:59:43 +0200
Subject: feat: command registry and context

---
 tests/unit/test_registry.py | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)
 create mode 100644 tests/unit/test_registry.py

(limited to 'tests/unit')

diff --git a/tests/unit/test_registry.py b/tests/unit/test_registry.py
new file mode 100644
index 0000000..f30a5df
--- /dev/null
+++ b/tests/unit/test_registry.py
@@ -0,0 +1,23 @@
+import unittest
+from firefly_cli import registry
+
+class TestRegistry(unittest.TestCase):
+    def setUp(self):
+        registry._COMMANDS.clear()
+
+    def test_command_decorator_registers(self):
+        @registry.command("tx add", help="add a tx")
+        def handler(args, ctx):
+            return 0
+        cmds = registry.all_commands()
+        self.assertEqual(len(cmds), 1)
+        self.assertEqual(cmds[0].name, "tx add")
+        self.assertIs(cmds[0].handler, handler)
+
+    def test_add_arguments_callback_stored(self):
+        def args_cb(p):
+            p.add_argument("name")
+        @registry.command("account get", args=args_cb)
+        def handler(args, ctx):
+            return 0
+        self.assertIs(registry.all_commands()[0].args, args_cb)
-- 
cgit v1.2.3