function EntityFetchByFieldTest::testActionExecutionWithLimit

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[[api-linebreak]] @group RulesAction

Namespace

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->accessCheck(TRUE)
    ->willReturn($query->reveal())
    ->shouldBeCalledTimes(1);
  $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'));
}