Complete the unit testing suite for APIs

[mish24]

I would like to add more tests to improve test coverage

[djmgit]

@mish24 go ahead

[vibhcool]

@mish24 , It is nice, go ahead. please create a specific issue for the test you are going to add to keep the PRs small :)

[mish24]

@vibhcool I'll send the PRs and simultaneously create issues to reference them!