add unit test coverage

TODO