newsupernova/vendor/magento/data-migration-tool/tests/unit/testsuite/Migration/Handler/SequenceTest.php

<?php
/**
 * Copyright © Magento, Inc. All rights reserved.
 * See COPYING.txt for license details.
 */

namespace Migration\Handler;

class SequenceTest extends \PHPUnit\Framework\TestCase
{
    /**
     * @param string $table
     * @param int $id
     *
     * @return void
     * @dataProvider dataProviderTables
     */
    public function testHandle($table, $id)
    {
        $fieldName = 'fieldname';
        $createdVersionField = 'fieldname_version';
        $version = 1;

        /** @var \Migration\ResourceModel\Structure|\PHPUnit_Framework_MockObject_MockObject $record */
        $structure = $this->getMockBuilder(\Migration\ResourceModel\Structure::class)
            ->setMethods(['getFields'])
            ->disableOriginalConstructor()
            ->getMock();
        $structure->expects($this->any())->method('getFields')->willReturn([$fieldName]);
        /** @var \Migration\ResourceModel\Record|\PHPUnit_Framework_MockObject_MockObject $record */
        $record = $this->getMockBuilder(\Migration\ResourceModel\Record::class)
            ->setMethods(['getValue', 'setValue', 'getFields', 'getStructure'])
            ->disableOriginalConstructor()
            ->getMock();
        $record->expects($this->any())->method('getStructure')->willReturn($structure);
        $record->expects($this->any())->method('getFields')->willReturn([$fieldName]);
        $record->expects($this->any())->method('getValue')->with($fieldName)->willReturn($id);
        $record->expects($this->any())->method('setValue')->with($fieldName, $id);

        /** @var \Migration\ResourceModel\Record|\PHPUnit_Framework_MockObject_MockObject $record2 */
        $record2 = $this->getMockBuilder(\Migration\ResourceModel\Record::class)
            ->setMethods(['getValue', 'setValue', 'getFields', 'getStructure'])
            ->disableOriginalConstructor()
            ->getMock();

        $record2->expects($this->any())->method('setValue')->willReturn($createdVersionField, $version);
        $record2->expects($this->any())->method('getFields')->willReturn([$fieldName]);
        $record2->expects($this->any())->method('getStructure')->willReturn($structure);

        /** @var \Migration\ResourceModel\Destination|\PHPUnit_Framework_MockObject_MockObject $destination */
        $destination = $this->getMockBuilder(\Migration\ResourceModel\Destination::class)
            ->setMethods(['clearDocument', 'saveRecords'])
            ->disableOriginalConstructor()
            ->getMock();

        $destination->expects($this->any())->method('clearDocument')->with($table);
        $destination->expects($this->any())->method('saveRecords')->with($table, [['sequence_value' => $id]])
            ->willReturnSelf();

        $handler = new Sequence($table, $destination);
        $handler->setField($fieldName);
        $this->assertNull($handler->handle($record, $record2));
    }

    /**
     * @return array
     */
    public function dataProviderTables()
    {
        return [
            ['table' => 'tablename', 'id' => 1],
            ['table' => '',          'id' => 1],
            ['table' => 'tablename', 'id' => 0],
            ['table' => '',          'id' => 0],
        ];
    }
}