class FormCacheTest

class FormCacheTest extends UnitTestCase {
    
    /**
     * The form cache object under test.
     *
     * @var \Drupal\Core\Form\FormCache
     */
    protected $formCache;
    
    /**
     * The expirable key value factory.
     *
     * @var \Drupal\Core\KeyValueStore\KeyValueExpirableFactoryInterface|\PHPUnit\Framework\MockObject\MockObject
     */
    protected $keyValueExpirableFactory;
    
    /**
     * The current user.
     *
     * @var \Drupal\Core\Session\AccountInterface|\PHPUnit\Framework\MockObject\MockObject
     */
    protected $account;
    
    /**
     * The CSRF token generator.
     *
     * @var \Drupal\Core\Access\CsrfTokenGenerator|\PHPUnit\Framework\MockObject\MockObject
     */
    protected $csrfToken;
    
    /**
     * The mocked module handler.
     *
     * @var \Prophecy\Prophecy\ObjectProphecy
     */
    protected $moduleHandler;
    
    /**
     * The expirable key value store used by form cache.
     *
     * @var \Drupal\Core\KeyValueStore\KeyValueStoreExpirableInterface|\PHPUnit\Framework\MockObject\MockObject
     */
    protected $formCacheStore;
    
    /**
     * The expirable key value store used by form state cache.
     *
     * @var \Drupal\Core\KeyValueStore\KeyValueStoreExpirableInterface|\PHPUnit\Framework\MockObject\MockObject
     */
    protected $formStateCacheStore;
    
    /**
     * The logger channel.
     *
     * @var \Psr\Log\LoggerInterface|\PHPUnit\Framework\MockObject\MockObject
     */
    protected $logger;
    
    /**
     * The request stack.
     *
     * @var \Symfony\Component\HttpFoundation\RequestStack|\PHPUnit\Framework\MockObject\MockObject
     */
    protected $requestStack;
    
    /**
     * A policy rule determining the cacheability of a request.
     *
     * @var \Drupal\Core\PageCache\RequestPolicyInterface|\PHPUnit\Framework\MockObject\MockObject
     */
    protected $requestPolicy;
    
    /**
     * {@inheritdoc}
     */
    protected function setUp() : void {
        parent::setUp();
        $this->moduleHandler = $this->prophesize('Drupal\\Core\\Extension\\ModuleHandlerInterface');
        $this->formCacheStore = $this->createMock('Drupal\\Core\\KeyValueStore\\KeyValueStoreExpirableInterface');
        $this->formStateCacheStore = $this->createMock('Drupal\\Core\\KeyValueStore\\KeyValueStoreExpirableInterface');
        $this->keyValueExpirableFactory = $this->createMock('Drupal\\Core\\KeyValueStore\\KeyValueExpirableFactoryInterface');
        $this->keyValueExpirableFactory
            ->expects($this->any())
            ->method('get')
            ->willReturnMap([
            [
                'form',
                $this->formCacheStore,
            ],
            [
                'form_state',
                $this->formStateCacheStore,
            ],
        ]);
        $this->csrfToken = $this->getMockBuilder('Drupal\\Core\\Access\\CsrfTokenGenerator')
            ->disableOriginalConstructor()
            ->getMock();
        $this->account = $this->createMock('Drupal\\Core\\Session\\AccountInterface');
        $this->logger = $this->createMock('Psr\\Log\\LoggerInterface');
        $this->requestStack = $this->createMock('\\Symfony\\Component\\HttpFoundation\\RequestStack');
        $this->requestPolicy = $this->createMock('\\Drupal\\Core\\PageCache\\RequestPolicyInterface');
        $this->formCache = new FormCache($this->root, $this->keyValueExpirableFactory, $this->moduleHandler
            ->reveal(), $this->account, $this->csrfToken, $this->logger, $this->requestStack, $this->requestPolicy);
    }
    
    /**
     * @covers ::getCache
     */
    public function testGetCacheValidToken() : void {
        $form_build_id = 'the_form_build_id';
        $form_state = new FormState();
        $cache_token = 'the_cache_token';
        $cached_form = [
            '#cache_token' => $cache_token,
        ];
        $this->formCacheStore
            ->expects($this->once())
            ->method('get')
            ->with($form_build_id)
            ->willReturn($cached_form);
        $this->csrfToken
            ->expects($this->once())
            ->method('validate')
            ->with($cache_token)
            ->willReturn(TRUE);
        $this->account
            ->expects($this->never())
            ->method('isAnonymous');
        $form = $this->formCache
            ->getCache($form_build_id, $form_state);
        $this->assertSame($cached_form, $form);
    }
    
