class RuboCop::Cop::Style::ObjectThen

def preferred_method(node)
  case style
  when :then
    node.method?(:yield_self)
  when :yield_self
    node.method?(:then)
  else
    false
  end
end