Add more javadocs to unit test