1
    2
    3
    4
    5
    6
    7
    8
    9
   10
   11
   12
   13
   14
   15
   16
   17
   18
   19
   20
   21
   22
   23
   24
   25
   26
   27
   28
   29
   30
   31
   32
   33
   34
   35
   36
   37
   38
   39
   40
   41
   42
   43
   44
   45
   46
   47
   48
   49
   50
   51
   52
   53
   54
   55
   56
   57
   58
   59
   60
   61
   62
   63
   64
   65
   66
   67
   68
   69
   70
   71


content / browser / picture_in_picture / document_picture_in_picture_navigation_throttle_unittest.cc [blame]

// Copyright 2023 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "content/browser/renderer_host/render_frame_host_impl.h"
#include "content/public/test/test_renderer_host.h"
#include "content/test/navigation_simulator_impl.h"
#include "content/test/test_web_contents.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace content {

class DocumentPictureInPictureNavigationThrottleTest
    : public RenderViewHostTestHarness {
 public:
  DocumentPictureInPictureNavigationThrottleTest()
      : RenderViewHostTestHarness(
            base::test::TaskEnvironment::TimeSource::MOCK_TIME) {}
  DocumentPictureInPictureNavigationThrottleTest(
      const DocumentPictureInPictureNavigationThrottleTest&) = delete;
  DocumentPictureInPictureNavigationThrottleTest& operator=(
      const DocumentPictureInPictureNavigationThrottleTest&) = delete;
  ~DocumentPictureInPictureNavigationThrottleTest() override = default;
};

TEST_F(DocumentPictureInPictureNavigationThrottleTest,
       ClosesPictureInPictureWindowOnNavigationStart) {
  // Simulate that we're inside a document picture-in-picture window.
  blink::mojom::PictureInPictureWindowOptions options;
  static_cast<TestWebContents*>(web_contents())
      ->SetPictureInPictureOptions(options);

  // Simulate a navigation, which should fail.
  auto nav_simulator = NavigationSimulatorImpl::CreateRendererInitiated(
      GURL("https://example.test/"), main_rfh());
  nav_simulator->Start();
  EXPECT_TRUE(nav_simulator->HasFailed());

  // The RenderFrameHost should be asynchronously asked to close. First,
  // RunUntilIdle() to request the close, and then wait out the unload timer via
  // FastForwardBy().
  task_environment()->RunUntilIdle();
  task_environment()->FastForwardBy(base::Milliseconds(500));
  EXPECT_TRUE(
      static_cast<RenderFrameHostImpl*>(main_rfh())->IsPageReadyToBeClosed());
}

TEST_F(DocumentPictureInPictureNavigationThrottleTest,
       SameDocumentNavigationsOkay) {
  auto* committed_rfh = NavigationSimulatorImpl::NavigateAndCommitFromDocument(
      GURL("https://example.test/"), main_rfh());
  EXPECT_EQ(committed_rfh, main_rfh());
  EXPECT_EQ(GURL("https://example.test/"), main_rfh()->GetLastCommittedURL());

  // Simulate that we're inside a document picture-in-picture window.
  blink::mojom::PictureInPictureWindowOptions options;
  static_cast<TestWebContents*>(web_contents())
      ->SetPictureInPictureOptions(options);

  // Simulate a same-document navigation, which should succeed.
  {
    auto nav_simulator = NavigationSimulatorImpl::CreateRendererInitiated(
        GURL("https://example.test/#"), main_rfh());
    nav_simulator->CommitSameDocument();
    EXPECT_FALSE(nav_simulator->HasFailed());
    EXPECT_EQ(GURL("https://example.test/#"),
              main_rfh()->GetLastCommittedURL());
  }
}

}  // namespace content