From 41f8797a4caba2e8c85c0bb7f9cc112481e01bbf Mon Sep 17 00:00:00 2001
From: Josh Wu <joshwu@google.com>
Date: Sun, 20 Jul 2025 18:25:11 +0800
Subject: [PATCH] Add more test cases for custom packets

---
 tests/hci_test.py | 42 +++++++++++++++++++++++++++++++++++++++---
 1 file changed, 39 insertions(+), 3 deletions(-)

diff --git a/tests/hci_test.py b/tests/hci_test.py
index 3ca5a93..4531588 100644
--- a/tests/hci_test.py
+++ b/tests/hci_test.py
@@ -238,14 +238,50 @@ def test_HCI_Command():
     command = hci.HCI_Command(op_code=0x5566, parameters=bytes.fromhex('AABBCC'))
     basic_check(command)
 
+
+# -----------------------------------------------------------------------------
+def test_custom_command():
     @hci.HCI_Command.command
     class CustomCommand(hci.HCI_Command):
         op_code = 0x7788
         name = 'Custom Command'
 
-    basic_check(CustomCommand())
-    assert CustomCommand().op_code == 0x7788
-    assert CustomCommand().name == 'Custom Command'
+    command = CustomCommand()
+    basic_check(command)
+    parsed = hci.HCI_Packet.from_bytes(bytes(command))
+    assert isinstance(parsed, CustomCommand)
+    assert parsed.op_code == 0x7788
+    assert parsed.name == 'Custom Command'
+
+
+# -----------------------------------------------------------------------------
+def test_custom_event():
+    @hci.HCI_Event.event
+    class CustomEvent(hci.HCI_Event):
+        event_code = 0x99
+        name = 'Custom Event'
+
+    event = CustomEvent()
+    basic_check(event)
+    parsed = hci.HCI_Packet.from_bytes(bytes(event))
+    assert isinstance(parsed, CustomEvent)
+    assert parsed.event_code == 0x99
+    assert parsed.name == 'Custom Event'
+
+
+# -----------------------------------------------------------------------------
+def test_custom_le_meta_event():
+    @hci.HCI_LE_Meta_Event.event
+    class CustomEvent(hci.HCI_LE_Meta_Event):
+        subevent_code = 0xFF
+        name = 'Custom Extended Event'
+
+    event = CustomEvent()
+    basic_check(event)
+    parsed = hci.HCI_Packet.from_bytes(bytes(event))
+    assert isinstance(parsed, CustomEvent)
+    assert parsed.subevent_code == 0xFF
+    assert parsed.name == 'Custom Extended Event'
 
 
 # -----------------------------------------------------------------------------