Refactor InspectorWebPerfAgent: update lifecycle management to be per Local Frame root; replace hea…

third_party/WebKit/Source/core/frame/LocalFrameTest.cpp

 // Copyright 2016 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 "core/frame/LocalFrame.h"
 
 #include "core/editing/FrameSelection.h"
 #include "core/frame/FrameHost.h"
 #include "core/frame/FrameView.h"
 #include "core/frame/VisualViewport.h"
 #include "core/html/HTMLElement.h"
+#include "core/inspector/InspectorWebPerfAgent.h"
 #include "core/layout/LayoutObject.h"
 #include "core/testing/DummyPageHolder.h"
+#include "core/timing/Performance.h"
 #include "platform/DragImage.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace blink {
 
 class LocalFrameTest : public ::testing::Test {
  protected:
   LocalFrameTest() = default;
   ~LocalFrameTest() override = default;
 
   Document& document() const { return m_dummyPageHolder->document(); }
   LocalFrame& frame() const { return *document().frame(); }
+  Performance* performance() const { return m_performance; }
 
   void setBodyContent(const std::string& bodyContent) {
     document().body()->setInnerHTML(String::fromUTF8(bodyContent.c_str()),
                                     ASSERT_NO_EXCEPTION);
     updateAllLifecyclePhases();
   }
 
   void updateAllLifecyclePhases() {
     document().view()->updateAllLifecyclePhases();
   }
 
+  bool hasWebPerformanceAgent() { return frame().m_inspectorWebPerfAgent; }
+  bool hasWebPerformanceAgentObservers() {
+    return frame().m_inspectorWebPerfAgent->hasWebPerformanceObservers();
+  }
+
  private:
   void SetUp() override {
     m_dummyPageHolder = DummyPageHolder::create(IntSize(800, 600));
+    m_performance = Performance::create(&frame());
   }
 
   std::unique_ptr<DummyPageHolder> m_dummyPageHolder;
+  Persistent<Performance> m_performance;
 };
 
 TEST_F(LocalFrameTest, nodeImage) {
   setBodyContent(
       "<style>"
       "#sample { width: 100px; height: 100px; }"
       "</style>"
       "<div id=sample></div>");
   Element* sample = document().getElementById("sample");
   const std::unique_ptr<DragImage> image = frame().nodeImage(*sample);
@@ skipping 78 matching lines @@
   frame().selection().selectAll();
   updateAllLifecyclePhases();
   const std::unique_ptr<DragImage> image2(frame().dragImageForSelection(0.75f));
 
   EXPECT_GT(image1->size().width(), 0);
   EXPECT_GT(image1->size().height(), 0);
   EXPECT_EQ(image1->size().width() * 2, image2->size().width());
   EXPECT_EQ(image1->size().height() * 2, image2->size().height());
 }
 
+TEST_F(LocalFrameTest, LongTaskObserverInstrumentation) {
+  EXPECT_FALSE(hasWebPerformanceAgent());
+  frame().enableInspectorWebPerfAgent(performance());
+  EXPECT_TRUE(hasWebPerformanceAgent());
+  EXPECT_TRUE(hasWebPerformanceAgentObservers());
+
+  frame().disableInspectorWebPerfAgent(performance());
+  EXPECT_FALSE(hasWebPerformanceAgent());
+}
+
 }  // namespace blink