cakephp/tests/TestCase/Collection/Iterator/TreeIteratorTest.php

<?php
/**
 * CakePHP(tm) : Rapid Development Framework (http://cakephp.org)
 * Copyright (c) Cake Software Foundation, Inc. (http://cakefoundation.org)
 *
 * Licensed under The MIT License
 * For full copyright and license information, please see the LICENSE.txt
 * Redistributions of files must retain the above copyright notice.
 *
 * @copyright     Copyright (c) Cake Software Foundation, Inc. (http://cakefoundation.org)
 * @link          http://cakephp.org CakePHP(tm) Project
 * @since         3.0.0
 * @license       http://www.opensource.org/licenses/mit-license.php MIT License
 */
namespace Cake\Test\TestCase\Collection\Iterator;

use Cake\Collection\Iterator\NestIterator;
use Cake\Collection\Iterator\TreeIterator;
use Cake\TestSuite\TestCase;

/**
 * TreeIterator Test
 *
 */
class TreeIteratorTest extends TestCase
{

    /**
     * Tests the printer function with defaults
     *
     * @return void
     */
    public function testPrinter()
    {
        $items = [
            [
                'id' => 1,
                'name' => 'a',
                'stuff' => [
                    ['id' => 2, 'name' => 'b', 'stuff' => [['id' => 3, 'name' => 'c']]]
                ]
            ],
            ['id' => 4, 'name' => 'd', 'stuff' => [['id' => 5, 'name' => 'e']]]
        ];
        $items = new NestIterator($items, 'stuff');
        $result = (new TreeIterator($items))->printer('name')->toArray();
        $expected = [
            'a',
            '__b',
            '____c',
            'd',
            '__e'
        ];
        $this->assertEquals($expected, $result);
    }

    /**
     * Tests the printer function with a custom key extractor and spacer
     *
     * @return void
     */
    public function testPrinterCustomKeyAndSpacer()
    {
        $items = [
            [
                'id' => 1,
                'name' => 'a',
                'stuff' => [
                    ['id' => 2, 'name' => 'b', 'stuff' => [['id' => 3, 'name' => 'c']]]
                ]
            ],
            ['id' => 4, 'name' => 'd', 'stuff' => [['id' => 5, 'name' => 'e']]]
        ];
        $items = new NestIterator($items, 'stuff');
        $result = (new TreeIterator($items))->printer('id', 'name', '@@')->toArray();
        $expected = [
            'a' => '1',
            'b' => '@@2',
            'c' => '@@@@3',
            'd' => '4',
            'e' => '@@5'
        ];
        $this->assertEquals($expected, $result);
    }

    /**
     * Tests the printer function with a closure extractor
     *
     * @return void
     */
    public function testPrinterWithClosure()
    {
        $items = [
            [
                'id' => 1,
                'name' => 'a',
                'stuff' => [
                    ['id' => 2, 'name' => 'b', 'stuff' => [['id' => 3, 'name' => 'c']]]
                ]
            ],
            ['id' => 4, 'name' => 'd', 'stuff' => [['id' => 5, 'name' => 'e']]]
        ];
        $items = new NestIterator($items, 'stuff');
        $result = (new TreeIterator($items))
            ->printer(function ($element, $key, $iterator) {
                return ($iterator->getDepth() + 1 ) . '.' . $key . ' ' . $element['name'];
            }, null, null)
            ->toArray();
        $expected = [
            '1.0 a',
            '2.0 b',
            '3.0 c',
            '1.1 d',
            '2.0 e'
        ];
        $this->assertEquals($expected, $result);
    }
}