test: Improve test coverage

[ysknsid25]

Describe the feature

Overall coverage is low.

May I write the test code? 🙋

Additional information

• Would you be willing to help implement this feature?