function KeyValueEntityStorageTest::testLoad

@covers ::load @covers ::postLoad

File

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

Class

KeyValueEntityStorageTest

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

Namespace

Code

public function testLoad() {
    $entity = $this->getMockEntity();
    $this->entityType
        ->expects($this->once())
        ->method('getClass')
        ->will($this->returnValue(get_class($entity)));
    $this->setUpKeyValueEntityStorage();
    $this->keyValueStore
        ->expects($this->once())
        ->method('getMultiple')
        ->with([
        'foo',
    ])
        ->will($this->returnValue([
        [
            'id' => 'foo',
        ],
    ]));
    $this->moduleHandler
        ->expects($this->at(0))
        ->method('getImplementations')
        ->with('entity_load')
        ->will($this->returnValue([]));
    $this->moduleHandler
        ->expects($this->at(1))
        ->method('getImplementations')
        ->with('test_entity_type_load')
        ->will($this->returnValue([]));
    $entity = $this->entityStorage
        ->load('foo');
    $this->assertInstanceOf('Drupal\\Core\\Entity\\EntityInterface', $entity);
    $this->assertSame('foo', $entity->id());
}