Fix/test: improve test coverage

[codelformat]

🚀 Summary of Changes

🔗 Related Issues

🛠️ Type of Change

• 🐛 Bug fix (non-breaking change which fixes an issue)
• ✨ New feature (non-breaking change which adds functionality)
• 📝 Documentation update (changes to documentation only)
• 💥 Breaking change (fix or feature that would cause existing functionality to not work as expected)
• 🔧 Refactor/Optimization (code improvement without changing logic)

✅ Checklist

• I have read and followed the Contributing Guidelines.
• I have performed a self-review of my own code.
• My code follows the project's coding style/standards.
• I have added/updated necessary comments and documentation.
• (If new feature) I have added corresponding test cases.
• All local tests pass successfully.

🧪 Test Plan

📸 Screenshots