BatchRangeIteratorTest.php

Go to the documentation of this file.

<?php
namespace Magento\Framework\Test\Unit\DB\Query;

use Magento\Framework\DB\Adapter\AdapterInterface;
use Magento\Framework\DB\Query\BatchRangeIterator;
use Magento\Framework\DB\Select;

class BatchRangeIteratorTest extends \PHPUnit\Framework\TestCase
{
    private $model;

    private $selectMock;

    private $wrapperSelectMock;

    private $connectionMock;

    private $batchSize;

    private $correlationName;

    private $rangeField;

    private $rangeFieldAlias;

    private $currentBatch = 0;

    protected function setUp()
    {
        $this->batchSize = 10;
        $this->currentBatch = 0;
        $this->correlationName = 'correlationName';
        $this->rangeField = 'rangeField';
        $this->rangeFieldAlias = 'rangeFieldAlias';

        $this->selectMock = $this->createMock(Select::class);
        $this->wrapperSelectMock = $this->createMock(Select::class);
        $this->connectionMock = $this->createMock(AdapterInterface::class);
        $this->connectionMock->expects($this->any())->method('select')->willReturn($this->wrapperSelectMock);
        $this->selectMock->expects($this->once())->method('getConnection')->willReturn($this->connectionMock);
        $this->connectionMock->expects($this->any())->method('quoteIdentifier')->willReturnArgument(0);

        $this->model = new BatchRangeIterator(
            $this->selectMock,
            $this->batchSize,
            $this->correlationName,
            $this->rangeField,
            $this->rangeFieldAlias
        );
    }

    public function testCurrent()
    {
        $this->selectMock->expects($this->once())->method('limit')->with($this->batchSize, $this->currentBatch);
        $this->selectMock->expects($this->once())->method('order')->with('correlationName.rangeField' . ' ASC');
        $this->assertEquals($this->selectMock, $this->model->current());
        $this->assertEquals(0, $this->model->key());
    }

    public function testIterations()
    {
        $iterations = 0;

        $this->connectionMock->expects($this->once())
            ->method('fetchRow')
            ->willReturn(['cnt' => 105]);

        foreach ($this->model as $key) {
            $this->assertEquals($this->selectMock, $key);
            $iterations++;
        }

        $this->assertEquals(11, $iterations);
    }
}