Introducing phased profiling framework

components/metrics/profiler/tracking_synchronizer_unittest.cc

Index: components/metrics/profiler/tracking_synchronizer_unittest.cc
diff --git a/components/metrics/profiler/tracking_synchronizer_unittest.cc b/components/metrics/profiler/tracking_synchronizer_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..a597a477c64a0f5a8cdb6303bd24231149c8928a
--- /dev/null
+++ b/components/metrics/profiler/tracking_synchronizer_unittest.cc
@@ -0,0 +1,118 @@
+// Copyright 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 "base/memory/ref_counted.h"
+#include "base/tracked_objects.h"
+#include "components/metrics/profiler/tracking_synchronizer.h"
+#include "components/metrics/profiler/tracking_synchronizer_observer.h"
+#include "content/public/test/test_browser_thread_bundle.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+using tracked_objects::ProcessDataPhaseSnapshot;
+using tracked_objects::TaskSnapshot;
+
+namespace metrics {
+
+namespace {
+
+class TestObserver : public TrackingSynchronizerObserver {
+ public:
+  TestObserver() {}
+
+  ~TestObserver() override {
+    EXPECT_TRUE(got_phase_0_);
+    EXPECT_TRUE(got_phase_1_);
+  }
+
+  void ReceivedProfilerData(
+      const tracked_objects::ProcessDataPhaseSnapshot& process_data_phase,
+      base::ProcessId process_id,
+      content::ProcessType process_type,
+      int profiling_phase,
+      base::TimeDelta phase_start,
+      base::TimeDelta phase_finish,
+      const ProfilerEvents& past_events) override {
+    EXPECT_EQ(static_cast<base::ProcessId>(239), process_id);
+    EXPECT_EQ(content::ProcessType::PROCESS_TYPE_PLUGIN, process_type);
+    ASSERT_EQ(1u, process_data_phase.tasks.size());
+
+    switch (profiling_phase) {
+      case 0:
+        EXPECT_FALSE(got_phase_0_);
+        got_phase_0_ = true;
+
+        EXPECT_EQ(base::TimeDelta::FromMilliseconds(0), phase_start);
+        EXPECT_EQ(base::TimeDelta::FromMilliseconds(222), phase_finish);
+
+        EXPECT_EQ("death_thread0",
+                  process_data_phase.tasks[0].death_thread_name);
+        EXPECT_EQ(0u, past_events.size());
+        break;
+
+      case 1:
+        EXPECT_FALSE(got_phase_1_);
+        got_phase_1_ = true;
+
+        EXPECT_EQ(base::TimeDelta::FromMilliseconds(222), phase_start);
+        EXPECT_EQ(base::TimeDelta::FromMilliseconds(666), phase_finish);
+
+        EXPECT_EQ("death_thread1",
+                  process_data_phase.tasks[0].death_thread_name);
+        ASSERT_EQ(1u, past_events.size());
+        EXPECT_EQ(ProfilerEventProto::EVENT_FIRST_NONEMPTY_PAINT,
+                  past_events[0]);
+        break;
+
+      default:
+        bool profiling_phase_is_neither_0_nor_1 = true;
+        EXPECT_FALSE(profiling_phase_is_neither_0_nor_1);
+    }
+  }
+
+ private:
+  bool got_phase_0_ = false;
+  bool got_phase_1_ = false;
+
+  DISALLOW_COPY_AND_ASSIGN(TestObserver);
+};
+
+base::TimeTicks TestTimeFromMs(int64 ms) {
+  return base::TimeTicks() + base::TimeDelta::FromMilliseconds(ms);
+}
+
+}  // namespace
+
+TEST(TrackingSynchronizerTest, ProfilerData) {
+  // Testing how TrackingSynchronizer reports 2 phases of profiling.
+#if !defined(OS_IOS)
+  content::TestBrowserThreadBundle thread_bundle;
+#endif
+  scoped_refptr<TrackingSynchronizer> tracking_synchronizer =
+      new TrackingSynchronizer(TestTimeFromMs(111));
+
+  // Mimic a phase change event.
+  tracking_synchronizer->phase_completion_events_sequence_.push_back(
+      ProfilerEventProto::EVENT_FIRST_NONEMPTY_PAINT);
+  tracking_synchronizer->phase_start_times_.push_back(TestTimeFromMs(333));
+
+  tracked_objects::ProcessDataSnapshot profiler_data;
+  ProcessDataPhaseSnapshot snapshot0;
+  tracked_objects::TaskSnapshot task_snapshot0;
+  task_snapshot0.death_thread_name = "death_thread0";
+  snapshot0.tasks.push_back(task_snapshot0);
+  ProcessDataPhaseSnapshot snapshot1;
+  profiler_data.phased_process_data_snapshots[0] = snapshot0;
+  tracked_objects::TaskSnapshot task_snapshot1;
+  task_snapshot1.death_thread_name = "death_thread1";
+  snapshot1.tasks.push_back(task_snapshot1);
+  profiler_data.phased_process_data_snapshots[1] = snapshot1;
+  profiler_data.process_id = 239;
+
+  TestObserver test_observer;
+  tracking_synchronizer->SendData(profiler_data,
+                                  content::ProcessType::PROCESS_TYPE_PLUGIN,
+                                  TestTimeFromMs(777), &test_observer);
+}
+
+}  // namespace metrics