class RenderElementTest

class RenderElementTest extends UnitTestCase {
  
  /**
   * The request stack.
   *
   * @var \Symfony\Component\HttpFoundation\RequestStack
   */
  protected $requestStack;
  
  /**
   * The container.
   *
   * @var \Drupal\Core\DependencyInjection\ContainerBuilder
   */
  protected $container;
  
  /**
   * {@inheritdoc}
   */
  protected function setUp() : void {
    parent::setUp();
    $this->requestStack = new RequestStack();
    $this->container = new ContainerBuilder();
    $this->container
      ->set('request_stack', $this->requestStack);
    \Drupal::setContainer($this->container);
  }
  
  /**
   * @covers ::preRenderAjaxForm
   */
  public function testPreRenderAjaxForm() : void {
    $request = Request::create('/test');
    $request->query
      ->set('foo', 'bar');
    $this->requestStack
      ->push($request);
    $prophecy = $this->prophesize('Drupal\\Core\\Routing\\UrlGeneratorInterface');
    $url = '/test?foo=bar&ajax_form=1';
    $prophecy->generateFromRoute('<current>', [], [
      'query' => [
        'foo' => 'bar',
        FormBuilderInterface::AJAX_FORM_REQUEST => TRUE,
      ],
    ], TRUE)
      ->willReturn((new GeneratedUrl())->setCacheContexts([
      'route',
    ])
      ->setGeneratedUrl($url));
    $url_generator = $prophecy->reveal();
    $this->container
      ->set('url_generator', $url_generator);
    $element = [
      '#type' => 'select',
      '#id' => 'test',
      '#ajax' => [
        'wrapper' => 'foo',
        'callback' => 'test-callback',
      ],
    ];
    $element = RenderElementBase::preRenderAjaxForm($element);
    $this->assertTrue($element['#ajax_processed']);
    $this->assertEquals($url, $element['#attached']['drupalSettings']['ajax']['test']['url']);
  }
  
  /**
   * @covers ::preRenderAjaxForm
   */
  public function testPreRenderAjaxFormWithQueryOptions() : void {
    $request = Request::create('/test');
    $request->query
      ->set('foo', 'bar');
    $this->requestStack
      ->push($request);
    $prophecy = $this->prophesize('Drupal\\Core\\Routing\\UrlGeneratorInterface');
    $url = '/test?foo=bar&other=query&ajax_form=1';
    $prophecy->generateFromRoute('<current>', [], [
      'query' => [
        'foo' => 'bar',
        'other' => 'query',
        FormBuilderInterface::AJAX_FORM_REQUEST => TRUE,
      ],
    ], TRUE)
      ->willReturn((new GeneratedUrl())->setCacheContexts([
      'route',
    ])
      ->setGeneratedUrl($url));
    $url_generator = $prophecy->reveal();
    $this->container
      ->set('url_generator', $url_generator);
    $element = [
      '#type' => 'select',
      '#id' => 'test',
      '#ajax' => [
        'wrapper' => 'foo',
        'callback' => 'test-callback',
        'options' => [
          'query' => [
            'other' => 'query',
          ],
        ],
      ],
    ];
    $element = RenderElementBase::preRenderAjaxForm($element);
    $this->assertTrue($element['#ajax_processed']);
    $this->assertEquals($url, $element['#attached']['drupalSettings']['ajax']['test']['url']);
  }

}