Adding previews information to PLM UKM

chrome/browser/page_load_metrics/observers/previews_ukm_observer_unittest.cc

Index: chrome/browser/page_load_metrics/observers/previews_ukm_observer_unittest.cc
diff --git a/chrome/browser/page_load_metrics/observers/previews_ukm_observer_unittest.cc b/chrome/browser/page_load_metrics/observers/previews_ukm_observer_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..79fc3038f910be3805f86814a698d7e0ee2b617a
--- /dev/null
+++ b/chrome/browser/page_load_metrics/observers/previews_ukm_observer_unittest.cc
@@ -0,0 +1,368 @@
+// Copyright 2017 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/browser/page_load_metrics/observers/previews_ukm_observer.h"
+
+#include "base/macros.h"
+#include "base/metrics/metrics_hashes.h"
+#include "base/optional.h"
+#include "chrome/browser/loader/chrome_navigation_data.h"
+#include "chrome/browser/page_load_metrics/observers/page_load_metrics_observer_test_harness.h"
+#include "chrome/browser/page_load_metrics/page_load_metrics_observer.h"
+#include "chrome/browser/previews/previews_infobar_delegate.h"
+#include "chrome/test/base/testing_browser_process.h"
+#include "components/data_reduction_proxy/core/browser/data_reduction_proxy_data.h"
+#include "components/ukm/test_ukm_recorder.h"
+#include "components/ukm/ukm_source.h"
+
+namespace previews {
+
+namespace {
+
+const char kDefaultTestUrl[] = "https://www.google.com/";
+
+data_reduction_proxy::DataReductionProxyData* DataForNavigationHandle(
+    content::WebContents* web_contents,
+    content::NavigationHandle* navigation_handle) {
+  ChromeNavigationData* chrome_navigation_data = new ChromeNavigationData();
+  content::WebContentsTester::For(web_contents)
+      ->SetNavigationData(navigation_handle,
+                          base::WrapUnique(chrome_navigation_data));
+  data_reduction_proxy::DataReductionProxyData* data =
+      new data_reduction_proxy::DataReductionProxyData();
+  chrome_navigation_data->SetDataReductionProxyData(base::WrapUnique(data));
+
+  return data;
+}
+
+class TestPreviewsUKMObserver : public PreviewsUKMObserver {
+ public:
+  TestPreviewsUKMObserver(content::WebContents* web_contents,
+                          bool data_reduction_proxy_used,
+                          bool lite_page_received)
+      : web_contents_(web_contents),
+        data_reduction_proxy_used_(data_reduction_proxy_used),
+        lite_page_received_(lite_page_received) {}
+
+  ~TestPreviewsUKMObserver() override {}
+
+  // page_load_metrics::PageLoadMetricsObserver implementation:
+  ObservePolicy OnCommit(content::NavigationHandle* navigation_handle,
+                         ukm::SourceId source_id) override {
+    data_reduction_proxy::DataReductionProxyData* data =
+        DataForNavigationHandle(web_contents_, navigation_handle);
+    data->set_used_data_reduction_proxy(data_reduction_proxy_used_);
+    data->set_request_url(GURL(kDefaultTestUrl));
+    data->set_lite_page_received(lite_page_received_);
+    return PreviewsUKMObserver::OnCommit(navigation_handle, source_id);
+  }
+
+ private:
+  content::WebContents* web_contents_;
+  bool data_reduction_proxy_used_;
+  bool lite_page_received_;
+
+  DISALLOW_COPY_AND_ASSIGN(TestPreviewsUKMObserver);
+};
+
+class PreviewsUKMObserverTest
+    : public page_load_metrics::PageLoadMetricsObserverTestHarness {
+ public:
+  PreviewsUKMObserverTest() {}
+  ~PreviewsUKMObserverTest() override {}
+
+  void RunTest(bool data_reduction_proxy_used, bool lite_page_received) {
+    data_reduction_proxy_used_ = data_reduction_proxy_used;
+    lite_page_received_ = lite_page_received;
+    NavigateAndCommit(GURL(kDefaultTestUrl));
+  }
+
+  void ValidateUKM(bool server_lofi_expected,
+                   bool client_lofi_expected,
+                   bool lite_page_expected,
+                   bool opt_out_expected) {
+    const ukm::UkmSource* source =
+        test_ukm_recorder().GetSourceForUrl(kDefaultTestUrl);
+    if (!server_lofi_expected && !client_lofi_expected && !lite_page_expected &&
+        !opt_out_expected) {
+      EXPECT_EQ(0u, test_ukm_recorder().entries_count());
+      return;
+    }
+
+    EXPECT_EQ(server_lofi_expected, test_ukm_recorder().HasMetric(
+                                        *source, "Previews", "server_lofi"));
+    EXPECT_EQ(client_lofi_expected, test_ukm_recorder().HasMetric(
+                                        *source, "Previews", "client_lofi"));
+    EXPECT_EQ(lite_page_expected,
+              test_ukm_recorder().HasMetric(*source, "Previews", "lite_page"));
+    EXPECT_EQ(opt_out_expected,
+              test_ukm_recorder().HasMetric(*source, "Previews", "opt_out"));
+  }
+
+ protected:
+  void RegisterObservers(page_load_metrics::PageLoadTracker* tracker) override {
+    tracker->AddObserver(base::MakeUnique<TestPreviewsUKMObserver>(
+        web_contents(), data_reduction_proxy_used_, lite_page_received_));
+    // Data is only added to the first navigation after RunTest().
+    data_reduction_proxy_used_ = false;
+    lite_page_received_ = false;
+  }
+
+ private:
+  bool data_reduction_proxy_used_ = false;
+  bool lite_page_received_ = false;
+
+  DISALLOW_COPY_AND_ASSIGN(PreviewsUKMObserverTest);
+};
+
+TEST_F(PreviewsUKMObserverTest, NoPreviewSeen) {
+  RunTest(false /* data_reduction_proxy_used */,
+          false /* lite_page_received */);
+  NavigateToUntrackedUrl();
+
+  ValidateUKM(false /* server_lofi_expected */,
+              false /* client_lofi_expected */, false /* lite_page_expected */,
+              false /* opt_out_expected */);
+}
+
+TEST_F(PreviewsUKMObserverTest, UntrackedPreviewTypeOptOut) {
+  RunTest(false /* data_reduction_proxy_used */,
+          false /* lite_page_received */);
+  observer()->BroadcastEventToObservers(
+      PreviewsInfoBarDelegate::OptOutEventKey());
+  NavigateToUntrackedUrl();
+
+  // Opt out should not be added sicne we don't track this type.
+  ValidateUKM(false /* server_lofi_expected */,
+              false /* client_lofi_expected */, false /* lite_page_expected */,
+              false /* opt_out_expected */);
+}
+
+TEST_F(PreviewsUKMObserverTest, LitePageSeen) {
+  RunTest(true /* data_reduction_proxy_used */, true /* lite_page_received */);
+
+  NavigateToUntrackedUrl();
+
+  ValidateUKM(false /* server_lofi_expected */,
+              false /* client_lofi_expected */, true /* lite_page_expected */,
+              false /* opt_out_expected */);
+}
+
+TEST_F(PreviewsUKMObserverTest, LitePageOptOut) {
+  RunTest(true /* data_reduction_proxy_used */, true /* lite_page_received */);
+
+  observer()->BroadcastEventToObservers(
+      PreviewsInfoBarDelegate::OptOutEventKey());
+  NavigateToUntrackedUrl();
+
+  ValidateUKM(false /* server_lofi_expected */,
+              false /* client_lofi_expected */, true /* lite_page_expected */,
+              true /* opt_out_expected */);
+}
+
+TEST_F(PreviewsUKMObserverTest, ClientLoFiSeen) {
+  RunTest(false /* data_reduction_proxy_used */,
+          false /* lite_page_received */);
+
+  std::unique_ptr<data_reduction_proxy::DataReductionProxyData> data =
+      base::MakeUnique<data_reduction_proxy::DataReductionProxyData>();
+  data->set_client_lofi_requested(true);
+
+  // Prepare 3 resources of varying size and configurations, 2 of which have
+  // client LoFi set.
+  page_load_metrics::ExtraRequestCompleteInfo resources[] = {
+      {GURL(kDefaultTestUrl), net::HostPortPair(), -1, false /*was_cached*/,
+       1024 * 40 /* raw_body_bytes */,
+       1024 * 40 * 5 /* original_network_content_length */, std::move(data),
+       content::ResourceType::RESOURCE_TYPE_IMAGE, 0},
+      // Uncached non-proxied request.
+      {GURL(kDefaultTestUrl), net::HostPortPair(), -1, false /*was_cached*/,
+       1024 * 40 /* raw_body_bytes */,
+       1024 * 40 /* original_network_content_length */,
+       nullptr /* data_reduction_proxy_data */,
+       content::ResourceType::RESOURCE_TYPE_IMAGE, 0},
+  };
+
+  for (const auto& request : resources)
+    SimulateLoadedResource(request);
+
+  NavigateToUntrackedUrl();
+
+  ValidateUKM(false /* server_lofi_expected */, true /* client_lofi_expected */,
+              false /* lite_page_expected */, false /* opt_out_expected */);
+}
+
+TEST_F(PreviewsUKMObserverTest, ClientLoFiOptOut) {
+  RunTest(false /* data_reduction_proxy_used */,
+          false /* lite_page_received */);
+
+  std::unique_ptr<data_reduction_proxy::DataReductionProxyData> data =
+      base::MakeUnique<data_reduction_proxy::DataReductionProxyData>();
+  data->set_client_lofi_requested(true);
+
+  // Prepare 3 resources of varying size and configurations, 2 of which have
+  // client LoFi set.
+  page_load_metrics::ExtraRequestCompleteInfo resources[] = {
+      {GURL(kDefaultTestUrl), net::HostPortPair(), -1, false /*was_cached*/,
+       1024 * 40 /* raw_body_bytes */,
+       1024 * 40 * 5 /* original_network_content_length */, std::move(data),
+       content::ResourceType::RESOURCE_TYPE_IMAGE, 0},
+      // Uncached non-proxied request.
+      {GURL(kDefaultTestUrl), net::HostPortPair(), -1, false /*was_cached*/,
+       1024 * 40 /* raw_body_bytes */,
+       1024 * 40 /* original_network_content_length */,
+       nullptr /* data_reduction_proxy_data */,
+       content::ResourceType::RESOURCE_TYPE_IMAGE, 0},
+  };
+
+  for (const auto& request : resources)
+    SimulateLoadedResource(request);
+  observer()->BroadcastEventToObservers(
+      PreviewsInfoBarDelegate::OptOutEventKey());
+  NavigateToUntrackedUrl();
+
+  ValidateUKM(false /* server_lofi_expected */, true /* client_lofi_expected */,
+              false /* lite_page_expected */, true /* opt_out_expected */);
+}
+
+TEST_F(PreviewsUKMObserverTest, ServerLoFiSeen) {
+  RunTest(true /* data_reduction_proxy_used */, false /* lite_page_received */);
+
+  std::unique_ptr<data_reduction_proxy::DataReductionProxyData> data =
+      base::MakeUnique<data_reduction_proxy::DataReductionProxyData>();
+  data->set_used_data_reduction_proxy(true);
+  data->set_lofi_received(true);
+
+  // Prepare 3 resources of varying size and configurations, 2 of which have
+  // client LoFi set.
+  page_load_metrics::ExtraRequestCompleteInfo resources[] = {
+      {GURL(kDefaultTestUrl), net::HostPortPair(), -1, false /*was_cached*/,
+       1024 * 40 /* raw_body_bytes */,
+       1024 * 40 * 5 /* original_network_content_length */, std::move(data),
+       content::ResourceType::RESOURCE_TYPE_IMAGE, 0},
+      {GURL(kDefaultTestUrl), net::HostPortPair(), -1, false /*was_cached*/,
+       1024 * 40 /* raw_body_bytes */,
+       1024 * 40 /* original_network_content_length */,
+       nullptr /* data_reduction_proxy_data */,
+       content::ResourceType::RESOURCE_TYPE_IMAGE, 0},
+  };
+
+  for (const auto& request : resources)
+    SimulateLoadedResource(request);
+
+  NavigateToUntrackedUrl();
+
+  ValidateUKM(true /* server_lofi_expected */, false /* client_lofi_expected */,
+              false /* lite_page_expected */, false /* opt_out_expected */);
+}
+
+TEST_F(PreviewsUKMObserverTest, ServerLoFiOptOut) {
+  RunTest(true /* data_reduction_proxy_used */, false /* lite_page_received */);
+
+  std::unique_ptr<data_reduction_proxy::DataReductionProxyData> data =
+      base::MakeUnique<data_reduction_proxy::DataReductionProxyData>();
+  data->set_used_data_reduction_proxy(true);
+  data->set_lofi_received(true);
+
+  // Prepare 3 resources of varying size and configurations, 2 of which have
+  // client LoFi set.
+  page_load_metrics::ExtraRequestCompleteInfo resources[] = {
+      {GURL(kDefaultTestUrl), net::HostPortPair(), -1, false /*was_cached*/,
+       1024 * 40 /* raw_body_bytes */,
+       1024 * 40 * 5 /* original_network_content_length */, std::move(data),
+       content::ResourceType::RESOURCE_TYPE_IMAGE, 0},
+      {GURL(kDefaultTestUrl), net::HostPortPair(), -1, false /*was_cached*/,
+       1024 * 40 /* raw_body_bytes */,
+       1024 * 40 /* original_network_content_length */,
+       nullptr /* data_reduction_proxy_data */,
+       content::ResourceType::RESOURCE_TYPE_IMAGE, 0},
+  };
+
+  for (const auto& request : resources)
+    SimulateLoadedResource(request);
+
+  observer()->BroadcastEventToObservers(
+      PreviewsInfoBarDelegate::OptOutEventKey());
+  NavigateToUntrackedUrl();
+
+  ValidateUKM(true /* server_lofi_expected */, false /* client_lofi_expected */,
+              false /* lite_page_expected */, true /* opt_out_expected */);
+}
+
+TEST_F(PreviewsUKMObserverTest, BothLoFiSeen) {
+  RunTest(true /* data_reduction_proxy_used */, false /* lite_page_received */);
+
+  std::unique_ptr<data_reduction_proxy::DataReductionProxyData> data1 =
+      base::MakeUnique<data_reduction_proxy::DataReductionProxyData>();
+  data1->set_used_data_reduction_proxy(true);
+  data1->set_lofi_received(true);
+
+  std::unique_ptr<data_reduction_proxy::DataReductionProxyData> data2 =
+      base::MakeUnique<data_reduction_proxy::DataReductionProxyData>();
+  data2->set_used_data_reduction_proxy(true);
+  data2->set_client_lofi_requested(true);
+
+  // Prepare 4 resources of varying size and configurations, 1 has Client LoFi,
+  // 1 has Server LoFi.
+  page_load_metrics::ExtraRequestCompleteInfo resources[] = {
+      // Uncached proxied request with .1 compression ratio.
+      {GURL(kDefaultTestUrl), net::HostPortPair(), -1, false /*was_cached*/,
+       1024 * 40 /* raw_body_bytes */,
+       1024 * 40 * 10 /* original_network_content_length */, std::move(data1),
+       content::ResourceType::RESOURCE_TYPE_IMAGE, 0},
+      // Uncached proxied request with .5 compression ratio.
+      {GURL(kDefaultTestUrl), net::HostPortPair(), -1, false /*was_cached*/,
+       1024 * 40 /* raw_body_bytes */,
+       1024 * 40 * 5 /* original_network_content_length */, std::move(data2),
+       content::ResourceType::RESOURCE_TYPE_IMAGE, 0},
+  };
+
+  for (const auto& request : resources)
+    SimulateLoadedResource(request);
+
+  NavigateToUntrackedUrl();
+  ValidateUKM(true /* server_lofi_expected */, true /* client_lofi_expected */,
+              false /* lite_page_expected */, false /* opt_out_expected */);
+}
+
+TEST_F(PreviewsUKMObserverTest, BothLoFiOptOut) {
+  RunTest(true /* data_reduction_proxy_used */, false /* lite_page_received */);
+
+  std::unique_ptr<data_reduction_proxy::DataReductionProxyData> data1 =
+      base::MakeUnique<data_reduction_proxy::DataReductionProxyData>();
+  data1->set_used_data_reduction_proxy(true);
+  data1->set_lofi_received(true);
+
+  std::unique_ptr<data_reduction_proxy::DataReductionProxyData> data2 =
+      base::MakeUnique<data_reduction_proxy::DataReductionProxyData>();
+  data2->set_used_data_reduction_proxy(true);
+  data2->set_client_lofi_requested(true);
+
+  // Prepare 4 resources of varying size and configurations, 1 has Client LoFi,
+  // 1 has Server LoFi.
+  page_load_metrics::ExtraRequestCompleteInfo resources[] = {
+      // Uncached proxied request with .1 compression ratio.
+      {GURL(kDefaultTestUrl), net::HostPortPair(), -1, false /*was_cached*/,
+       1024 * 40 /* raw_body_bytes */,
+       1024 * 40 * 10 /* original_network_content_length */, std::move(data1),
+       content::ResourceType::RESOURCE_TYPE_IMAGE, 0},
+      // Uncached proxied request with .5 compression ratio.
+      {GURL(kDefaultTestUrl), net::HostPortPair(), -1, false /*was_cached*/,
+       1024 * 40 /* raw_body_bytes */,
+       1024 * 40 * 5 /* original_network_content_length */, std::move(data2),
+       content::ResourceType::RESOURCE_TYPE_IMAGE, 0},
+  };
+
+  for (const auto& request : resources)
+    SimulateLoadedResource(request);
+  observer()->BroadcastEventToObservers(
+      PreviewsInfoBarDelegate::OptOutEventKey());
+  NavigateToUntrackedUrl();
+  ValidateUKM(true /* server_lofi_expected */, true /* client_lofi_expected */,
+              false /* lite_page_expected */, true /* opt_out_expected */);
+}
+
+}  // namespace
+
+}  // namespace previews