ADD Unit Test for the existing APIs

[ankit2001]

[shwetagurnani]

I would like to work on it.
Thanks

[shwetagurnani]

/assign

[github-actions]

This issue has been assigned to @shwetagurnani!
It will become unassigned if it isn't closed within 14 days. A maintainer can also add the WIP label to prevent it from being unassigned.

[github-actions]

@shwetagurnani, this issue hasn't had any activity in 7 days. It will become unassigned in 7 days to make room for someone else to contribute.