function MakeUniqueEntityFieldTest::entityQueryExpects

Helper function to add expectations to the mock entity query object.

Parameters

int $count: The number of unique values to be set up.

File

core/modules/migrate/tests/src/Unit/process/MakeUniqueEntityFieldTest.php, line 162

Class

MakeUniqueEntityFieldTest

@coversDefaultClass \Drupal\migrate\Plugin\migrate\process\MakeUniqueEntityField @group migrate

Namespace

Code

protected function entityQueryExpects($count) {
    $this->entityQuery
        ->expects($this->exactly($count + 1))
        ->method('condition')
        ->willReturn($this->entityQuery);
    $this->entityQuery
        ->expects($this->exactly($count + 1))
        ->method('count')
        ->willReturn($this->entityQuery);
    $this->entityQuery
        ->expects($this->exactly($count + 1))
        ->method('execute')
        ->willReturnCallback(function () use (&$count) {
        return $count--;
    });
}