BatchIteratorTest.php

Go to the documentation of this file.

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

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

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

    private $selectMock;

    private $wrapperSelectMock;

    private $connectionMock;

    private $batchSize;

    private $correlationName;

    private $rangeField;

    private $rangeFieldAlias;

    protected function setUp()
    {
        $this->batchSize = 10;
        $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 BatchIterator(
            $this->selectMock,
            $this->batchSize,
            $this->correlationName,
            $this->rangeField,
            $this->rangeFieldAlias
        );
    }

    public function testCurrent()
    {
        $filed = $this->correlationName . '.' . $this->rangeField;

        $this->selectMock->expects($this->once())->method('where')->with($filed . ' > ?', 0);
        $this->selectMock->expects($this->once())->method('limit')->with($this->batchSize);
        $this->selectMock->expects($this->once())->method('order')->with($filed . ' ASC');
        $this->assertEquals($this->selectMock, $this->model->current());
        $this->assertEquals($this->selectMock, $this->model->current());
        $this->assertEquals(0, $this->model->key());
    }

    public function testIterations()
    {
        $startCallIndex = 3;
        $stepCall = 4;

        $this->connectionMock->expects($this->at($startCallIndex))
            ->method('fetchRow')
            ->willReturn(['max' => 10, 'cnt' => 10]);

        $this->connectionMock->expects($this->at($startCallIndex += $stepCall))
            ->method('fetchRow')
            ->willReturn(['max' => 20, 'cnt' => 10]);

        $this->connectionMock->expects($this->at($startCallIndex += $stepCall))
            ->method('fetchRow')
            ->willReturn(['max' => 25, 'cnt' => 5]);

        $this->connectionMock->expects($this->at($startCallIndex += $stepCall))
            ->method('fetchRow')
            ->willReturn(['max' => null, 'cnt' => 0]);

        $iteration = 0;
        $result = [];
        foreach ($this->model as $key => $select) {
            $result[] = $select;
            $this->assertEquals($iteration, $key);
            $iteration++;
        }
        $this->assertCount(3, $result);
    }

    public function testNext()
    {
        $filed = $this->correlationName . '.' . $this->rangeField;
        $this->selectMock->expects($this->at(0))->method('where')->with($filed . ' > ?', 0);
        $this->selectMock->expects($this->exactly(3))->method('limit')->with($this->batchSize);
        $this->selectMock->expects($this->exactly(3))->method('order')->with($filed . ' ASC');
        $this->selectMock->expects($this->at(3))->method('where')->with($filed . ' > ?', 25);

        $this->wrapperSelectMock->expects($this->exactly(3))->method('from')->with(
            $this->selectMock,
            [
                new \Zend_Db_Expr('MAX(' . $this->rangeFieldAlias . ') as max'),
                new \Zend_Db_Expr('COUNT(*) as cnt')
            ]
        );
        $this->connectionMock->expects($this->exactly(3))
            ->method('fetchRow')
            ->with($this->wrapperSelectMock)
            ->willReturn(['max' => 25, 'cnt' => 10]);

        $this->assertEquals($this->selectMock, $this->model->next());
        $this->assertEquals(1, $this->model->key());

        $this->assertEquals($this->selectMock, $this->model->next());
        $this->assertEquals($this->selectMock, $this->model->current());
        $this->assertEquals(2, $this->model->key());
    }
}