Add Unit test coverage

[THEBOSS0369]

Currently, the project lacks sufficient unit test coverage, which is essential for ensuring code reliability, This issue aims to improve the overall test coverage and maintainability of the codebase.

Read This Comment in the PR For more info.