59 lines
1.5 KiB
PHP
59 lines
1.5 KiB
PHP
|
<?php
|
||
|
|
|
||
|
|
namespace Tests\Unit\Http\Middleware;
|
||
|
|
|
||
|
|
use Mockery as m;
|
||
|
|
use Tests\TestCase;
|
||
|
|
use Illuminate\Http\Request;
|
||
|
|
use Pterodactyl\Models\User;
|
||
|
|
use Tests\Traits\Http\MocksMiddlewareClosure;
|
||
|
|
use Symfony\Component\HttpFoundation\ParameterBag;
|
||
|
|
use Tests\Assertions\MiddlewareAttributeAssertionsTrait;
|
||
|
|
|
||
|
|
abstract class MiddlewareTestCase extends TestCase
|
||
|
|
{
|
||
|
|
use MiddlewareAttributeAssertionsTrait, MocksMiddlewareClosure;
|
||
|
|
|
||
|
|
/**
|
||
|
|
* @var \Illuminate\Http\Request|\Mockery\Mock
|
||
|
|
*/
|
||
|
|
protected $request;
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Setup tests with a mocked request object and normal attributes.
|
||
|
|
*/
|
||
|
|
public function setUp()
|
||
|
|
{
|
||
|
|
parent::setUp();
|
||
|
|
|
||
|
|
$this->request = m::mock(Request::class);
|
||
|
|
$this->request->attributes = new ParameterBag();
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Set a request attribute on the mock object.
|
||
|
|
*
|
||
|
|
* @param string $attribute
|
||
|
|
* @param mixed $value
|
||
|
|
*/
|
||
|
|
protected function setRequestAttribute(string $attribute, $value)
|
||
|
|
{
|
||
|
|
$this->request->attributes->set($attribute, $value);
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Sets the mocked request user. If a user model is not provided, a factory model
|
||
|
|
* will be created and returned.
|
||
|
|
*
|
||
|
|
* @param \Pterodactyl\Models\User|null $user
|
||
|
|
* @return \Pterodactyl\Models\User
|
||
|
|
*/
|
||
|
|
protected function setRequestUser(User $user = null): User
|
||
|
|
{
|
||
|
|
$user = $user instanceof User ? $user : factory(User::class)->make();
|
||
|
|
$this->request->shouldReceive('user')->withNoArgs()->andReturn($user);
|
||
|
|
|
||
|
|
return $user;
|
||
|
|
}
|
||
|
|
}
|