Add tests for comment's model