FriendsOfSymfony / FOSElasticaBundle

        $previousSlice = [];

        foreach ($objectsByIteration as $i => $objects) {

            $offset = $objects[0] - 1;

            $this->sliceFetcher->expects($this->at($i))

                ->method('fetch')

                ->with($queryBuilder, $batchSize, $offset, $previousSlice, ['id'])

                ->will($this->returnValue($objects));

            $this->objectManager->expects($this->at($i))

                ->method('clear');

            $previousSlice = $objects;

        }

        $this->objectPersister->expects($this->exactly(count($objectsByIteration)))

            ->method('insertMany');

        $providerInvocationOffset = 2;

        foreach ($objectsByIteration as $i => $objects) {

            $offset = $objects[0] - 1;

            $provider->expects($this->at($providerInvocationOffset + $i))

                ->method('fetchSlice')

                ->with($queryBuilder, $batchSize, $offset)

                ->will($this->returnValue($objects));

            $this->objectManager->expects($this->at($i))

                ->method('clear');

        }

        $this->objectPersister->expects($this->exactly(count($objectsByIteration)))

            ->method('insertMany');