src/tests/ceftests/dom_unittest.cc - cef - Git at Google

 // 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 = "http://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) {
     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());
     CefRect divRect = divNode->GetElementBounds();
     EXPECT_EQ(divRect.width, 50);
     EXPECT_EQ(divRect.height, 25);
     EXPECT_EQ(divRect.x, 150);
     EXPECT_EQ(divRect.y, 100);
     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"), "http://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);

     CefRefPtr<CefDOMNode> bodyNode = headNode->GetNextSibling();
     TestBodyNodeStructure(bodyNode);

     // Retrieve the head node directly.
     headNode = document->GetHead();
     TestHeadNodeStructure(headNode);

     // Retrieve the body node directly.
     bodyNode = document->GetBody();
     TestBodyNodeStructure(bodyNode);
   }

   // 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);
   }

   CefRefPtr<CefBrowser> browser_;
   DOMTestType test_type_;

   IMPLEMENT_REFCOUNTING(TestDOMVisitor);
 };

 // Used in the render process.
 class DOMRendererTest : public ClientAppRenderer::Delegate {
  public:
   DOMRendererTest() {}

   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 {
     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);
 }
	// 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 = "http://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) {
	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());
	CefRect divRect = divNode->GetElementBounds();
	EXPECT_EQ(divRect.width, 50);
	EXPECT_EQ(divRect.height, 25);
	EXPECT_EQ(divRect.x, 150);
	EXPECT_EQ(divRect.y, 100);
	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"), "http://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);

	CefRefPtr<CefDOMNode> bodyNode = headNode->GetNextSibling();
	TestBodyNodeStructure(bodyNode);

	// Retrieve the head node directly.
	headNode = document->GetHead();
	TestHeadNodeStructure(headNode);

	// Retrieve the body node directly.
	bodyNode = document->GetBody();
	TestBodyNodeStructure(bodyNode);
	}

	// 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);
	}

	CefRefPtr<CefBrowser> browser_;
	DOMTestType test_type_;

	IMPLEMENT_REFCOUNTING(TestDOMVisitor);
	};

	// Used in the render process.
	class DOMRendererTest : public ClientAppRenderer::Delegate {
	public:
	DOMRendererTest() {}

	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 {
	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);
	}