class NegotiationMiddlewareTest

class NegotiationMiddlewareTest extends UnitTestCase {
    
    /**
     * @var \Symfony\Component\HttpKernel\HttpKernelInterface
     */
    protected $app;
    
    /**
     * @var \Drupal\Tests\Core\StackMiddleware\StubNegotiationMiddleware
     */
    protected $contentNegotiation;
    
    /**
     * {@inheritdoc}
     */
    protected function setUp() : void {
        parent::setUp();
        $this->app = $this->prophesize(MockedHttpKernelInterface::class);
        $this->contentNegotiation = new StubNegotiationMiddleware($this->app
            ->reveal());
    }
    
    /**
     * Tests the getContentType() method with AJAX iframe upload.
     *
     * @covers ::getContentType
     */
    public function testAjaxIframeUpload() {
        $request = new Request();
        $request->request
            ->set('ajax_iframe_upload', '1');
        $this->assertSame('iframeupload', $this->contentNegotiation
            ->getContentType($request));
    }
    
    /**
     * Tests the specifying a format via query parameters gets used.
     *
     * @covers ::getContentType
     */
    public function testFormatViaQueryParameter() {
        $request = new Request();
        $request->query
            ->set('_format', 'bob');
        $this->assertSame('bob', $this->contentNegotiation
            ->getContentType($request));
    }
    
    /**
     * Tests the getContentType() method when no priority format is found.
     *
     * @covers ::getContentType
     */
    public function testUnknownContentTypeReturnsNull() {
        $request = new Request();
        $this->assertNull($this->contentNegotiation
            ->getContentType($request));
    }
    
    /**
     * Tests the getContentType() method when no priority format is found but it's an AJAX request.
     *
     * @covers ::getContentType
     */
    public function testUnknownContentTypeButAjaxRequest() {
        $request = new Request();
        $request->headers
            ->set('X-Requested-With', 'XMLHttpRequest');
        $this->assertNull($this->contentNegotiation
            ->getContentType($request));
    }
    
    /**
     * Tests that handle() correctly hands off to sub application.
     *
     * @covers ::handle
     */
    public function testHandle() {
        $request = $this->prophesize(Request::class);
        // Default empty format list should not set any formats.
        $request->setFormat()
            ->shouldNotBeCalled();
        // Request format will be set with default format.
        $request->setRequestFormat()
            ->shouldNotBeCalled();
        // Some getContentType calls we don't really care about but have to mock.
        $request_data = $this->prophesize(ParameterBag::class);
        $request_data->get('ajax_iframe_upload', FALSE)
            ->shouldBeCalled();
        $request_mock = $request->reveal();
        $request_mock->query = new ParameterBag([]);
        $request_mock->request = $request_data->reveal();
        // Calling kernel app with default arguments.
        $this->app
            ->handle($request_mock, HttpKernelInterface::MASTER_REQUEST, TRUE)
            ->shouldBeCalled();
        $this->contentNegotiation
            ->handle($request_mock);
        // Calling kernel app with specified arguments.
        $this->app
            ->handle($request_mock, HttpKernelInterface::SUB_REQUEST, FALSE)
            ->shouldBeCalled();
        $this->contentNegotiation
            ->handle($request_mock, HttpKernelInterface::SUB_REQUEST, FALSE);
    }
    
    /**
     * @covers ::registerFormat
     */
    public function testSetFormat() {
        $app = $this->createMock(MockedHttpKernelInterface::class);
        $app->expects($this->once())
            ->method('handle')
            ->willReturn($this->createMock(Response::class));
        $content_negotiation = new StubNegotiationMiddleware($app);
        $request = $this->prophesize(Request::class);
        // Default empty format list should not set any formats.
        $request->setFormat('david', 'geeky/david')
            ->shouldBeCalled();
        // Some calls we don't care about.
        $request->setRequestFormat()
            ->shouldNotBeCalled();
        $request_data = $this->prophesize(ParameterBag::class);
        $request_data->get('ajax_iframe_upload', FALSE)
            ->shouldBeCalled();
        $request_mock = $request->reveal();
        $request_mock->query = new ParameterBag([]);
        $request_mock->request = $request_data->reveal();
        // Trigger handle.
        $content_negotiation->registerFormat('david', 'geeky/david');
        $content_negotiation->handle($request_mock);
    }

}