Add comment junit tests