Adding DRP specfic UMA for FirstContentfulPaint

chrome/browser/renderer_host/chrome_resource_dispatcher_host_delegate_unittest.cc

+// Copyright 2014 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/renderer_host/chrome_resource_dispatcher_host_delegate.h"
+
+#include <memory>
+
+#include "base/message_loop/message_loop.h"
+#include "chrome/browser/renderer_host/chrome_navigation_data.h"
+#include "components/data_reduction_proxy/core/browser/data_reduction_proxy_data.h"
+#include "content/public/browser/navigation_data.h"
+#include "content/public/test/test_browser_thread.h"
+#include "net/base/request_priority.h"
+#include "net/url_request/url_request.h"
+#include "net/url_request/url_request_context.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+class ChromeResourceDispatcherHostDelegateTest : public testing::Test {
+ public:
+  ChromeResourceDispatcherHostDelegateTest()
+      : ui_thread_(content::BrowserThread::UI, &message_loop_) {}
+  ~ChromeResourceDispatcherHostDelegateTest() override {}
+
+ private:
+  base::MessageLoopForIO message_loop_;
+  content::TestBrowserThread ui_thread_;
+};
+
+TEST_F(ChromeResourceDispatcherHostDelegateTest, GetNavigationDataWithDRPData) {
+  std::unique_ptr<net::URLRequestContext> context(new net::URLRequestContext());
+  std::unique_ptr<net::URLRequest> fake_request(context->CreateRequest(
+      GURL("content.request.any.port"), net::RequestPriority::IDLE, nullptr));
+  // Add DRP data to URLRequest
+  data_reduction_proxy::DataReductionProxyData* drp_data =
+      data_reduction_proxy::DataReductionProxyData::DataForRequest(
+          fake_request.get(), true);
+  drp_data->set_used_data_reduction_proxy(true);
+  std::unique_ptr<ChromeResourceDispatcherHostDelegate> delegate(
+      new ChromeResourceDispatcherHostDelegate());
+  std::unique_ptr<content::NavigationData> navigation_data =
+      delegate->GetNavigationData(fake_request.get());
+  ChromeNavigationData* chrome_navigation_data =
+      static_cast<ChromeNavigationData*>(navigation_data.get());
+  data_reduction_proxy::DataReductionProxyData* drp_data_copy =
+      chrome_navigation_data->GetDataReductionProxyData();
+  // The DRP_data should be a copy off the one on URLRequest
+  EXPECT_NE(drp_data_copy, drp_data);
+  // Make sure DRP Data was copied.
+  EXPECT_TRUE(drp_data_copy->get_used_data_reduction_proxy());
+}
+
+TEST_F(ChromeResourceDispatcherHostDelegateTest,
+       GetNavigationDataWithoutDRPData) {
+  std::unique_ptr<net::URLRequestContext> context(new net::URLRequestContext());
+  std::unique_ptr<net::URLRequest> fake_request(context->CreateRequest(
+      GURL("content.request.any.port"), net::RequestPriority::IDLE, nullptr));
+  std::unique_ptr<ChromeResourceDispatcherHostDelegate> delegate(
+      new ChromeResourceDispatcherHostDelegate());
+  std::unique_ptr<content::NavigationData> navigation_data =
+      delegate->GetNavigationData(fake_request.get());
+  ChromeNavigationData* chrome_navigation_data =
+      static_cast<ChromeNavigationData*>(navigation_data.get());
+  EXPECT_TRUE(!chrome_navigation_data->GetDataReductionProxyData());
+}