Feat: Add test for database

[NancyWei123]

Describe the feature

Test coverage is currently 51%, and database functions are not yet covered.

Add ScreenShots

Record

• I agree to follow this project's Code of Conduct
• I want to work on this issue

[github-actions]

👋 Thanks for opening this issue! The maintainers will review it shortly. Till then, do not open any PR. This is a strict rule we like to follow around here :)