public function KeyValueEntityStorageTest::testLoadMultipleAll in Drupal 8

@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

Drupal\Tests\Core\Entity\KeyValueStore

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());
  }
}