mirror of
https://github.com/librenms/librenms.git
synced 2024-10-07 16:52:45 +00:00
239 lines
10 KiB
PHP
239 lines
10 KiB
PHP
<?php
|
|
/**
|
|
* PermissionsTest.php
|
|
*
|
|
* -Description-
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
*
|
|
* @link https://www.librenms.org
|
|
*
|
|
* @copyright 2019 Tony Murray
|
|
* @author Tony Murray <murraytony@gmail.com>
|
|
*/
|
|
|
|
namespace LibreNMS\Tests\Unit;
|
|
|
|
use App\Models\Bill;
|
|
use App\Models\Device;
|
|
use App\Models\Port;
|
|
use App\Models\User;
|
|
use LibreNMS\Tests\TestCase;
|
|
use Mockery\Mock;
|
|
|
|
class PermissionsTest extends TestCase
|
|
{
|
|
private function devicePermissionData($user)
|
|
{
|
|
$user_id = $user instanceof User ? $user->user_id : (is_numeric($user) ? (int) $user : \Auth::id());
|
|
|
|
$data = null;
|
|
|
|
switch ($user_id) {
|
|
case 43:
|
|
$data = [
|
|
(object) ['user_id' => 43, 'device_id' => 54],
|
|
(object) ['user_id' => 43, 'device_id' => 32],
|
|
];
|
|
break;
|
|
case 14:
|
|
$data = [
|
|
(object) ['user_id' => 14, 'device_id' => 54],
|
|
];
|
|
break;
|
|
}
|
|
|
|
return collect($data);
|
|
}
|
|
|
|
public function testUserCanAccessDevice()
|
|
{
|
|
$perms = \Mockery::mock(\LibreNMS\Cache\PermissionsCache::class)->makePartial();
|
|
$perms->shouldReceive('getDevicePermissions')->andReturnUsing(function ($user) {
|
|
return self::devicePermissionData($user);
|
|
});
|
|
|
|
$device = Device::factory()->make(['device_id' => 54]); /** @var Device $device */
|
|
$user = User::factory()->make(['user_id' => 43]); /** @var User $user */
|
|
$this->assertTrue($perms->canAccessDevice($device, 43));
|
|
$this->assertTrue($perms->canAccessDevice($device, $user));
|
|
$this->assertTrue($perms->canAccessDevice(54, $user));
|
|
$this->assertTrue($perms->canAccessDevice(54, 43));
|
|
$this->assertTrue($perms->canAccessDevice(54, 43));
|
|
$this->assertFalse($perms->canAccessDevice(54, 23));
|
|
$this->assertFalse($perms->canAccessDevice(23, 43));
|
|
$this->assertFalse($perms->canAccessDevice(54));
|
|
|
|
\Auth::shouldReceive('id')->once()->andReturn(43);
|
|
$this->assertTrue($perms->canAccessDevice(54));
|
|
\Auth::shouldReceive('id')->once()->andReturn(23);
|
|
$this->assertFalse($perms->canAccessDevice(54));
|
|
}
|
|
|
|
public function testDevicesForUser()
|
|
{
|
|
$perms = \Mockery::mock(\LibreNMS\Cache\PermissionsCache::class)->makePartial();
|
|
$perms->shouldReceive('getDevicePermissions')->andReturnUsing(function ($user) {
|
|
return self::devicePermissionData($user);
|
|
});
|
|
|
|
$this->assertEquals(collect([54, 32]), $perms->devicesForUser(43));
|
|
$user = User::factory()->make(['user_id' => 43]); /** @var User $user */
|
|
$this->assertEquals(collect([54, 32]), $perms->devicesForUser($user));
|
|
$this->assertEmpty($perms->devicesForUser(9));
|
|
$this->assertEquals(collect(), $perms->devicesForUser());
|
|
\Auth::shouldReceive('id')->once()->andReturn(14);
|
|
$this->assertEquals(collect([54]), $perms->devicesForUser());
|
|
}
|
|
|
|
/*
|
|
public function testUsersForDevice()
|
|
{
|
|
$perms = \Mockery::mock(\LibreNMS\Permissions::class)->makePartial();
|
|
$perms->shouldReceive('getDevicePermissions')->andReturn(collect([
|
|
(object)['user_id' => 3, 'device_id' => 7],
|
|
(object)['user_id' => 3, 'device_id' => 2],
|
|
(object)['user_id' => 4, 'device_id' => 5],
|
|
(object)['user_id' => 6, 'device_id' => 5],
|
|
]));
|
|
|
|
$this->assertEquals(collect([4, 6]), $perms->usersForDevice(5));
|
|
$this->assertEquals(collect([3]), $perms->usersForDevice(Device::factory()->make(['device_id' => 7])));
|
|
$this->assertEquals(collect(), $perms->usersForDevice(6));
|
|
$this->assertEmpty($perms->usersForDevice(9));
|
|
}
|
|
*/
|
|
public function testUserCanAccessPort()
|
|
{
|
|
$perms = \Mockery::mock(\LibreNMS\Cache\PermissionsCache::class)->makePartial();
|
|
$perms->shouldReceive('getPortPermissions')->andReturn(collect([
|
|
(object) ['user_id' => 43, 'port_id' => 54],
|
|
(object) ['user_id' => 43, 'port_id' => 32],
|
|
(object) ['user_id' => 14, 'port_id' => 54],
|
|
]));
|
|
|
|
$port = Port::factory()->make(['port_id' => 54]); /** @var Port $port */
|
|
$user = User::factory()->make(['user_id' => 43]); /** @var User $user */
|
|
$this->assertTrue($perms->canAccessPort($port, 43));
|
|
$this->assertTrue($perms->canAccessPort($port, $user));
|
|
$this->assertTrue($perms->canAccessPort(54, $user));
|
|
$this->assertTrue($perms->canAccessPort(54, 43));
|
|
$this->assertTrue($perms->canAccessPort(54, 43));
|
|
$this->assertFalse($perms->canAccessPort(54, 23));
|
|
$this->assertFalse($perms->canAccessPort(23, 43));
|
|
$this->assertFalse($perms->canAccessPort(54));
|
|
|
|
\Auth::shouldReceive('id')->once()->andReturn(43);
|
|
$this->assertTrue($perms->canAccessPort(54));
|
|
\Auth::shouldReceive('id')->once()->andReturn(23);
|
|
$this->assertFalse($perms->canAccessPort(54));
|
|
}
|
|
|
|
public function testPortsForUser()
|
|
{
|
|
$perms = \Mockery::mock(\LibreNMS\Cache\PermissionsCache::class)->makePartial();
|
|
$perms->shouldReceive('getPortPermissions')->andReturn(collect([
|
|
(object) ['user_id' => 3, 'port_id' => 7],
|
|
(object) ['user_id' => 3, 'port_id' => 2],
|
|
(object) ['user_id' => 4, 'port_id' => 5],
|
|
]));
|
|
|
|
$this->assertEquals(collect([7, 2]), $perms->portsForUser(3));
|
|
$user = User::factory()->make(['user_id' => 3]); /** @var User $user */
|
|
$this->assertEquals(collect([7, 2]), $perms->portsForUser($user));
|
|
$this->assertEmpty($perms->portsForUser(9));
|
|
$this->assertEquals(collect(), $perms->portsForUser());
|
|
\Auth::shouldReceive('id')->once()->andReturn(4);
|
|
$this->assertEquals(collect([5]), $perms->portsForUser());
|
|
}
|
|
|
|
public function testUsersForPort()
|
|
{
|
|
$perms = \Mockery::mock(\LibreNMS\Cache\PermissionsCache::class)->makePartial();
|
|
$perms->shouldReceive('getPortPermissions')->andReturn(collect([
|
|
(object) ['user_id' => 3, 'port_id' => 7],
|
|
(object) ['user_id' => 3, 'port_id' => 2],
|
|
(object) ['user_id' => 4, 'port_id' => 5],
|
|
(object) ['user_id' => 6, 'port_id' => 5],
|
|
]));
|
|
|
|
$port = Port::factory()->make(['port_id' => 7]); /** @var Port $port */
|
|
$this->assertEquals(collect([4, 6]), $perms->usersForPort(5));
|
|
$this->assertEquals(collect([3]), $perms->usersForPort($port));
|
|
$this->assertEquals(collect(), $perms->usersForPort(6));
|
|
$this->assertEmpty($perms->usersForPort(9));
|
|
}
|
|
|
|
public function testUserCanAccessBill()
|
|
{
|
|
$perms = \Mockery::mock(\LibreNMS\Cache\PermissionsCache::class)->makePartial();
|
|
$perms->shouldReceive('getBillPermissions')->andReturn(collect([
|
|
(object) ['user_id' => 43, 'bill_id' => 54],
|
|
(object) ['user_id' => 43, 'bill_id' => 32],
|
|
(object) ['user_id' => 14, 'bill_id' => 54],
|
|
]));
|
|
|
|
$bill = Bill::factory()->make(['bill_id' => 54]); /** @var Bill $bill */
|
|
$user = User::factory()->make(['user_id' => 43]); /** @var User $user */
|
|
$this->assertTrue($perms->canAccessBill($bill, 43));
|
|
$this->assertTrue($perms->canAccessBill($bill, $user));
|
|
$this->assertTrue($perms->canAccessBill(54, $user));
|
|
$this->assertTrue($perms->canAccessBill(54, 43));
|
|
$this->assertTrue($perms->canAccessBill(54, 43));
|
|
$this->assertFalse($perms->canAccessBill(54, 23));
|
|
$this->assertFalse($perms->canAccessBill(23, 43));
|
|
$this->assertFalse($perms->canAccessBill(54));
|
|
|
|
\Auth::shouldReceive('id')->once()->andReturn(43);
|
|
$this->assertTrue($perms->canAccessBill(54));
|
|
\Auth::shouldReceive('id')->once()->andReturn(23);
|
|
$this->assertFalse($perms->canAccessBill(54));
|
|
}
|
|
|
|
public function testBillsForUser()
|
|
{
|
|
$perms = \Mockery::mock(\LibreNMS\Cache\PermissionsCache::class)->makePartial();
|
|
$perms->shouldReceive('getBillPermissions')->andReturn(collect([
|
|
(object) ['user_id' => 3, 'bill_id' => 7],
|
|
(object) ['user_id' => 3, 'bill_id' => 2],
|
|
(object) ['user_id' => 4, 'bill_id' => 5],
|
|
]));
|
|
|
|
$this->assertEquals(collect([7, 2]), $perms->billsForUser(3));
|
|
$user = User::factory()->make(['user_id' => 3]); /** @var User $user */
|
|
$this->assertEquals(collect([7, 2]), $perms->billsForUser($user));
|
|
$this->assertEmpty($perms->billsForUser(9));
|
|
$this->assertEquals(collect(), $perms->billsForUser());
|
|
\Auth::shouldReceive('id')->once()->andReturn(4);
|
|
$this->assertEquals(collect([5]), $perms->billsForUser());
|
|
}
|
|
|
|
public function testUsersForBill()
|
|
{
|
|
$perms = \Mockery::mock(\LibreNMS\Cache\PermissionsCache::class)->makePartial();
|
|
$perms->shouldReceive('getBillPermissions')->andReturn(collect([
|
|
(object) ['user_id' => 3, 'bill_id' => 7],
|
|
(object) ['user_id' => 3, 'bill_id' => 2],
|
|
(object) ['user_id' => 4, 'bill_id' => 5],
|
|
(object) ['user_id' => 6, 'bill_id' => 5],
|
|
]));
|
|
|
|
$this->assertEquals(collect([4, 6]), $perms->usersForBill(5));
|
|
$bill = Bill::factory()->make(['bill_id' => 7]); /** @var Bill $bill */
|
|
$this->assertEquals(collect([3]), $perms->usersForBill($bill));
|
|
$this->assertEquals(collect(), $perms->usersForBill(6));
|
|
$this->assertEmpty($perms->usersForBill(9));
|
|
}
|
|
}
|