Add Prerender type to NavigationType for nav timing 2

third_party/WebKit/Source/core/timing/PerformanceBaseTest.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/timing/Performance.h"
 
 #include "bindings/core/v8/PerformanceObserverCallback.h"
 #include "bindings/core/v8/V8BindingForTesting.h"
+#include "core/testing/DummyPageHolder.h"
 #include "core/timing/PerformanceBase.h"
 #include "core/timing/PerformanceLongTaskTiming.h"
 #include "core/timing/PerformanceObserver.h"
 #include "core/timing/PerformanceObserverInit.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace blink {
 
 class TestPerformanceBase : public PerformanceBase {
  public:
@@ skipping 16 matching lines @@
 class PerformanceBaseTest : public ::testing::Test {
  protected:
   void initialize(ScriptState* scriptState) {
     v8::Local<v8::Function> callback =
         v8::Function::New(scriptState->context(), nullptr).ToLocalChecked();
     m_base = new TestPerformanceBase();
     m_cb = PerformanceObserverCallback::create(scriptState, callback);
     m_observer = PerformanceObserver::create(scriptState, m_base, m_cb);
   }
 
+  void SetUp() override {
+    m_pageHolder = DummyPageHolder::create(IntSize(800, 600));
+  }
+
   int numPerformanceEntriesInObserver() {
     return m_observer->m_performanceEntries.size();
   }
 
+  PerformanceNavigationTiming::NavigationType getNavigationType(
+      NavigationType type,
+      Document* document) {
+    return PerformanceBase::getNavigationType(type, document);
+  }
+
   Persistent<TestPerformanceBase> m_base;
   Persistent<PerformanceObserver> m_observer;
+  std::unique_ptr<DummyPageHolder> m_pageHolder;
   Persistent<PerformanceObserverCallback> m_cb;
 };
 
 TEST_F(PerformanceBaseTest, Register) {
   V8TestingScope scope;
   initialize(scope.getScriptState());
 
   EXPECT_EQ(0, m_base->numObservers());
   EXPECT_EQ(0, m_base->numActiveObservers());
 
@@ skipping 41 matching lines @@
   Vector<String> entryTypeVec;
   entryTypeVec.append("longtask");
   options.setEntryTypes(entryTypeVec);
   m_observer->observe(options, exceptionState);
 
   EXPECT_TRUE(m_base->hasPerformanceObserverFor(PerformanceEntry::LongTask));
   // Add a long task entry
   m_base->addLongTaskTiming(1234, 5678, "www.foo.com/bar", nullptr);
   EXPECT_EQ(1, numPerformanceEntriesInObserver());  // added an entry
 }
+
+TEST_F(PerformanceBaseTest, GetNavigationType) {
+  m_pageHolder->page().setVisibilityState(PageVisibilityStatePrerender, false);
+  PerformanceNavigationTiming::NavigationType returnedType =
+      getNavigationType(NavigationTypeBackForward, &m_pageHolder->document());
+  EXPECT_EQ(returnedType,
+            PerformanceNavigationTiming::NavigationType::Prerender);
+
+  m_pageHolder->page().setVisibilityState(PageVisibilityStateHidden, false);
+  returnedType =
+      getNavigationType(NavigationTypeBackForward, &m_pageHolder->document());
+  EXPECT_EQ(returnedType,
+            PerformanceNavigationTiming::NavigationType::BackForward);
+
+  m_pageHolder->page().setVisibilityState(PageVisibilityStateVisible, false);
+  returnedType = getNavigationType(NavigationTypeFormResubmitted,
+                                   &m_pageHolder->document());
+  EXPECT_EQ(returnedType,
+            PerformanceNavigationTiming::NavigationType::Navigate);
+}
 }  // namespace blink