Add integration of the new heuristics

components/dom_distiller/content/distillable_page_utils_browsertest.cc

+// Copyright 2015 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/bind.h"
+#include "base/path_service.h"
+#include "base/run_loop.h"
+#include "components/dom_distiller/content/distillable_page_utils.h"
+#include "components/dom_distiller/core/distillable_page_detector.h"
+#include "components/dom_distiller/core/page_features.h"
+#include "content/public/browser/browser_context.h"
+#include "content/public/browser/render_frame_host.h"
+#include "content/public/browser/web_contents_observer.h"
+#include "content/public/test/content_browser_test.h"
+#include "content/shell/browser/shell.h"
+#include "net/test/embedded_test_server/embedded_test_server.h"
+#include "ui/base/resource/resource_bundle.h"
+
+namespace dom_distiller {
+namespace {
+
+const char* kArticlePath = "/og_article.html";
+const char* kNonArticlePath = "/non_og_article.html";
+
+class DomDistillerDistillablePageUtilsTest : public content::ContentBrowserTest,
+                                             content::WebContentsObserver {
+ public:
+  void SetUpOnMainThread() override {
+    AddComponentsResources();
+    SetUpTestServer();
+    ContentBrowserTest::SetUpOnMainThread();
+  }
+
+  void LoadURL(const std::string& url) {
+    content::WebContents* current_web_contents = shell()->web_contents();
+    Observe(current_web_contents);
+    base::RunLoop url_loaded_runner;
+    main_frame_loaded_callback_ = url_loaded_runner.QuitClosure();
+    current_web_contents->GetController().LoadURL(
+        embedded_test_server()->GetURL(url),
+        content::Referrer(),
+        ui::PAGE_TRANSITION_TYPED,
+        std::string());
+    url_loaded_runner.Run();
+    main_frame_loaded_callback_ = base::Closure();
+    Observe(nullptr);
+  }
+
+ private:
+  void AddComponentsResources() {
+    base::FilePath pak_file;
+    base::FilePath pak_dir;
+    PathService::Get(base::DIR_MODULE, &pak_dir);
+    pak_file =
+        pak_dir.Append(FILE_PATH_LITERAL("components_tests_resources.pak"));
+    ui::ResourceBundle::GetSharedInstance().AddDataPackFromPath(
+        pak_file, ui::SCALE_FACTOR_NONE);
+  }
+
+  void SetUpTestServer() {
+    base::FilePath path;
+    PathService::Get(base::DIR_SOURCE_ROOT, &path);
+    path = path.AppendASCII("components/test/data/dom_distiller");
+    embedded_test_server()->ServeFilesFromDirectory(path);
+    ASSERT_TRUE(embedded_test_server()->InitializeAndWaitUntilReady());
+  }
+
+  void DocumentLoadedInFrame(
+      content::RenderFrameHost* render_frame_host) override {
+    if (!render_frame_host->GetParent())
+      main_frame_loaded_callback_.Run();
+  }
+
+  base::Closure main_frame_loaded_callback_;
+};
+
+class ResultHolder {
+ public:
+  ResultHolder(base::Closure callback) : callback_(callback) {}
+
+  void OnResult(bool result) {
+    result_ = result;
+    callback_.Run();
+  }
+
+  bool GetResult() {
+    return result_;
+  }
+
+  base::Callback<void(bool)> GetCallback() {
+    return base::Bind(&ResultHolder::OnResult, base::Unretained(this));
+  }
+
+ private:
+  base::Closure callback_;
+  bool result_;
+};
+
+}  // namespace
+
+IN_PROC_BROWSER_TEST_F(DomDistillerDistillablePageUtilsTest, TestIsOGArticle) {
+  LoadURL(kArticlePath);
+  base::RunLoop run_loop_;
+  ResultHolder holder(run_loop_.QuitClosure());
+  IsOpenGraphArticle(shell()->web_contents(), holder.GetCallback());
+  run_loop_.Run();
+  ASSERT_TRUE(holder.GetResult());
+}
+
+IN_PROC_BROWSER_TEST_F(DomDistillerDistillablePageUtilsTest,
+                       TestIsNotOGArticle) {
+  LoadURL(kNonArticlePath);
+  base::RunLoop run_loop_;
+  ResultHolder holder(run_loop_.QuitClosure());
+  IsOpenGraphArticle(shell()->web_contents(), holder.GetCallback());
+  run_loop_.Run();
+  ASSERT_FALSE(holder.GetResult());
+}
+
+IN_PROC_BROWSER_TEST_F(DomDistillerDistillablePageUtilsTest,
+                       TestIsDistillablePage) {
+  scoped_ptr<AdaBoostProto> proto(new AdaBoostProto);
+  proto->set_num_features(kDerivedFeaturesCount);
+  proto->set_num_stumps(1);
+
+  StumpProto* stump = proto->add_stump();
+  stump->set_feature_number(0);
+  stump->set_weight(1);
+  stump->set_split(-1);
+  scoped_ptr<DistillablePageDetector> detector(
+      new DistillablePageDetector(proto.Pass()));
+  EXPECT_DOUBLE_EQ(0.5, detector->GetThreshold());
+  // The first value of the first feature is either 0 or 1. Since the stump's
+  // split is -1, the stump weight will be applied to any set of derived
+  // features.
+  LoadURL(kArticlePath);
+  base::RunLoop run_loop_;
+  ResultHolder holder(run_loop_.QuitClosure());
+  IsDistillablePageForDetector(shell()->web_contents(), detector.get(),
+                               holder.GetCallback());
+  run_loop_.Run();
+  ASSERT_TRUE(holder.GetResult());
+}
+
+IN_PROC_BROWSER_TEST_F(DomDistillerDistillablePageUtilsTest,
+                       TestIsNotDistillablePage) {
+  scoped_ptr<AdaBoostProto> proto(new AdaBoostProto);
+  proto->set_num_features(kDerivedFeaturesCount);
+  proto->set_num_stumps(1);
+  StumpProto* stump = proto->add_stump();
+  stump->set_feature_number(0);
+  stump->set_weight(-1);
+  stump->set_split(-1);
+  scoped_ptr<DistillablePageDetector> detector(
+      new DistillablePageDetector(proto.Pass()));
+  EXPECT_DOUBLE_EQ(-0.5, detector->GetThreshold());
+  // The first value of the first feature is either 0 or 1. Since the stump's
+  // split is -1, the stump weight will be applied to any set of derived
+  // features.
+  LoadURL(kArticlePath);
+  base::RunLoop run_loop_;
+  ResultHolder holder(run_loop_.QuitClosure());
+  IsDistillablePageForDetector(shell()->web_contents(), detector.get(),
+                               holder.GetCallback());
+  run_loop_.Run();
+  ASSERT_FALSE(holder.GetResult());
+}
+
+}  // namespace dom_distiller