    /**
     * @covers ::getCache
     */
    public function testGetCacheInvalidToken() : void {
        $form_build_id = 'the_form_build_id';
        $form_state = new FormState();
        $cache_token = 'the_cache_token';
        $cached_form = [
            '#cache_token' => $cache_token,
        ];
        $this->formCacheStore
            ->expects($this->once())
            ->method('get')
            ->with($form_build_id)
            ->willReturn($cached_form);
        $this->csrfToken
            ->expects($this->once())
            ->method('validate')
            ->with($cache_token)
            ->willReturn(FALSE);
        $this->account
            ->expects($this->never())
            ->method('isAnonymous');
        $form = $this->formCache
            ->getCache($form_build_id, $form_state);
        $this->assertNull($form);
    }
    
    /**
     * @covers ::getCache
     */
    public function testGetCacheAnonUser() : void {
        $form_build_id = 'the_form_build_id';
        $form_state = new FormState();
        $cached_form = [
            '#cache_token' => NULL,
        ];
        $this->formCacheStore
            ->expects($this->once())
            ->method('get')
            ->with($form_build_id)
            ->willReturn($cached_form);
        $this->account
            ->expects($this->once())
            ->method('isAnonymous')
            ->willReturn(TRUE);
        $this->csrfToken
            ->expects($this->never())
            ->method('validate');
        $form = $this->formCache
            ->getCache($form_build_id, $form_state);
        $this->assertSame($cached_form, $form);
    }
    
    /**
     * @covers ::getCache
     */
    public function testGetCacheAuthUser() : void {
        $form_build_id = 'the_form_build_id';
        $form_state = new FormState();
        $cached_form = [
            '#cache_token' => NULL,
        ];
        $this->formCacheStore
            ->expects($this->once())
            ->method('get')
            ->with($form_build_id)
            ->willReturn($cached_form);
        $this->account
            ->expects($this->once())
            ->method('isAnonymous')
            ->willReturn(FALSE);
        $form = $this->formCache
            ->getCache($form_build_id, $form_state);
        $this->assertNull($form);
    }
    
    /**
     * @covers ::getCache
     */
    public function testGetCacheNoForm() : void {
        $form_build_id = 'the_form_build_id';
        $form_state = new FormState();
        $cached_form = NULL;
        $this->formCacheStore
            ->expects($this->once())
            ->method('get')
            ->with($form_build_id)
            ->willReturn($cached_form);
        $this->account
            ->expects($this->never())
            ->method('isAnonymous');
        $form = $this->formCache
            ->getCache($form_build_id, $form_state);
        $this->assertNull($form);
    }
    
    /**
     * @covers ::loadCachedFormState
     */
    public function testLoadCachedFormState() : void {
        $form_build_id = 'the_form_build_id';
        $form_state = new FormState();
        $cached_form = [
            '#cache_token' => NULL,
        ];
        $this->formCacheStore
            ->expects($this->once())
            ->method('get')
            ->with($form_build_id)
            ->willReturn($cached_form);
        $this->account
            ->expects($this->once())
            ->method('isAnonymous')
            ->willReturn(TRUE);
        $cached_form_state = [
            'storage' => [
                'foo' => 'bar',
            ],
        ];
        $this->formStateCacheStore
            ->expects($this->once())
            ->method('get')
            ->with($form_build_id)
            ->willReturn($cached_form_state);
        $this->formCache
            ->getCache($form_build_id, $form_state);
        $this->assertSame($cached_form_state['storage'], $form_state->getStorage());
    }
    
