function DedupeEntityTest::entityQueryExpects

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

Parameters

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

File

core/modules/migrate/tests/src/Unit/process/DedupeEntityTest.php, line 160

Class

DedupeEntityTest

@coversDefaultClass \Drupal\migrate\Plugin\migrate\process\DedupeEntity @group migrate @group legacy

Namespace

Code

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