function KeyValueEntityStorageTest::testLoadMultipleAll

@covers ::loadMultiple @covers ::postLoad @covers ::mapFromStorageRecords @covers ::doLoadMultiple

File

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

Class

KeyValueEntityStorageTest

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

Namespace

Code

public function testLoadMultipleAll() {
    $expected['foo'] = $this->getMockEntity('Drupal\\Core\\Entity\\EntityBase', [
        [
            'id' => 'foo',
        ],
    ]);
    $expected['bar'] = $this->getMockEntity('Drupal\\Core\\Entity\\EntityBase', [
        [
            'id' => 'bar',
        ],
    ]);
    $this->entityType
        ->expects($this->once())
        ->method('getClass')
        ->will($this->returnValue(get_class(reset($expected))));
    $this->setUpKeyValueEntityStorage();
    $this->keyValueStore
        ->expects($this->once())
        ->method('getAll')
        ->will($this->returnValue([
        [
            'id' => 'foo',
        ],
        [
            'id' => 'bar',
        ],
    ]));
    $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([]));
    $entities = $this->entityStorage
        ->loadMultiple();
    foreach ($entities as $id => $entity) {
        $this->assertInstanceOf('Drupal\\Core\\Entity\\EntityInterface', $entity);
        $this->assertSame($id, $entity->id());
        $this->assertSame($id, $expected[$id]->id());
    }
}