cakephp/tests/TestCase/Http/MiddlewareQueueTest.php

<?php
declare(strict_types=1);
/**
 * CakePHP(tm) : Rapid Development Framework (https://cakephp.org)
 * Copyright (c) Cake Software Foundation, Inc. (https://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. (https://cakefoundation.org)
 * @link          https://cakephp.org CakePHP(tm) Project
 * @since         3.3.0
 * @license       https://opensource.org/licenses/mit-license.php MIT License
 */
namespace Cake\Test\TestCase\Http;

use Cake\Core\Configure;
use Cake\Http\Middleware\CallableMiddleware;
use Cake\Http\MiddlewareQueue;
use Cake\TestSuite\TestCase;
use TestApp\Middleware\DumbMiddleware;
use TestApp\Middleware\SampleMiddleware;

/**
 * Test case for the MiddlewareQueue
 */
class MiddlewareQueueTest extends TestCase
{
    /**
     * setUp
     *
     * @return void
     */
    public function setUp()
    {
        parent::setUp();

        $this->appNamespace = Configure::read('App.namespace');
        static::setAppNamespace();
    }

    /**
     * tearDown
     *
     * @return void
     */
    public function tearDown()
    {
        parent::tearDown();
        static::setAppNamespace($this->appNamespace);
    }

    public function testConstructorAddingMiddleware()
    {
        $cb = new CallableMiddleware(function () {
        });
        $queue = new MiddlewareQueue([$cb]);
        $this->assertCount(1, $queue);
        $this->assertSame($cb, $queue->get(0));
    }

    /**
     * Test get()
     *
     * @return void
     */
    public function testGet()
    {
        $queue = new MiddlewareQueue();
        $cb = new CallableMiddleware(function () {
        });
        $queue->add($cb);
        $this->assertSame($cb, $queue->get(0));
        $this->assertNull($queue->get(1));
    }

    /**
     * Test the return value of add()
     *
     * @return void
     */
    public function testAddReturn()
    {
        $queue = new MiddlewareQueue();
        $cb = new CallableMiddleware(function () {
        });
        $this->assertSame($queue, $queue->add($cb));
    }

    /**
     * Test the add orders correctly
     *
     * @return void
     */
    public function testAddOrdering()
    {
        $one = new CallableMiddleware(function () {
        });
        $two = new CallableMiddleware(function () {
        });

        $queue = new MiddlewareQueue();
        $this->assertCount(0, $queue);

        $queue->add($one);
        $this->assertCount(1, $queue);

        $queue->add($two);
        $this->assertCount(2, $queue);

        $this->assertSame($one, $queue->get(0));
        $this->assertSame($two, $queue->get(1));
    }

    /**
     * Test the prepend can be chained
     *
     * @return void
     */
    public function testPrependReturn()
    {
        $cb = new CallableMiddleware(function () {
        });
        $queue = new MiddlewareQueue();
        $this->assertSame($queue, $queue->prepend($cb));
    }

    /**
     * Test the prepend orders correctly.
     *
     * @return void
     */
    public function testPrependOrdering()
    {
        $one = new CallableMiddleware(function () {
        });
        $two = new CallableMiddleware(function () {
        });

        $queue = new MiddlewareQueue();
        $this->assertCount(0, $queue);

        $queue->add($one);
        $this->assertCount(1, $queue);

        $queue->prepend($two);
        $this->assertCount(2, $queue);

        $this->assertSame($two, $queue->get(0));
        $this->assertSame($one, $queue->get(1));
    }

    /**
     * Test updating queue using class name
     *
     * @return void
     */
    public function testAddingPrependingUsingString()
    {
        $queue = new MiddlewareQueue();
        $queue->add('Sample');
        $queue->prepend('TestApp\Middleware\SampleMiddleware');

        $this->assertInstanceOf('TestApp\Middleware\SampleMiddleware', $queue->get(0)->getCallable());
        $this->assertInstanceOf('TestApp\Middleware\SampleMiddleware', $queue->get(1)->getCallable());
    }

    /**
     * Test updating queue using array
     *
     * @return void
     */
    public function testAddingPrependingUsingArray()
    {
        $one = new CallableMiddleware(function () {
        });

        $queue = new MiddlewareQueue();
        $queue->add([$one]);
        $queue->prepend(['TestApp\Middleware\SampleMiddleware']);

        $this->assertInstanceOf('TestApp\Middleware\SampleMiddleware', $queue->get(0)->getCallable());
        $this->assertSame($one, $queue->get(1));
    }

    /**
     * Test insertAt ordering
     *
     * @return void
     */
    public function testInsertAt()
    {
        $one = new CallableMiddleware(function () {
        });
        $two = new CallableMiddleware(function () {
        });
        $three = new CallableMiddleware(function () {
        });
        $four = new SampleMiddleware();

        $queue = new MiddlewareQueue();
        $queue->add($one)->add($two)->insertAt(0, $three)->insertAt(2, $four);
        $this->assertSame($three, $queue->get(0));
        $this->assertSame($one, $queue->get(1));
        $this->assertSame($four, $queue->get(2)->getCallable());
        $this->assertSame($two, $queue->get(3));

        $queue = new MiddlewareQueue();
        $queue->add($one)->add($two)->insertAt(1, $three);
        $this->assertSame($one, $queue->get(0));
        $this->assertSame($three, $queue->get(1));
        $this->assertSame($two, $queue->get(2));
    }

