Fix and write all the test cases

[Rupeshiya]

Describe the enhancement you'd like
As of now, the test cases are failing and also other test cases need to written for left API's

Describe approaches if you have thought of any
Write and fix the test case for all the API's