Implement unit test coverage
Test coverage for integration tests