feat: add test cases for comments