module Selenium::WebDriver::DriverExtensions::Rotatable

def orientation
  bridge.screen_orientation.to_sym.downcase
end

def rotation=(orientation)
  unless ORIENTATIONS.include?(orientation)
    raise ArgumentError, "expected #{ORIENTATIONS.inspect}, got #{orientation.inspect}"
  end
  bridge.screen_orientation = orientation.to_s.upcase
end