add unit test for code coverage