test: Improve test coverage

### Describe the feature

Overall coverage is low.

![Image](https://github.com/user-attachments/assets/ddaf293d-cbf3-4f3b-bea5-c3777fd35cfe)

May I write the test code? 🙋

### Additional information

- [x] Would you be willing to help implement this feature?