Tests - Improve code coverage

[bhoopesh369]

• Currently the code coverage is less than 50%
• We need people to contribute to cover more lines of code and thus improve codcov score

[bhoopesh369]

refer - codecov report