module ActionDispatch::IntegrationTest::Behavior

def app
  super || self.class.app
end

def document_root_element
  html_document.root
end