Add functions to glue to get the full html of the page and also to check if t...

webkit/glue/webframe_unittest.cc

 // Copyright (c) 2006-2008 The Chromium 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 "base/string_util.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "webkit/glue/webframe.h"
 #include "webkit/glue/webview.h"
 #include "webkit/tools/test_shell/test_shell_test.h"
 
@@ skipping 34 matching lines @@
   test_shell_->WaitTestFinished();
 
   frame->GetContentAsPlainText(std::numeric_limits<int>::max(), &text);
   EXPECT_EQ("Hello world\n\nsub\ntext", WideToUTF8(text));
 
   // Get the frame text where the subframe separator falls on the boundary of
   // what we'll take. There used to be a crash in this case.
   frame->GetContentAsPlainText(12, &text);
   EXPECT_EQ("Hello world", WideToUTF8(text));
 }
+
+TEST_F(WebFrameTest, GetFullHtmlOfPage) {
+  WebView* view = test_shell_->webView();
+  WebFrame* frame = view->GetMainFrame();
+
+  // Generate a simple test case.
+  const char simple_source[] = "<p>Hello</p><p>World</p>";
+  GURL test_url("http://hello/");
+  frame->LoadHTMLString(simple_source, test_url);
+  test_shell_->WaitTestFinished();
+
+  std::wstring text;
+  frame->GetContentAsPlainText(std::numeric_limits<int>::max(), &text);
+  EXPECT_EQ("Hello\n\nWorld", WideToUTF8(text));
+
+  const std::string html = frame->GetFullPageHtml();
+
+  // Load again with the output html.
+  frame->LoadHTMLString(html, test_url);
+  test_shell_->WaitTestFinished();
+
+  EXPECT_EQ(html, frame->GetFullPageHtml());
+
+  text = L"";
+  frame->GetContentAsPlainText(std::numeric_limits<int>::max(), &text);
+  EXPECT_EQ("Hello\n\nWorld", WideToUTF8(text));
+
+  // Test selection check
+  EXPECT_FALSE(frame->HasSelection());
+  frame->SelectAll();
+  EXPECT_TRUE(frame->HasSelection());
+  frame->ClearSelection();
+  EXPECT_FALSE(frame->HasSelection());
+}