mirror of
https://github.com/flarum/framework.git
synced 2024-11-30 05:13:37 +08:00
Make AbstractPolicy compatible with both object and class as $model (#1977)
This commit is contained in:
parent
738e95f024
commit
7143e13073
|
@ -35,7 +35,7 @@ abstract class AbstractPolicy
|
|||
*/
|
||||
public function getPermission(GetPermission $event)
|
||||
{
|
||||
if (! $event->model instanceof $this->model) {
|
||||
if (! $event->model instanceof $this->model && $event->model !== $this->model) {
|
||||
return;
|
||||
}
|
||||
|
||||
|
|
48
framework/core/tests/unit/User/AbstractPolicyTest.php
Normal file
48
framework/core/tests/unit/User/AbstractPolicyTest.php
Normal file
|
@ -0,0 +1,48 @@
|
|||
<?php
|
||||
|
||||
/*
|
||||
* This file is part of Flarum.
|
||||
*
|
||||
* For detailed copyright and license information, please view the
|
||||
* LICENSE file that was distributed with this source code.
|
||||
*/
|
||||
|
||||
namespace Flarum\Tests\unit\User;
|
||||
|
||||
use Flarum\Event\GetPermission;
|
||||
use Flarum\Tests\unit\TestCase;
|
||||
use Flarum\User\User;
|
||||
use Illuminate\Events\Dispatcher;
|
||||
use Mockery as m;
|
||||
|
||||
class AbstractPolicyTest extends TestCase
|
||||
{
|
||||
private $policy;
|
||||
private $dispatcher;
|
||||
|
||||
public function setUp()
|
||||
{
|
||||
$this->policy = m::mock(UserPolicy::class)->makePartial();
|
||||
$this->dispatcher = new Dispatcher();
|
||||
$this->dispatcher->subscribe($this->policy);
|
||||
User::setEventDispatcher($this->dispatcher);
|
||||
}
|
||||
|
||||
public function test_policy_can_be_called_with_object()
|
||||
{
|
||||
$this->policy->shouldReceive('edit')->andReturn(true);
|
||||
|
||||
$allowed = $this->dispatcher->until(new GetPermission(new User(), 'edit', new User()));
|
||||
|
||||
$this->assertTrue($allowed);
|
||||
}
|
||||
|
||||
public function test_policy_can_be_called_with_class()
|
||||
{
|
||||
$this->policy->shouldReceive('create')->andReturn(true);
|
||||
|
||||
$allowed = $this->dispatcher->until(new GetPermission(new User(), 'create', User::class));
|
||||
|
||||
$this->assertTrue($allowed);
|
||||
}
|
||||
}
|
28
framework/core/tests/unit/User/UserPolicy.php
Normal file
28
framework/core/tests/unit/User/UserPolicy.php
Normal file
|
@ -0,0 +1,28 @@
|
|||
<?php
|
||||
|
||||
/*
|
||||
* This file is part of Flarum.
|
||||
*
|
||||
* For detailed copyright and license information, please view the
|
||||
* LICENSE file that was distributed with this source code.
|
||||
*/
|
||||
|
||||
namespace Flarum\Tests\unit\User;
|
||||
|
||||
use Flarum\User\AbstractPolicy;
|
||||
use Flarum\User\User;
|
||||
|
||||
class UserPolicy extends AbstractPolicy
|
||||
{
|
||||
protected $model = User::class;
|
||||
|
||||
public function create(User $actor)
|
||||
{
|
||||
return true;
|
||||
}
|
||||
|
||||
public function edit(User $actor, User $user)
|
||||
{
|
||||
return true;
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue
Block a user