class EntityStorageBaseTest

Same name and namespace in other branches
  1. 9 core/tests/Drupal/Tests/Core/Entity/EntityStorageBaseTest.php \Drupal\Tests\Core\Entity\EntityStorageBaseTest
  2. 11.x core/tests/Drupal/Tests/Core/Entity/EntityStorageBaseTest.php \Drupal\Tests\Core\Entity\EntityStorageBaseTest

@coversDefaultClass \Drupal\Core\Entity\EntityStorageBase
@group Entity

Hierarchy

Expanded class hierarchy of EntityStorageBaseTest

File

core/tests/Drupal/Tests/Core/Entity/EntityStorageBaseTest.php, line 15

Namespace

Drupal\Tests\Core\Entity
View source
class EntityStorageBaseTest extends UnitTestCase {
  
  /**
   * Generate a mocked entity object.
   *
   * @param string $id
   *   ID value for this entity.
   *
   * @return \Drupal\Core\Entity\EntityInterface|\PHPUnit\Framework\MockObject\MockObject
   *   The mocked entity.
   */
  public function generateEntityInterface(string $id) : EntityInterface&MockObject {
    $mock_entity = $this->createMock(EntityInterface::class);
    $mock_entity->expects($this->any())
      ->method('id')
      ->willReturn($id);
    return $mock_entity;
  }
  
  /**
   * Data provider for testLoad().
   */
  public static function providerLoad() : \Generator {
    // Data set for a matching value.
    (yield 'matching-value' => [
      '1',
      [
        '1' => '1',
      ],
      '1',
    ]);
    // Data set for no matching value.
    (yield 'no-matching-value' => [
      NULL,
      [],
      '0',
    ]);
  }
  
  /**
   * @covers ::load
   *
   * @dataProvider providerLoad
   */
  public function testLoad(string|null $expected, array $entity_fixture, string $query) : void {
    if (!is_null($expected)) {
      $expected = $this->generateEntityInterface($expected);
    }
    $entity_fixture = array_map([
      $this,
      'generateEntityInterface',
    ], $entity_fixture);
    $mock_base = $this->getMockBuilder('\\Drupal\\Core\\Entity\\EntityStorageBase')
      ->disableOriginalConstructor()
      ->onlyMethods([
      'loadMultiple',
    ])
      ->getMockForAbstractClass();
    // load() always calls loadMultiple().
    $mock_base->expects($this->once())
      ->method('loadMultiple')
      ->with([
      $query,
    ])
      ->willReturn($entity_fixture);
    $this->assertEquals($expected, $mock_base->load($query));
  }
  
  /**
   * Data provider for testLoadMultiple.
   */
  public static function providerLoadMultiple() : \Generator {
    // Data set for NULL ID parameter.
    (yield 'null-id-parameter' => [
      range(1, 10),
      range(1, 10),
      NULL,
    ]);
    // Data set for no results.
    (yield 'no-results' => [
      [],
      [],
      [
        '11',
      ],
    ]);
    // Data set for 0 results for multiple IDs.
    (yield 'no-results-multiple-ids' => [
      [],
      [],
      [
        '11',
        '12',
        '13',
      ],
    ]);
    // Data set for 1 result for 1 ID.
    (yield '1-result-for-1-id' => [
      [
        '1' => '1',
      ],
      [
        '1' => '1',
      ],
      [
        '1',
      ],
    ]);
    // Data set for results for all IDs.
    $ids = [
      '1',
      '2',
      '3',
    ];
    (yield 'results-for-all-ids' => [
      $ids,
      $ids,
      $ids,
    ]);
    // Data set for partial results for multiple IDs.
    (yield 'partial-results-for-multiple-ids' => [
      $ids,
      $ids,
      array_merge($ids, [
        '11',
        '12',
      ]),
    ]);
  }
  
  /**
   * Test loadMultiple().
   *
   * Does not cover statically-cached results.
   *
   * @covers ::loadMultiple
   *
   * @dataProvider providerLoadMultiple
   */
  public function testLoadMultiple(array $expected, array $load_multiple, array|null $query) : void {
    $expected = array_map([
      $this,
      'generateEntityInterface',
    ], $expected);
    $load_multiple = array_map([
      $this,
      'generateEntityInterface',
    ], $load_multiple);
    // Make our EntityStorageBase mock.
    $mock_base = $this->getMockBuilder('\\Drupal\\Core\\Entity\\EntityStorageBase')
      ->disableOriginalConstructor()
      ->onlyMethods([
      'doLoadMultiple',
      'postLoad',
    ])
      ->getMockForAbstractClass();
    // For all non-cached queries, we call doLoadMultiple().
    $mock_base->expects($this->once())
      ->method('doLoadMultiple')
      ->with($query)
      ->willReturn($load_multiple);
    // Make our EntityTypeInterface mock so that we can turn off static caching.
    $mock_entity_type = $this->createMock('\\Drupal\\Core\\Entity\\EntityTypeInterface');
    // Disallow caching.
    $mock_entity_type->expects($this->any())
      ->method('isStaticallyCacheable')
      ->willReturn(FALSE);
    // Add the EntityTypeInterface to the storage object.
    $ref_entity_type = new \ReflectionProperty($mock_base, 'entityType');
    $ref_entity_type->setValue($mock_base, $mock_entity_type);
    // Set up expectations for postLoad(), which we only call if there are
    // results from loadMultiple().
    $mock_base->expects($this->exactly(empty($load_multiple) ? 0 : 1))
      ->method('postLoad');
    $this->assertEquals($expected, $mock_base->loadMultiple($query));
  }

}

Members

Title Sort descending Deprecated Modifiers Object type Summary Overrides
EntityStorageBaseTest::generateEntityInterface public function Generate a mocked entity object.
EntityStorageBaseTest::providerLoad public static function Data provider for testLoad().
EntityStorageBaseTest::providerLoadMultiple public static function Data provider for testLoadMultiple.
EntityStorageBaseTest::testLoad public function @covers ::load[[api-linebreak]]
EntityStorageBaseTest::testLoadMultiple public function Test loadMultiple().
PhpUnitWarnings::$deprecationWarnings private static property Deprecation warnings from PHPUnit to raise with @trigger_error().
PhpUnitWarnings::addWarning public function Converts PHPUnit deprecation warnings to E_USER_DEPRECATED.
RandomGeneratorTrait::getRandomGenerator protected function Gets the random generator for the utility methods.
RandomGeneratorTrait::randomMachineName protected function Generates a unique random string containing letters and numbers.
RandomGeneratorTrait::randomObject public function Generates a random PHP object.
RandomGeneratorTrait::randomString public function Generates a pseudo-random string of ASCII characters of codes 32 to 126.
RandomGeneratorTrait::randomStringValidate Deprecated public function Callback for random string validation.
UnitTestCase::$root protected property The app root. 1
UnitTestCase::getClassResolverStub protected function Returns a stub class resolver.
UnitTestCase::getConfigFactoryStub public function Returns a stub config factory that behaves according to the passed array.
UnitTestCase::getConfigStorageStub public function Returns a stub config storage that returns the supplied configuration.
UnitTestCase::getContainerWithCacheTagsInvalidator protected function Sets up a container with a cache tags invalidator.
UnitTestCase::getStringTranslationStub public function Returns a stub translation manager that just returns the passed string.
UnitTestCase::setUp protected function 358
UnitTestCase::setUpBeforeClass public static function
UnitTestCase::__get public function

Buggy or inaccurate documentation? Please file an issue. Need support? Need help programming? Connect with the Drupal community.