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 @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'));
}