wiki-grav/plugins/email/vendor/symfony/messenger/Test/Middleware/MiddlewareTestCase.php
2023-05-12 15:06:01 +02:00

61 lines
1.7 KiB
PHP

<?php
/*
* This file is part of the Symfony package.
*
* (c) Fabien Potencier <fabien@symfony.com>
*
* For the full copyright and license information, please view the LICENSE
* file that was distributed with this source code.
*/
namespace Symfony\Component\Messenger\Test\Middleware;
use PHPUnit\Framework\TestCase;
use Symfony\Component\Messenger\Envelope;
use Symfony\Component\Messenger\Middleware\MiddlewareInterface;
use Symfony\Component\Messenger\Middleware\StackInterface;
use Symfony\Component\Messenger\Middleware\StackMiddleware;
/**
* @author Nicolas Grekas <p@tchwork.com>
*/
abstract class MiddlewareTestCase extends TestCase
{
protected function getStackMock(bool $nextIsCalled = true)
{
if (!$nextIsCalled) {
$stack = $this->createMock(StackInterface::class);
$stack
->expects($this->never())
->method('next')
;
return $stack;
}
$nextMiddleware = $this->createMock(MiddlewareInterface::class);
$nextMiddleware
->expects($this->once())
->method('handle')
->willReturnCallback(function (Envelope $envelope, StackInterface $stack): Envelope {
return $envelope;
})
;
return new StackMiddleware($nextMiddleware);
}
protected function getThrowingStackMock(\Throwable $throwable = null)
{
$nextMiddleware = $this->createMock(MiddlewareInterface::class);
$nextMiddleware
->expects($this->once())
->method('handle')
->willThrowException($throwable ?? new \RuntimeException('Thrown from next middleware.'))
;
return new StackMiddleware($nextMiddleware);
}
}