function KeyValueEntityStorageTest::setUpKeyValueEntityStorage

Prepares the key value entity storage.

@covers ::__construct

Parameters

string $uuid_key: (optional) The entity key used for the UUID. Defaults to 'uuid'.

File

core/tests/Drupal/Tests/Core/Entity/KeyValueStore/KeyValueEntityStorageTest.php, line 100

Class

KeyValueEntityStorageTest

@coversDefaultClass \Drupal\Core\Entity\KeyValueStore\KeyValueEntityStorage @group Entity

Namespace

Code

protected function setUpKeyValueEntityStorage($uuid_key = 'uuid') {
    $this->entityType
        ->expects($this->atLeastOnce())
        ->method('getKey')
        ->will($this->returnValueMap([
        [
            'id',
            'id',
        ],
        [
            'uuid',
            $uuid_key,
        ],
        [
            'langcode',
            'langcode',
        ],
    ]));
    $this->entityType
        ->expects($this->atLeastOnce())
        ->method('id')
        ->will($this->returnValue('test_entity_type'));
    $this->entityType
        ->expects($this->any())
        ->method('getListCacheTags')
        ->willReturn([
        'test_entity_type_list',
    ]);
    $this->entityTypeManager = $this->createMock(EntityTypeManagerInterface::class);
    $this->entityTypeManager
        ->expects($this->any())
        ->method('getDefinition')
        ->with('test_entity_type')
        ->will($this->returnValue($this->entityType));
    $this->entityFieldManager = $this->createMock(EntityFieldManagerInterface::class);
    $this->cacheTagsInvalidator = $this->createMock('Drupal\\Core\\Cache\\CacheTagsInvalidatorInterface');
    $this->keyValueStore = $this->createMock('Drupal\\Core\\KeyValueStore\\KeyValueStoreInterface');
    $this->moduleHandler = $this->createMock('Drupal\\Core\\Extension\\ModuleHandlerInterface');
    $this->uuidService = $this->createMock('Drupal\\Component\\Uuid\\UuidInterface');
    $this->languageManager = $this->createMock('Drupal\\Core\\Language\\LanguageManagerInterface');
    $language = new Language([
        'langcode' => 'en',
    ]);
    $this->languageManager
        ->expects($this->any())
        ->method('getDefaultLanguage')
        ->will($this->returnValue($language));
    $this->languageManager
        ->expects($this->any())
        ->method('getCurrentLanguage')
        ->will($this->returnValue($language));
    $this->entityStorage = new KeyValueEntityStorage($this->entityType, $this->keyValueStore, $this->uuidService, $this->languageManager, new MemoryCache());
    $this->entityStorage
        ->setModuleHandler($this->moduleHandler);
    $container = new ContainerBuilder();
    $container->set('entity_field.manager', $this->entityFieldManager);
    $container->set('entity_type.manager', $this->entityTypeManager);
    $container->set('language_manager', $this->languageManager);
    $container->set('cache_tags.invalidator', $this->cacheTagsInvalidator);
    \Drupal::setContainer($container);
}