CollectionTest.php

Go to the documentation of this file.

<?php
namespace Magento\Framework\Test\Unit\Data;

use Magento\Framework\Data\Collection;
use Magento\Framework\Data\Collection\EntityFactoryInterface;

class CollectionTest extends \PHPUnit\Framework\TestCase
{
    private $collection;

    public function setUp()
    {
        $factoryMock = $this->createMock(EntityFactoryInterface::class);
        $this->collection = new Collection($factoryMock);
    }

    public function testWalk()
    {
        $objOne = new \Magento\Framework\DataObject(['id' => 1, 'name' => 'one']);
        $objTwo = new \Magento\Framework\DataObject(['id' => 2, 'name' => 'two']);
        $objThree = new \Magento\Framework\DataObject(['id' => 3, 'name' => 'three']);

        $this->collection->addItem($objOne);
        $this->collection->addItem($objTwo);
        $this->collection->addItem($objThree);

        $this->assertEquals([1, 2, 3], $this->collection->getAllIds(), 'Items added incorrectly to the collection');
        $this->collection->walk([$this, 'modifyObjectNames'], ['test prefix']);

        $this->assertEquals([1, 2, 3], $this->collection->getAllIds(), 'Incorrect IDs after callback function');
        $expectedNames = [
            'test prefix one',
            'test prefix two',
            'test prefix three'
        ];

        $this->assertEquals(
            $expectedNames,
            $this->collection->getColumnValues('name'),
            'Incorrect Names after callback function'
        );
    }

    public function modifyObjectNames(\Magento\Framework\DataObject $object, $prefix)
    {
        $object->setData('name', $prefix . ' ' . $object->getData('name'));
    }
}