[test] Adding more tests for coverage