mirror of
				https://bitbucket.org/chromiumembedded/cef
				synced 2025-06-05 21:39:12 +02:00 
			
		
		
		
	
		
			
				
	
	
		
			380 lines
		
	
	
		
			13 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			380 lines
		
	
	
		
			13 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
// Copyright (c) 2012 The Chromium Embedded Framework Authors. All rights
 | 
						|
// reserved. Use of this source code is governed by a BSD-style license that
 | 
						|
// can be found in the LICENSE file.
 | 
						|
 | 
						|
#include "include/cef_dom.h"
 | 
						|
#include "tests/ceftests/test_handler.h"
 | 
						|
#include "tests/gtest/include/gtest/gtest.h"
 | 
						|
#include "tests/shared/renderer/client_app_renderer.h"
 | 
						|
 | 
						|
using client::ClientAppRenderer;
 | 
						|
 | 
						|
namespace {
 | 
						|
 | 
						|
const char* kTestUrl = "https://tests/DOMTest.Test";
 | 
						|
const char* kTestMessage = "DOMTest.Message";
 | 
						|
 | 
						|
enum DOMTestType {
 | 
						|
  DOM_TEST_STRUCTURE,
 | 
						|
  DOM_TEST_MODIFY,
 | 
						|
};
 | 
						|
 | 
						|
class TestDOMVisitor : public CefDOMVisitor {
 | 
						|
 public:
 | 
						|
  explicit TestDOMVisitor(CefRefPtr<CefBrowser> browser, DOMTestType test_type)
 | 
						|
      : browser_(browser), test_type_(test_type) {}
 | 
						|
 | 
						|
  void TestHeadNodeStructure(CefRefPtr<CefDOMNode> headNode) {
 | 
						|
    EXPECT_TRUE(headNode.get());
 | 
						|
    EXPECT_TRUE(headNode->IsElement());
 | 
						|
    EXPECT_FALSE(headNode->IsText());
 | 
						|
    EXPECT_EQ(headNode->GetName(), "HEAD");
 | 
						|
    EXPECT_EQ(headNode->GetElementTagName(), "HEAD");
 | 
						|
 | 
						|
    EXPECT_TRUE(headNode->HasChildren());
 | 
						|
    EXPECT_FALSE(headNode->HasElementAttributes());
 | 
						|
 | 
						|
    CefRefPtr<CefDOMNode> titleNode = headNode->GetFirstChild();
 | 
						|
    EXPECT_TRUE(titleNode.get());
 | 
						|
    EXPECT_TRUE(titleNode->IsElement());
 | 
						|
    EXPECT_FALSE(titleNode->IsText());
 | 
						|
    EXPECT_EQ(titleNode->GetName(), "TITLE");
 | 
						|
    EXPECT_EQ(titleNode->GetElementTagName(), "TITLE");
 | 
						|
    EXPECT_TRUE(titleNode->GetParent()->IsSame(headNode));
 | 
						|
 | 
						|
    EXPECT_FALSE(titleNode->GetNextSibling().get());
 | 
						|
    EXPECT_FALSE(titleNode->GetPreviousSibling().get());
 | 
						|
    EXPECT_TRUE(titleNode->HasChildren());
 | 
						|
    EXPECT_FALSE(titleNode->HasElementAttributes());
 | 
						|
 | 
						|
    CefRefPtr<CefDOMNode> textNode = titleNode->GetFirstChild();
 | 
						|
    EXPECT_TRUE(textNode.get());
 | 
						|
    EXPECT_FALSE(textNode->IsElement());
 | 
						|
    EXPECT_TRUE(textNode->IsText());
 | 
						|
    EXPECT_EQ(textNode->GetValue(), "The Title");
 | 
						|
    EXPECT_TRUE(textNode->GetParent()->IsSame(titleNode));
 | 
						|
 | 
						|
    EXPECT_FALSE(textNode->GetNextSibling().get());
 | 
						|
    EXPECT_FALSE(textNode->GetPreviousSibling().get());
 | 
						|
    EXPECT_FALSE(textNode->HasChildren());
 | 
						|
  }
 | 
						|
 | 
						|
  void TestBodyNodeStructure(CefRefPtr<CefDOMNode> bodyNode,
 | 
						|
                             float devicePixelRatio) {
 | 
						|
    EXPECT_TRUE(bodyNode.get());
 | 
						|
    EXPECT_TRUE(bodyNode->IsElement());
 | 
						|
    EXPECT_FALSE(bodyNode->IsText());
 | 
						|
    EXPECT_EQ(bodyNode->GetName(), "BODY");
 | 
						|
    EXPECT_EQ(bodyNode->GetElementTagName(), "BODY");
 | 
						|
 | 
						|
    EXPECT_TRUE(bodyNode->HasChildren());
 | 
						|
    EXPECT_FALSE(bodyNode->HasElementAttributes());
 | 
						|
 | 
						|
    CefRefPtr<CefDOMNode> h1Node = bodyNode->GetFirstChild();
 | 
						|
    EXPECT_TRUE(h1Node.get());
 | 
						|
    EXPECT_TRUE(h1Node->IsElement());
 | 
						|
    EXPECT_FALSE(h1Node->IsText());
 | 
						|
    EXPECT_EQ(h1Node->GetName(), "H1");
 | 
						|
    EXPECT_EQ(h1Node->GetElementTagName(), "H1");
 | 
						|
 | 
						|
    EXPECT_TRUE(h1Node->GetNextSibling().get());
 | 
						|
    EXPECT_FALSE(h1Node->GetPreviousSibling().get());
 | 
						|
    EXPECT_TRUE(h1Node->HasChildren());
 | 
						|
    EXPECT_FALSE(h1Node->HasElementAttributes());
 | 
						|
 | 
						|
    CefRefPtr<CefDOMNode> textNode = h1Node->GetFirstChild();
 | 
						|
    EXPECT_TRUE(textNode.get());
 | 
						|
    EXPECT_FALSE(textNode->IsElement());
 | 
						|
    EXPECT_TRUE(textNode->IsText());
 | 
						|
    EXPECT_EQ(textNode->GetValue(), "Hello From");
 | 
						|
 | 
						|
    EXPECT_FALSE(textNode->GetPreviousSibling().get());
 | 
						|
    EXPECT_FALSE(textNode->HasChildren());
 | 
						|
 | 
						|
    CefRefPtr<CefDOMNode> brNode = textNode->GetNextSibling();
 | 
						|
    EXPECT_TRUE(brNode.get());
 | 
						|
    EXPECT_TRUE(brNode->IsElement());
 | 
						|
    EXPECT_FALSE(brNode->IsText());
 | 
						|
    EXPECT_EQ(brNode->GetName(), "BR");
 | 
						|
    EXPECT_EQ(brNode->GetElementTagName(), "BR");
 | 
						|
 | 
						|
    EXPECT_FALSE(brNode->HasChildren());
 | 
						|
 | 
						|
    EXPECT_TRUE(brNode->HasElementAttributes());
 | 
						|
    EXPECT_TRUE(brNode->HasElementAttribute("class"));
 | 
						|
    EXPECT_EQ(brNode->GetElementAttribute("class"), "some_class");
 | 
						|
    EXPECT_TRUE(brNode->HasElementAttribute("id"));
 | 
						|
    EXPECT_EQ(brNode->GetElementAttribute("id"), "some_id");
 | 
						|
    EXPECT_FALSE(brNode->HasElementAttribute("no_existing"));
 | 
						|
 | 
						|
    CefDOMNode::AttributeMap map;
 | 
						|
    brNode->GetElementAttributes(map);
 | 
						|
    ASSERT_EQ(map.size(), (size_t)2);
 | 
						|
    EXPECT_EQ(map["class"], "some_class");
 | 
						|
    EXPECT_EQ(map["id"], "some_id");
 | 
						|
 | 
						|
    // Can also retrieve by ID.
 | 
						|
    brNode = bodyNode->GetDocument()->GetElementById("some_id");
 | 
						|
    EXPECT_TRUE(brNode.get());
 | 
						|
    EXPECT_TRUE(brNode->IsElement());
 | 
						|
    EXPECT_FALSE(brNode->IsText());
 | 
						|
    EXPECT_EQ(brNode->GetName(), "BR");
 | 
						|
    EXPECT_EQ(brNode->GetElementTagName(), "BR");
 | 
						|
 | 
						|
    textNode = brNode->GetNextSibling();
 | 
						|
    EXPECT_TRUE(textNode.get());
 | 
						|
    EXPECT_FALSE(textNode->IsElement());
 | 
						|
    EXPECT_TRUE(textNode->IsText());
 | 
						|
    EXPECT_EQ(textNode->GetValue(), "Main Frame");
 | 
						|
 | 
						|
    EXPECT_FALSE(textNode->GetNextSibling().get());
 | 
						|
    EXPECT_FALSE(textNode->HasChildren());
 | 
						|
 | 
						|
    CefRefPtr<CefDOMNode> divNode = h1Node->GetNextSibling();
 | 
						|
    EXPECT_TRUE(divNode.get());
 | 
						|
    EXPECT_TRUE(divNode->IsElement());
 | 
						|
    EXPECT_FALSE(divNode->IsText());
 | 
						|
 | 
						|
    // Returned bounds are in device pixels.
 | 
						|
    CefRect divRect = divNode->GetElementBounds();
 | 
						|
    EXPECT_NEAR(divRect.width, 50.0 * devicePixelRatio, 1);
 | 
						|
    EXPECT_NEAR(divRect.height, 25.0 * devicePixelRatio, 1);
 | 
						|
    EXPECT_NEAR(divRect.x, 150.0 * devicePixelRatio, 1);
 | 
						|
    EXPECT_NEAR(divRect.y, 100.0 * devicePixelRatio, 1);
 | 
						|
 | 
						|
    EXPECT_FALSE(divNode->GetNextSibling().get());
 | 
						|
  }
 | 
						|
 | 
						|
  // Test document structure by iterating through the DOM tree.
 | 
						|
  void TestStructure(CefRefPtr<CefDOMDocument> document) {
 | 
						|
    EXPECT_EQ(document->GetTitle(), "The Title");
 | 
						|
    EXPECT_EQ(document->GetBaseURL(), kTestUrl);
 | 
						|
    EXPECT_EQ(document->GetCompleteURL("foo.html"), "https://tests/foo.html");
 | 
						|
 | 
						|
    // Navigate the complete document structure.
 | 
						|
    CefRefPtr<CefDOMNode> docNode = document->GetDocument();
 | 
						|
    EXPECT_TRUE(docNode.get());
 | 
						|
    EXPECT_FALSE(docNode->IsElement());
 | 
						|
    EXPECT_FALSE(docNode->IsText());
 | 
						|
 | 
						|
    CefRefPtr<CefDOMNode> htmlNode = docNode->GetFirstChild();
 | 
						|
    EXPECT_TRUE(htmlNode.get());
 | 
						|
    EXPECT_TRUE(htmlNode->IsElement());
 | 
						|
    EXPECT_FALSE(htmlNode->IsText());
 | 
						|
    EXPECT_EQ(htmlNode->GetName(), "HTML");
 | 
						|
    EXPECT_EQ(htmlNode->GetElementTagName(), "HTML");
 | 
						|
 | 
						|
    EXPECT_TRUE(htmlNode->HasChildren());
 | 
						|
    EXPECT_FALSE(htmlNode->HasElementAttributes());
 | 
						|
 | 
						|
    CefRefPtr<CefDOMNode> headNode = htmlNode->GetFirstChild();
 | 
						|
    TestHeadNodeStructure(headNode);
 | 
						|
 | 
						|
    const float devicePixelRatio = GetDevicePixelRatio();
 | 
						|
 | 
						|
    CefRefPtr<CefDOMNode> bodyNode = headNode->GetNextSibling();
 | 
						|
    TestBodyNodeStructure(bodyNode, devicePixelRatio);
 | 
						|
 | 
						|
    // Retrieve the head node directly.
 | 
						|
    headNode = document->GetHead();
 | 
						|
    TestHeadNodeStructure(headNode);
 | 
						|
 | 
						|
    // Retrieve the body node directly.
 | 
						|
    bodyNode = document->GetBody();
 | 
						|
    TestBodyNodeStructure(bodyNode, devicePixelRatio);
 | 
						|
  }
 | 
						|
 | 
						|
  // Test document modification by changing the H1 tag.
 | 
						|
  void TestModify(CefRefPtr<CefDOMDocument> document) {
 | 
						|
    CefRefPtr<CefDOMNode> bodyNode = document->GetBody();
 | 
						|
    CefRefPtr<CefDOMNode> h1Node = bodyNode->GetFirstChild();
 | 
						|
 | 
						|
    ASSERT_EQ(h1Node->GetAsMarkup(),
 | 
						|
              "<h1>Hello From<br class=\"some_class\" id=\"some_id\">"
 | 
						|
              "Main Frame</h1>");
 | 
						|
 | 
						|
    CefRefPtr<CefDOMNode> textNode = h1Node->GetFirstChild();
 | 
						|
    ASSERT_EQ(textNode->GetValue(), "Hello From");
 | 
						|
    ASSERT_TRUE(textNode->SetValue("A Different Message From"));
 | 
						|
    ASSERT_EQ(textNode->GetValue(), "A Different Message From");
 | 
						|
 | 
						|
    CefRefPtr<CefDOMNode> brNode = textNode->GetNextSibling();
 | 
						|
    EXPECT_EQ(brNode->GetElementAttribute("class"), "some_class");
 | 
						|
    EXPECT_TRUE(brNode->SetElementAttribute("class", "a_different_class"));
 | 
						|
    EXPECT_EQ(brNode->GetElementAttribute("class"), "a_different_class");
 | 
						|
 | 
						|
    ASSERT_EQ(h1Node->GetAsMarkup(),
 | 
						|
              "<h1>A Different Message From<br class=\"a_different_class\" "
 | 
						|
              "id=\"some_id\">Main Frame</h1>");
 | 
						|
 | 
						|
    ASSERT_FALSE(h1Node->SetValue("Something Different"));
 | 
						|
  }
 | 
						|
 | 
						|
  void Visit(CefRefPtr<CefDOMDocument> document) override {
 | 
						|
    if (test_type_ == DOM_TEST_STRUCTURE) {
 | 
						|
      TestStructure(document);
 | 
						|
    } else if (test_type_ == DOM_TEST_MODIFY) {
 | 
						|
      TestModify(document);
 | 
						|
    }
 | 
						|
 | 
						|
    DestroyTest();
 | 
						|
  }
 | 
						|
 | 
						|
 protected:
 | 
						|
  // Return from the test.
 | 
						|
  void DestroyTest() {
 | 
						|
    // Check if the test has failed.
 | 
						|
    bool result = !TestFailed();
 | 
						|
 | 
						|
    // Return the result to the browser process.
 | 
						|
    CefRefPtr<CefProcessMessage> return_msg =
 | 
						|
        CefProcessMessage::Create(kTestMessage);
 | 
						|
    EXPECT_TRUE(return_msg->GetArgumentList()->SetBool(0, result));
 | 
						|
    browser_->GetMainFrame()->SendProcessMessage(PID_BROWSER, return_msg);
 | 
						|
  }
 | 
						|
 | 
						|
  // Used to convert between device pixels and CSS pixels.
 | 
						|
  float GetDevicePixelRatio() {
 | 
						|
    auto context = browser_->GetMainFrame()->GetV8Context();
 | 
						|
    EXPECT_TRUE(context);
 | 
						|
 | 
						|
    CefRefPtr<CefV8Value> retval;
 | 
						|
    CefRefPtr<CefV8Exception> exception;
 | 
						|
    EXPECT_TRUE(context->Eval("window.devicePixelRatio", CefString(), 0, retval,
 | 
						|
                              exception));
 | 
						|
    if (exception) {
 | 
						|
      ADD_FAILURE() << exception->GetMessage().c_str();
 | 
						|
      return 1.0;
 | 
						|
    }
 | 
						|
 | 
						|
    if (retval->IsValid() && retval->IsDouble()) {
 | 
						|
      return static_cast<float>(retval->GetDoubleValue());
 | 
						|
    }
 | 
						|
 | 
						|
    ADD_FAILURE() << "Failed to retrieve devicePixelRatio";
 | 
						|
    return 1.0;
 | 
						|
  }
 | 
						|
 | 
						|
  CefRefPtr<CefBrowser> browser_;
 | 
						|
  DOMTestType test_type_;
 | 
						|
 | 
						|
  IMPLEMENT_REFCOUNTING(TestDOMVisitor);
 | 
						|
};
 | 
						|
 | 
						|
// Used in the render process.
 | 
						|
class DOMRendererTest : public ClientAppRenderer::Delegate {
 | 
						|
 public:
 | 
						|
  DOMRendererTest() = default;
 | 
						|
 | 
						|
  bool OnProcessMessageReceived(CefRefPtr<ClientAppRenderer> app,
 | 
						|
                                CefRefPtr<CefBrowser> browser,
 | 
						|
                                CefRefPtr<CefFrame> frame,
 | 
						|
                                CefProcessId source_process,
 | 
						|
                                CefRefPtr<CefProcessMessage> message) override {
 | 
						|
    if (message->GetName() == kTestMessage) {
 | 
						|
      EXPECT_EQ(message->GetArgumentList()->GetSize(), (size_t)1);
 | 
						|
      int test_type = message->GetArgumentList()->GetInt(0);
 | 
						|
 | 
						|
      browser->GetMainFrame()->VisitDOM(
 | 
						|
          new TestDOMVisitor(browser, static_cast<DOMTestType>(test_type)));
 | 
						|
      return true;
 | 
						|
    }
 | 
						|
    return false;
 | 
						|
  }
 | 
						|
 | 
						|
 private:
 | 
						|
  IMPLEMENT_REFCOUNTING(DOMRendererTest);
 | 
						|
};
 | 
						|
 | 
						|
// Used in the browser process.
 | 
						|
class TestDOMHandler : public TestHandler {
 | 
						|
 public:
 | 
						|
  explicit TestDOMHandler(DOMTestType test) : test_type_(test) {}
 | 
						|
 | 
						|
  void RunTest() override {
 | 
						|
    // Specified values are in CSS pixels.
 | 
						|
    std::stringstream mainHtml;
 | 
						|
    mainHtml << "<html>"
 | 
						|
                "<head><title>The Title</title></head>"
 | 
						|
                "<body>"
 | 
						|
                "<h1>Hello From<br class=\"some_class\"/ id=\"some_id\"/>"
 | 
						|
                "Main Frame</h1>"
 | 
						|
                "<div id=\"sized_element\" style=\"width: 50px; height: 25px; "
 | 
						|
                "position: fixed; top: 100px; left: 150px;\"/>"
 | 
						|
                "</body>"
 | 
						|
                "</html>";
 | 
						|
 | 
						|
    AddResource(kTestUrl, mainHtml.str(), "text/html");
 | 
						|
    CreateBrowser(kTestUrl);
 | 
						|
 | 
						|
    // Time out the test after a reasonable period of time.
 | 
						|
    SetTestTimeout();
 | 
						|
  }
 | 
						|
 | 
						|
  void OnLoadEnd(CefRefPtr<CefBrowser> browser,
 | 
						|
                 CefRefPtr<CefFrame> frame,
 | 
						|
                 int httpStatusCode) override {
 | 
						|
    if (frame->IsMain()) {
 | 
						|
      // Start the test in the render process.
 | 
						|
      CefRefPtr<CefProcessMessage> message(
 | 
						|
          CefProcessMessage::Create(kTestMessage));
 | 
						|
      message->GetArgumentList()->SetInt(0, test_type_);
 | 
						|
      frame->SendProcessMessage(PID_RENDERER, message);
 | 
						|
    }
 | 
						|
  }
 | 
						|
 | 
						|
  bool OnProcessMessageReceived(CefRefPtr<CefBrowser> browser,
 | 
						|
                                CefRefPtr<CefFrame> frame,
 | 
						|
                                CefProcessId source_process,
 | 
						|
                                CefRefPtr<CefProcessMessage> message) override {
 | 
						|
    EXPECT_STREQ(message->GetName().ToString().c_str(), kTestMessage);
 | 
						|
 | 
						|
    got_message_.yes();
 | 
						|
 | 
						|
    if (message->GetArgumentList()->GetBool(0)) {
 | 
						|
      got_success_.yes();
 | 
						|
    }
 | 
						|
 | 
						|
    // Test is complete.
 | 
						|
    DestroyTest();
 | 
						|
 | 
						|
    return true;
 | 
						|
  }
 | 
						|
 | 
						|
  DOMTestType test_type_;
 | 
						|
  TrackCallback got_message_;
 | 
						|
  TrackCallback got_success_;
 | 
						|
 | 
						|
  IMPLEMENT_REFCOUNTING(TestDOMHandler);
 | 
						|
};
 | 
						|
 | 
						|
}  // namespace
 | 
						|
 | 
						|
