Extract timing info from distiller result

components/dom_distiller/core/distiller_unittest.cc

 // Copyright 2013 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 <algorithm>
 #include <map>
 #include <string>
 #include <vector>
 
 #include "base/bind.h"
@@ skipping 16 matching lines @@
 #include "third_party/dom_distiller_js/dom_distiller_json_converter.h"
 
 using std::vector;
 using std::string;
 using ::testing::Invoke;
 using ::testing::Return;
 using ::testing::_;
 
 using dom_distiller::proto::DomDistillerOptions;
 using dom_distiller::proto::DomDistillerResult;
+using dom_distiller::proto::TimingEntry;
 
 namespace {
 const char kTitle[] = "Title";
 const char kContent[] = "Content";
 const char kURL[] = "http://a.com/";
 const size_t kTotalImages = 2;
 const char* kImageURLs[kTotalImages] = {"http://a.com/img1.jpg",
                                         "http://a.com/img2.jpg"};
 const char* kImageData[kTotalImages] = {"abcde", "12345"};
 const char kDebugLog[] = "Debug Log";
@@ skipping 306 matching lines @@
   scoped_ptr<base::Value> result =
       dom_distiller::proto::json::DomDistillerResult::WriteToValue(dd_result);
   distiller_.reset(
       new DistillerImpl(url_fetcher_factory_, DomDistillerOptions()));
   DistillPage(kURL, CreateMockDistillerPage(result.get(), GURL(kURL)).Pass());
   base::MessageLoop::current()->RunUntilIdle();
   const DistilledPageProto& first_page = article_proto_->pages(0);
   EXPECT_EQ(kDebugLog, first_page.debug_info().log());
 }
 
+void SetTimingEntry(TimingEntry* entry, const std::string& name, double time) {
+  entry->set_name(name);
+  entry->set_time(time);
+}
+
+TEST_F(DistillerTest, DistillPageWithTimingInfo) {
+  base::MessageLoopForUI loop;
+  DomDistillerResult dd_result;
+  dd_result.mutable_timing_info()->set_total_time(1.0);
+  dd_result.mutable_timing_info()->set_markup_parsing_time(2.0);
+  dd_result.mutable_timing_info()->set_document_construction_time(3.0);
+  dd_result.mutable_timing_info()->set_article_processing_time(4.0);
+  dd_result.mutable_timing_info()->set_formatting_time(5.0);
+  SetTimingEntry(
+      dd_result.mutable_timing_info()->add_other_times(), "time0", 6.0);
+  SetTimingEntry(
+      dd_result.mutable_timing_info()->add_other_times(), "time1", 7.0);
+  scoped_ptr<base::Value> result =
+      dom_distiller::proto::json::DomDistillerResult::WriteToValue(dd_result);
+  distiller_.reset(
+      new DistillerImpl(url_fetcher_factory_, DomDistillerOptions()));
+  DistillPage(kURL, CreateMockDistillerPage(result.get(), GURL(kURL)).Pass());
+  base::MessageLoop::current()->RunUntilIdle();
+  const DistilledPageProto& first_page = article_proto_->pages(0);
+  std::map<std::string, double> timings;
+  for (int i = 0; i < first_page.timing_info_size(); ++i) {
+    DistilledPageProto::TimingInfo timing = first_page.timing_info(i);
+    timings[timing.name()] = timing.time();
+  }
+  EXPECT_EQ(7u, timings.size());
+  EXPECT_EQ(1.0, timings["total"]);
+  EXPECT_EQ(2.0, timings["markup_parsing"]);
+  EXPECT_EQ(3.0, timings["document_construction"]);
+  EXPECT_EQ(4.0, timings["article_processing"]);
+  EXPECT_EQ(5.0, timings["formatting"]);
+  EXPECT_EQ(6.0, timings["time0"]);
+  EXPECT_EQ(7.0, timings["time1"]);
+}
+
 TEST_F(DistillerTest, DistillPageWithImages) {
   base::MessageLoopForUI loop;
   vector<int> image_indices;
   image_indices.push_back(0);
   image_indices.push_back(1);
   scoped_ptr<base::Value> result =
       CreateDistilledValueReturnedFromJS(kTitle, kContent, image_indices, "");
   distiller_.reset(
       new DistillerImpl(url_fetcher_factory_, DomDistillerOptions()));
   DistillPage(kURL, CreateMockDistillerPage(result.get(), GURL(kURL)).Pass());
@@ skipping 272 matching lines @@
 
   ASSERT_TRUE(distiller_page);
   // Post the task to execute javascript and then delete the distiller.
   distiller_page->OnDistillationDone(GURL(kURL), distilled_value.get());
   distiller_.reset();
 
   base::MessageLoop::current()->RunUntilIdle();
 }
 
 }  // namespace dom_distiller