[Page Load Metrics] PageLoadMetrics Mojofication.

chrome/renderer/page_load_metrics/metrics_render_frame_observer_unittest.cc

 // Copyright (c) 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 "chrome/renderer/page_load_metrics/metrics_render_frame_observer.h"
 
 #include <memory>
 #include <utility>
 
 #include "base/memory/ptr_util.h"
+#include "base/message_loop/message_loop.h"
+#include "base/run_loop.h"
 #include "base/time/time.h"
 #include "base/timer/mock_timer.h"
 #include "chrome/common/page_load_metrics/page_load_timing.h"
+#include "chrome/renderer/page_load_metrics/fake_page_load_metrics.h"
 #include "chrome/renderer/page_load_metrics/fake_page_timing_metrics_ipc_sender.h"
+#include "mojo/public/cpp/bindings/interface_request.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace page_load_metrics {
 
 // Implementation of the MetricsRenderFrameObserver class we're testing,
 // with the GetTiming() and ShouldSendMetrics() methods stubbed out to make
 // the rest of the class more testable.
 class TestMetricsRenderFrameObserver : public MetricsRenderFrameObserver {
  public:
   TestMetricsRenderFrameObserver() : MetricsRenderFrameObserver(nullptr) {}
 
   std::unique_ptr<base::Timer> CreateTimer() const override {
     if (!mock_timer_)
       ADD_FAILURE() << "CreateTimer() called, but no MockTimer available.";
     return std::move(mock_timer_);
   }
 
   // We intercept sent messages and dispatch them to our
+  // FakePageLoadMetricsImpl, which we use to verify that the expected timing
+  // get sent.
+  void SendTiming(const PageLoadTiming& timing,
+                  const PageLoadMetadata& metadata) override {
+    if (!page_load_metrics_ptr_) {
+      mojom::PageLoadMetricsRequest request =
+          MakeRequest(&page_load_metrics_ptr_);
+      fake_page_load_metrics_.Bind(std::move(request));
+    }
+    page_load_metrics_ptr_->UpdateTiming(timing, metadata);
+    base::RunLoop().RunUntilIdle();
+  }
+
+  // We intercept sent messages and dispatch them to our
   // FakePageTimingMetricsIPCSender, which we use to verify that the expected
   // IPC messages get sent.
   bool Send(IPC::Message* message) override {
     return fake_timing_ipc_sender_.Send(message);
   }
 
   void set_mock_timer(std::unique_ptr<base::Timer> timer) {
     ASSERT_EQ(nullptr, mock_timer_);
     mock_timer_ = std::move(timer);
   }
 
   void set_is_main_frame(bool is_main_frame) { is_main_frame_ = is_main_frame; }
 
   bool WasFakeTimingConsumed() const { return fake_timing_.IsEmpty(); }
 
   void ExpectPageLoadTiming(const PageLoadTiming& timing) {
     SetFakePageLoadTiming(timing);
     fake_timing_ipc_sender_.ExpectPageLoadTiming(timing);
+    fake_page_load_metrics_.ExpectPageLoadTiming(timing);
   }
 
   void SetFakePageLoadTiming(const PageLoadTiming& timing) {
     EXPECT_TRUE(fake_timing_.IsEmpty());
     fake_timing_ = timing;
   }
 
   PageLoadTiming GetTiming() const override {
     PageLoadTiming tmp = fake_timing_;
     fake_timing_ = PageLoadTiming();
     return tmp;
   }
 
   void VerifyExpectedTimings() const {
     EXPECT_TRUE(fake_timing_.IsEmpty());
     fake_timing_ipc_sender_.VerifyExpectedTimings();
+    fake_page_load_metrics_.VerifyExpectedTimings();
   }
 
   bool ShouldSendMetrics() const override { return true; }
   bool HasNoRenderFrame() const override { return false; }
   bool IsMainFrame() const override { return is_main_frame_; }
 
  private:
+  base::MessageLoop loop_;
   FakePageTimingMetricsIPCSender fake_timing_ipc_sender_;
+  mojom::PageLoadMetricsPtr page_load_metrics_ptr_;
+  FakePageLoadMetricsImpl fake_page_load_metrics_;
   mutable PageLoadTiming fake_timing_;
   mutable std::unique_ptr<base::Timer> mock_timer_;
   bool is_main_frame_ = true;
 };
 
 typedef testing::Test MetricsRenderFrameObserverTest;
 
 TEST_F(MetricsRenderFrameObserverTest, NoMetrics) {
   TestMetricsRenderFrameObserver observer;
   base::MockTimer* mock_timer = new base::MockTimer(false, false);
@@ skipping 135 matching lines @@
   observer.DidCommitProvisionalLoad(true, false);
   ASSERT_FALSE(observer.WasFakeTimingConsumed());
   ASSERT_FALSE(mock_timer->IsRunning());
 
   observer.DidChangePerformanceTiming();
   ASSERT_FALSE(observer.WasFakeTimingConsumed());
   ASSERT_FALSE(mock_timer->IsRunning());
 }
 
 }  // namespace page_load_metrics