    /**
     * @covers ::loadCachedFormState
     */
    public function testLoadCachedFormStateWithFiles() : void {
        $form_build_id = 'the_form_build_id';
        $form_state = new FormState();
        $cached_form = [
            '#cache_token' => NULL,
        ];
        $this->formCacheStore
            ->expects($this->once())
            ->method('get')
            ->with($form_build_id)
            ->willReturn($cached_form);
        $this->account
            ->expects($this->once())
            ->method('isAnonymous')
            ->willReturn(TRUE);
        $cached_form_state = [
            'build_info' => [
                'files' => [
                    [
                        'module' => 'a_module',
                        'type' => 'the_type',
                        'name' => 'some_name',
                    ],
                    [
                        'module' => 'another_module',
                    ],
                ],
            ],
        ];
        $this->moduleHandler
            ->loadInclude('a_module', 'the_type', 'some_name')
            ->shouldBeCalledOnce();
        $this->moduleHandler
            ->loadInclude('another_module', 'inc', 'another_module')
            ->shouldBeCalledOnce();
        $this->formStateCacheStore
            ->expects($this->once())
            ->method('get')
            ->with($form_build_id)
            ->willReturn($cached_form_state);
        $this->formCache
            ->getCache($form_build_id, $form_state);
    }
    
    /**
     * @covers ::setCache
     */
    public function testSetCacheWithForm() : void {
        $form_build_id = 'the_form_build_id';
        $form = [
            '#form_id' => 'the_form_id',
        ];
        $form_state = new FormState();
        $this->formCacheStore
            ->expects($this->once())
            ->method('setWithExpire')
            ->with($form_build_id, $form, $this->isType('int'));
        $form_state_data = $form_state->getCacheableArray();
        $this->formStateCacheStore
            ->expects($this->once())
            ->method('setWithExpire')
            ->with($form_build_id, $form_state_data, $this->isType('int'));
        $this->formCache
            ->setCache($form_build_id, $form, $form_state);
    }
    
    /**
     * @covers ::setCache
     */
    public function testSetCacheWithoutForm() : void {
        $form_build_id = 'the_form_build_id';
        $form = NULL;
        $form_state = new FormState();
        $this->formCacheStore
            ->expects($this->never())
            ->method('setWithExpire');
        $form_state_data = $form_state->getCacheableArray();
        $this->formStateCacheStore
            ->expects($this->once())
            ->method('setWithExpire')
            ->with($form_build_id, $form_state_data, $this->isType('int'));
        $this->formCache
            ->setCache($form_build_id, $form, $form_state);
    }
    
    /**
     * @covers ::setCache
     */
    public function testSetCacheAuthUser() : void {
        $form_build_id = 'the_form_build_id';
        $form = [];
        $form_state = new FormState();
        $cache_token = 'the_cache_token';
        $form_data = $form;
        $form_data['#cache_token'] = $cache_token;
        $this->formCacheStore
            ->expects($this->once())
            ->method('setWithExpire')
            ->with($form_build_id, $form_data, $this->isType('int'));
        $form_state_data = $form_state->getCacheableArray();
        $this->formStateCacheStore
            ->expects($this->once())
            ->method('setWithExpire')
            ->with($form_build_id, $form_state_data, $this->isType('int'));
        $this->csrfToken
            ->expects($this->once())
            ->method('get')
            ->willReturn($cache_token);
        $this->account
            ->expects($this->once())
            ->method('isAuthenticated')
            ->willReturn(TRUE);
        $this->formCache
            ->setCache($form_build_id, $form, $form_state);
    }
    
    /**
     * @covers ::setCache
     */
    public function testSetCacheBuildIdMismatch() : void {
        $form_build_id = 'the_form_build_id';
        $form = [
            '#form_id' => 'the_form_id',
            '#build_id' => 'stale_form_build_id',
        ];
        $form_state = new FormState();
        $this->formCacheStore
            ->expects($this->never())
            ->method('setWithExpire');
        $this->formStateCacheStore
            ->expects($this->never())
            ->method('setWithExpire');
        $this->logger
            ->expects($this->once())
            ->method('error')
            ->with('Form build-id mismatch detected while attempting to store a form in the cache.');
        $this->formCache
            ->setCache($form_build_id, $form, $form_state);
    }
    
    /**
     * @covers ::deleteCache
     */
    public function testDeleteCache() : void {
        $form_build_id = 'the_form_build_id';
        $this->formCacheStore
            ->expects($this->once())
            ->method('delete')
            ->with($form_build_id);
        $this->formStateCacheStore
            ->expects($this->once())
            ->method('delete')
            ->with($form_build_id);
        $this->formCache
            ->deleteCache($form_build_id);
    }

}