class AccessAwareRouterTest

class AccessAwareRouterTest extends UnitTestCase {
    
    /**
     * @var \Symfony\Component\Routing\RouterInterface|\PHPUnit\Framework\MockObject\MockObject
     */
    protected RouterInterface|MockObject $router;
    
    /**
     * @var \Symfony\Component\Routing\Route
     */
    protected $route;
    
    /**
     * @var \Drupal\Core\Routing\Router|\PHPUnit\Framework\MockObject\MockObject
     */
    protected $coreRouter;
    
    /**
     * @var \Drupal\Core\Access\AccessManagerInterface|\PHPUnit\Framework\MockObject\MockObject
     */
    protected $accessManager;
    
    /**
     * @var \Drupal\Core\Session\AccountInterface|\PHPUnit\Framework\MockObject\MockObject
     */
    protected $currentUser;
    
    /**
     * @var \Drupal\Core\Routing\AccessAwareRouter
     */
    protected $accessAwareRouter;
    
    /**
     * {@inheritdoc}
     */
    protected function setUp() : void {
        parent::setUp();
        $this->route = new Route('test');
        $this->accessManager = $this->createMock('Drupal\\Core\\Access\\AccessManagerInterface');
        $this->currentUser = $this->createMock('Drupal\\Core\\Session\\AccountInterface');
    }
    
    /**
     * Sets up a chain router with matchRequest.
     */
    protected function setupRouter() : void {
        $this->router = $this->getMockBuilder('Drupal\\Core\\Routing\\Router')
            ->disableOriginalConstructor()
            ->getMock();
        $this->router
            ->expects($this->once())
            ->method('matchRequest')
            ->willReturn([
            RouteObjectInterface::ROUTE_OBJECT => $this->route,
        ]);
        $this->accessAwareRouter = new AccessAwareRouter($this->router, $this->accessManager, $this->currentUser);
    }
    
    /**
     * Tests the matchRequest() function for access allowed.
     */
    public function testMatchRequestAllowed() : void {
        $this->setupRouter();
        $request = new Request();
        $access_result = AccessResult::allowed();
        $this->accessManager
            ->expects($this->once())
            ->method('checkRequest')
            ->with($request)
            ->willReturn($access_result);
        $parameters = $this->accessAwareRouter
            ->matchRequest($request);
        $expected = [
            RouteObjectInterface::ROUTE_OBJECT => $this->route,
            AccessAwareRouterInterface::ACCESS_RESULT => $access_result,
        ];
        $this->assertSame($expected, $request->attributes
            ->all());
        $this->assertSame($expected, $parameters);
    }
    
    /**
     * Tests the matchRequest() function for access denied.
     */
    public function testMatchRequestDenied() : void {
        $this->setupRouter();
        $request = new Request();
        $access_result = AccessResult::forbidden();
        $this->accessManager
            ->expects($this->once())
            ->method('checkRequest')
            ->with($request)
            ->willReturn($access_result);
        $this->expectException(AccessDeniedHttpException::class);
        $this->accessAwareRouter
            ->matchRequest($request);
    }
    
    /**
     * Tests the matchRequest() function for access denied with reason message.
     */
    public function testCheckAccessResultWithReason() : void {
        $this->setupRouter();
        $request = new Request();
        $reason = $this->getRandomGenerator()
            ->string();
        $access_result = AccessResult::forbidden($reason);
        $this->accessManager
            ->expects($this->once())
            ->method('checkRequest')
            ->with($request)
            ->willReturn($access_result);
        $this->expectException(AccessDeniedHttpException::class);
        $this->expectExceptionMessage($reason);
        $this->accessAwareRouter
            ->matchRequest($request);
    }
    
    /**
     * Ensure that methods are passed to the wrapped router.
     *
     * @covers ::__call
     */
    public function testCall() : void {
        $mock_router = $this->createMock(RouterInterface::class);
        $this->router = $this->getMockBuilder(MockRouterInterface::class)
            ->disableOriginalConstructor()
            ->onlyMethods([
            'getRouteCollection',
            'match',
            'getContext',
            'setContext',
            'generate',
            'add',
        ])
            ->getMock();
        $this->router
            ->expects($this->once())
            ->method('add')
            ->with($mock_router)
            ->willReturnSelf();
        $this->accessAwareRouter = new AccessAwareRouter($this->router, $this->accessManager, $this->currentUser);
        $this->accessAwareRouter
            ->add($mock_router);
    }

}