// Test DOM structure reading.
 | 
						|
TEST(DOMTest, Read) {
 | 
						|
  CefRefPtr<TestDOMHandler> handler = new TestDOMHandler(DOM_TEST_STRUCTURE);
 | 
						|
  handler->ExecuteTest();
 | 
						|
 | 
						|
  EXPECT_TRUE(handler->got_message_);
 | 
						|
  EXPECT_TRUE(handler->got_success_);
 | 
						|
 | 
						|
  ReleaseAndWaitForDestructor(handler);
 | 
						|
}
 | 
						|
 | 
						|
// Test DOM modifications.
 | 
						|
TEST(DOMTest, Modify) {
 | 
						|
  CefRefPtr<TestDOMHandler> handler = new TestDOMHandler(DOM_TEST_MODIFY);
 | 
						|
  handler->ExecuteTest();
 | 
						|
 | 
						|
  EXPECT_TRUE(handler->got_message_);
 | 
						|
  EXPECT_TRUE(handler->got_success_);
 | 
						|
 | 
						|
  ReleaseAndWaitForDestructor(handler);
 | 
						|
}
 | 
						|
 | 
						|
// Entry point for creating DOM renderer test objects.
 | 
						|
// Called from client_app_delegates.cc.
 | 
						|
void CreateDOMRendererTests(ClientAppRenderer::DelegateSet& delegates) {
 | 
						|
  delegates.insert(new DOMRendererTest);
 | 
						|
}
 |