1 files changed, 94 insertions, 0 deletions
diff --git a/chromium/chrome/renderer/chrome_render_frame_observer_browsertest.cc b/chromium/chrome/renderer/chrome_render_frame_observer_browsertest.cc
new file mode 100644
index 00000000000..be42e42f39e
--- /dev/null
+++ b/chromium/chrome/renderer/chrome_render_frame_observer_browsertest.cc
@@ -0,0 +1,94 @@
+// Copyright 2016 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 "chrome/renderer/chrome_render_frame_observer.h"
+
+#include <tuple>
+
+#include "base/bind.h"
+#include "base/run_loop.h"
+#include "base/test/metrics/histogram_tester.h"
+#include "chrome/test/base/chrome_render_view_test.h"
+#include "components/translate/content/common/translate.mojom.h"
+#include "components/translate/content/renderer/translate_helper.h"
+#include "components/translate/core/common/translate_constants.h"
+#include "content/public/renderer/render_frame.h"
+#include "content/public/renderer/render_view.h"
+#include "mojo/public/cpp/bindings/pending_receiver.h"
+#include "mojo/public/cpp/bindings/pending_remote.h"
+#include "mojo/public/cpp/bindings/receiver_set.h"
+#include "services/service_manager/public/cpp/interface_provider.h"
+#include "third_party/blink/public/web/web_view.h"
+#include "third_party/blink/public/web/web_widget.h"
+
+namespace {
+
+class FakeContentTranslateDriver
+    : public translate::mojom::ContentTranslateDriver {
+ public:
+  FakeContentTranslateDriver()
+      : called_new_page_(false), page_needs_translation_(false) {}
+  ~FakeContentTranslateDriver() override {}
+
+  void BindHandle(mojo::ScopedMessagePipeHandle handle) {
+    receivers_.Add(
+        this, mojo::PendingReceiver<translate::mojom::ContentTranslateDriver>(
+                  std::move(handle)));
+  }
+
+  // translate::mojom::ContentTranslateDriver implementation.
+  void RegisterPage(mojo::PendingRemote<translate::mojom::Page> page,
+                    const translate::LanguageDetectionDetails& details,
+                    bool page_needs_translation) override {
+    called_new_page_ = true;
+    page_needs_translation_ = page_needs_translation;
+  }
+
+  bool called_new_page_;
+  bool page_needs_translation_;
+
+ private:
+  mojo::ReceiverSet<translate::mojom::ContentTranslateDriver> receivers_;
+};
+
+}  // namespace
+
+// Constants for UMA statistic collection.
+static const char kTranslateCaptureText[] = "Translate.CaptureText";
+
+class ChromeRenderFrameObserverTest : public ChromeRenderViewTest {
+ protected:
+  void SetUp() override {
+    ChromeRenderViewTest::SetUp();
+
+    service_manager::InterfaceProvider* remote_interfaces =
+        view_->GetMainRenderFrame()->GetRemoteInterfaces();
+    service_manager::InterfaceProvider::TestApi test_api(remote_interfaces);
+    test_api.SetBinderForName(
+        translate::mojom::ContentTranslateDriver::Name_,
+        base::Bind(&FakeContentTranslateDriver::BindHandle,
+                   base::Unretained(&fake_translate_driver_)));
+  }
+
+  FakeContentTranslateDriver fake_translate_driver_;
+};
+
+TEST_F(ChromeRenderFrameObserverTest, SkipCapturingSubFrames) {
+  base::HistogramTester histogram_tester;
+  LoadHTML(
+      "<!DOCTYPE html><body>"
+      "This is a main document"
+      "<iframe srcdoc=\"This a document in an iframe.\">"
+      "</body>");
+  view_->GetWebView()->MainFrameWidget()->UpdateAllLifecyclePhases(
+      blink::WebWidget::LifecycleUpdateReason::kTest);
+
+  base::RunLoop().RunUntilIdle();
+  ASSERT_TRUE(fake_translate_driver_.called_new_page_);
+  EXPECT_TRUE(fake_translate_driver_.page_needs_translation_)
+      << "Page should be translatable.";
+  // Should have 2 samples: one for preliminary capture, one for final capture.
+  // If there are more, then subframes are being captured more than once.
+  histogram_tester.ExpectTotalCount(kTranslateCaptureText, 2);
+}