class RouteSubscriberTest

class RouteSubscriberTest extends UnitTestCase {

  /**
   * The mocked entity type manager.
   *
   * @var \Drupal\Core\Entity\EntityTypeManagerInterface|\PHPUnit\Framework\MockObject\MockObject
   */
  protected $entityTypeManager;

  /**
   * The mocked config entity storage.
   *
   * @var \Drupal\Core\Config\Entity\ConfigEntityStorageInterface|\PHPUnit\Framework\MockObject\MockObject
   */
  protected $viewStorage;

  /**
   * The tested views route subscriber.
   *
   * @var \Drupal\views\EventSubscriber\RouteSubscriber|\Drupal\Tests\views\Unit\EventSubscriber\TestRouteSubscriber
   */
  protected $routeSubscriber;

  /**
   * The mocked key value storage.
   *
   * @var \Drupal\Core\State\StateInterface|\PHPUnit\Framework\MockObject\MockObject
   */
  protected $state;

  /**
   * {@inheritdoc}
   */
  protected function setUp() : void {
    parent::setUp();
    $this->entityTypeManager = $this
      ->createMock(EntityTypeManagerInterface::class);
    $this->viewStorage = $this
      ->getMockBuilder('Drupal\\Core\\Config\\Entity\\ConfigEntityStorage')
      ->disableOriginalConstructor()
      ->getMock();
    $this->entityTypeManager
      ->expects($this
      ->any())
      ->method('getStorage')
      ->with('view')
      ->willReturn($this->viewStorage);
    $this->state = $this
      ->createMock('\\Drupal\\Core\\State\\StateInterface');
    $this->routeSubscriber = new TestRouteSubscriber($this->entityTypeManager, $this->state);
  }

  /**
   * @covers ::routeRebuildFinished
   */
  public function testRouteRebuildFinished() {
    [
      $display_1,
      $display_2,
    ] = $this
      ->setupMocks();
    $display_1
      ->expects($this
      ->once())
      ->method('collectRoutes')
      ->willReturn([
      'test_id.page_1' => 'views.test_id.page_1',
    ]);
    $display_2
      ->expects($this
      ->once())
      ->method('collectRoutes')
      ->willReturn([
      'test_id.page_2' => 'views.test_id.page_2',
    ]);
    $this->routeSubscriber
      ->routes();
    $this->state
      ->expects($this
      ->once())
      ->method('set')
      ->with('views.view_route_names', [
      'test_id.page_1' => 'views.test_id.page_1',
      'test_id.page_2' => 'views.test_id.page_2',
    ]);
    $this->routeSubscriber
      ->routeRebuildFinished();
  }

  /**
   * Tests the onAlterRoutes method.
   *
   * @see \Drupal\views\EventSubscriber\RouteSubscriber::onAlterRoutes()
   */
  public function testOnAlterRoutes() {
    $collection = new RouteCollection();

    // The first route will be overridden later.
    $collection
      ->add('test_route', new Route('test_route', [
      '_controller' => 'Drupal\\Tests\\Core\\Controller\\TestController',
    ]));
    $route_2 = new Route('test_route/example', [
      '_controller' => 'Drupal\\Tests\\Core\\Controller\\TestController',
    ]);
    $collection
      ->add('test_route_2', $route_2);
    $route_event = new RouteBuildEvent($collection);
    [
      $display_1,
      $display_2,
    ] = $this
      ->setupMocks();

    // The page_1 display overrides an existing route, so the dynamicRoutes
    // should only call the second display.
    $display_1
      ->expects($this
      ->once())
      ->method('collectRoutes')
      ->willReturnCallback(function () use ($collection) {
      $collection
        ->add('views.test_id.page_1', new Route('test_route', [
        '_controller' => 'Drupal\\views\\Routing\\ViewPageController',
      ]));
      return [
        'test_id.page_1' => 'views.test_id.page_1',
      ];
    });
    $display_1
      ->expects($this
      ->once())
      ->method('alterRoutes')
      ->willReturn([
      'test_id.page_1' => 'test_route',
    ]);
    $display_2
      ->expects($this
      ->once())
      ->method('collectRoutes')
      ->willReturnCallback(function () use ($collection) {
      $collection
        ->add('views.test_id.page_2', new Route('test_route', [
        '_controller' => 'Drupal\\views\\Routing\\ViewPageController',
      ]));
      return [
        'test_id.page_2' => 'views.test_id.page_2',
      ];
    });
    $display_2
      ->expects($this
      ->once())
      ->method('alterRoutes')
      ->willReturn([]);

    // Ensure that even both the collectRoutes() and alterRoutes() methods
    // are called on the displays, we ensure that the route first defined by
    // views is dropped.
    $this->routeSubscriber
      ->routes();
    $this
      ->assertNull($this->routeSubscriber
      ->onAlterRoutes($route_event));
    $this->state
      ->expects($this
      ->once())
      ->method('set')
      ->with('views.view_route_names', [
      'test_id.page_1' => 'test_route',
      'test_id.page_2' => 'views.test_id.page_2',
    ]);
    $collection = $route_event
      ->getRouteCollection();
    $this
      ->assertEquals([
      'test_route',
      'test_route_2',
      'views.test_id.page_2',
    ], array_keys($collection
      ->all()));
    $this->routeSubscriber
      ->routeRebuildFinished();
  }

  /**
   * Sets up mocks of Views objects needed for testing.
   *
   * @return \Drupal\views\Plugin\views\display\DisplayRouterInterface[]|\PHPUnit\Framework\MockObject\MockObject[]
   *   An array of two mocked view displays.
   */
  protected function setupMocks() {
    $executable = $this
      ->getMockBuilder('Drupal\\views\\ViewExecutable')
      ->disableOriginalConstructor()
      ->getMock();
    $view = $this
      ->getMockBuilder('Drupal\\views\\Entity\\View')
      ->disableOriginalConstructor()
      ->getMock();
    $this->viewStorage
      ->expects($this
      ->any())
      ->method('load')
      ->willReturn($view);
    $view
      ->expects($this
      ->any())
      ->method('getExecutable')
      ->willReturn($executable);
    $view
      ->expects($this
      ->any())
      ->method('id')
      ->willReturn('test_id');
    $executable->storage = $view;
    $executable
      ->expects($this
      ->any())
      ->method('setDisplay')
      ->willReturnMap([
      [
        'page_1',
        TRUE,
      ],
      [
        'page_2',
        TRUE,
      ],
      [
        'page_3',
        FALSE,
      ],
    ]);

    // Ensure that only the first two displays are actually called.
    $display_1 = $this
      ->createMock('Drupal\\views\\Plugin\\views\\display\\DisplayRouterInterface');
    $display_2 = $this
      ->createMock('Drupal\\views\\Plugin\\views\\display\\DisplayRouterInterface');
    $display_collection = $this
      ->getMockBuilder('Drupal\\views\\DisplayPluginCollection')
      ->disableOriginalConstructor()
      ->getMock();
    $display_collection
      ->expects($this
      ->any())
      ->method('get')
      ->willReturnMap([
      [
        'page_1',
        $display_1,
      ],
      [
        'page_2',
        $display_2,
      ],
    ]);
    $executable->displayHandlers = $display_collection;
    $this->routeSubscriber->applicableViews = [];
    $this->routeSubscriber->applicableViews[] = [
      'test_id',
      'page_1',
    ];
    $this->routeSubscriber->applicableViews[] = [
      'test_id',
      'page_2',
    ];
    $this->routeSubscriber->applicableViews[] = [
      'test_id',
      'page_3',
    ];
    return [
      $display_1,
      $display_2,
    ];
  }

}