Refactor comment features test