    /**
     * Test insertAt out of the existing range
     *
     * @return void
     */
    public function testInsertAtOutOfBounds()
    {
        $one = new CallableMiddleware(function () {
        });
        $two = new CallableMiddleware(function () {
        });

        $queue = new MiddlewareQueue();
        $queue->add($one)->insertAt(99, $two);

        $this->assertCount(2, $queue);
        $this->assertSame($one, $queue->get(0));
        $this->assertSame($two, $queue->get(1));
    }

    /**
     * Test insertAt with a negative index
     *
     * @return void
     */
    public function testInsertAtNegative()
    {
        $one = new CallableMiddleware(function () {
        });
        $two = new CallableMiddleware(function () {
        });
        $three = new SampleMiddleware();

        $queue = new MiddlewareQueue();
        $queue->add($one)->insertAt(-1, $two)->insertAt(-1, $three);

        $this->assertCount(3, $queue);
        $this->assertSame($two, $queue->get(0));
        $this->assertSame($three, $queue->get(1)->getCallable());
        $this->assertSame($one, $queue->get(2));
    }

    /**
     * Test insertBefore
     *
     * @return void
     */
    public function testInsertBefore()
    {
        $one = new CallableMiddleware(function () {
        });
        $two = new SampleMiddleware();
        $three = new CallableMiddleware(function () {
        });
        $four = new DumbMiddleware();

        $queue = new MiddlewareQueue();
        $queue->add($one)->add($two)->insertBefore(SampleMiddleware::class, $three)->insertBefore(SampleMiddleware::class, $four);

        $this->assertCount(4, $queue);
        $this->assertSame($one, $queue->get(0));
        $this->assertSame($three, $queue->get(1));
        $this->assertSame($four, $queue->get(2)->getCallable());
        $this->assertSame($two, $queue->get(3)->getCallable());

        $two = SampleMiddleware::class;
        $queue = new MiddlewareQueue();
        $queue
            ->add($one)
            ->add($two)
            ->insertBefore(SampleMiddleware::class, $three);

        $this->assertCount(3, $queue);
        $this->assertSame($one, $queue->get(0));
        $this->assertSame($three, $queue->get(1));
        $this->assertInstanceOf(SampleMiddleware::class, $queue->get(2)->getCallable());
    }

    /**
     * Test insertBefore an invalid classname
     *
     * @return void
     */
    public function testInsertBeforeInvalid()
    {
        $this->expectException(\LogicException::class);
        $this->expectExceptionMessage('No middleware matching \'InvalidClassName\' could be found.');
        $one = function () {
        };
        $two = new SampleMiddleware();
        $three = function () {
        };
        $queue = new MiddlewareQueue();
        $queue->add($one)->add($two)->insertBefore('InvalidClassName', $three);
    }

    /**
     * Test insertAfter
     *
     * @return void
     */
    public function testInsertAfter()
    {
        $one = new SampleMiddleware();
        $two = new CallableMiddleware(function () {
        });
        $three = new CallableMiddleware(function () {
        });
        $four = new DumbMiddleware();
        $queue = new MiddlewareQueue();
        $queue
            ->add($one)
            ->add($two)
            ->insertAfter(SampleMiddleware::class, $three)
            ->insertAfter(SampleMiddleware::class, $four);

        $this->assertCount(4, $queue);
        $this->assertSame($one, $queue->get(0)->getCallable());
        $this->assertSame($four, $queue->get(1)->getCallable());
        $this->assertSame($three, $queue->get(2));
        $this->assertSame($two, $queue->get(3));

        $one = 'Sample';
        $queue = new MiddlewareQueue();
        $queue
            ->add($one)
            ->add($two)
            ->insertAfter('Sample', $three);

        $this->assertCount(3, $queue);
        $this->assertInstanceOf(SampleMiddleware::class, $queue->get(0)->getCallable());
        $this->assertSame($three, $queue->get(1));
        $this->assertSame($two, $queue->get(2));
    }

    /**
     * Test insertAfter an invalid classname
     *
     * @return void
     */
    public function testInsertAfterInvalid()
    {
        $one = new SampleMiddleware();
        $two = new CallableMiddleware(function () {
        });
        $three = new CallableMiddleware(function () {
        });
        $queue = new MiddlewareQueue();
        $queue->add($one)->add($two)->insertAfter('InvalidClass', $three);

        $this->assertCount(3, $queue);
        $this->assertSame($one, $queue->get(0)->getCallable());
        $this->assertSame($two, $queue->get(1));
        $this->assertSame($three, $queue->get(2));
    }
}