module ActionController::TestCase::Behavior::ClassMethods

def controller_class=(new_class)
  prepare_controller_class(new_class) if new_class
  self._controller_class = new_class
end