public function EntityFetchByFieldTest::testActionExecutionWithLimit in Rules 8.3

Tests action execution when a value for limit is provided.

@covers ::execute

File

tests/src/Unit/Integration/RulesAction/EntityFetchByFieldTest.php, line 82

Class

EntityFetchByFieldTest

@coversDefaultClass \Drupal\rules\Plugin\RulesAction\EntityFetchByField @group RulesAction

Namespace

Drupal\Tests\rules\Unit\Integration\RulesAction

Code

public function testActionExecutionWithLimit() {
  $entity_type = 'entity_test';
  $field_name = 'test_field';
  $field_value = 'llama';
  $limit = 2;

  // Create an array of dummy entities.
  $entities = array_map(function () {
    return $this
      ->prophesize(EntityInterface::class)
      ->reveal();
  }, range(1, $limit));

  // Creates entity ids for new dummy array of entities.
  $entity_ids = range(1, $limit);

  // Create dummy query object.
  $query = $this
    ->prophesize(QueryInterface::class);
  $query
    ->condition($field_name, $field_value, '=')
    ->willReturn($query
    ->reveal())
    ->shouldBeCalledTimes(1);
  $query
    ->range(0, $limit)
    ->willReturn($query
    ->reveal())
    ->shouldBeCalledTimes(1);
  $query
    ->execute()
    ->willReturn($entity_ids)
    ->shouldBeCalledTimes(1);

  // Create dummy entity storage object.
  $entity_storage = $this
    ->prophesize(EntityStorageInterface::class);
  $entity_storage
    ->loadMultiple($entity_ids)
    ->willReturn($entities)
    ->shouldBeCalledTimes(1);
  $entity_storage
    ->getQuery()
    ->willReturn($query)
    ->shouldBeCalledTimes(1);
  $this->entityTypeManager
    ->getStorage($entity_type)
    ->willReturn($entity_storage
    ->reveal())
    ->shouldBeCalledTimes(1);

  // Set context values for EntityFetchByField action and execute.
  $this->action
    ->setContextValue('type', $entity_type)
    ->setContextValue('field_name', $field_name)
    ->setContextValue('field_value', $field_value)
    ->setContextValue('limit', $limit)
    ->execute();

  // Test that executing action with a value for limit returns the dummy
  // entities array.
  $this
    ->assertEquals($entities, $this->action
    ->getProvidedContext('entity_fetched')
    ->getContextValue('entity_fetched'));
}