szurubooru/tests/Services/PrivilegeServiceTest.php

<?php
namespace Szurubooru\Tests\Services;

class PrivilegeServiceTest extends \Szurubooru\Tests\AbstractTestCase
{
	private $configMock;
	private $authServiceMock;

	public function setUp()
	{
		parent::setUp();
		$this->configMock = $this->mockConfig();
		$this->authServiceMock = $this->mock(\Szurubooru\Services\AuthService::class);
	}

	public function testReadingConfig()
	{
		$testUser = new \Szurubooru\Entities\User();
		$testUser->setName('dummy');
		$testUser->setAccessRank(\Szurubooru\Entities\User::ACCESS_RANK_POWER_USER);
		$this->authServiceMock->expects($this->atLeastOnce())->method('getLoggedInUser')->willReturn($testUser);

		$privilege = \Szurubooru\Privilege::LIST_USERS;
		$this->configMock->set('security/privileges/' . $privilege, 'powerUser');

		$privilegeService = $this->getPrivilegeService();
		$this->assertEquals([$privilege], $privilegeService->getCurrentPrivileges());
		$this->assertTrue($privilegeService->hasPrivilege($privilege));
	}

	public function testIsLoggedInByNameString()
	{
		$testUser1 = new \Szurubooru\Entities\User();
		$testUser1->setName('dummy');
		$testUser2 = new \Szurubooru\Entities\User();
		$testUser2->setName('godzilla');
		$this->authServiceMock->expects($this->atLeastOnce())->method('getLoggedInUser')->willReturn($testUser1);

		$privilegeService = $this->getPrivilegeService();
		$this->assertTrue($privilegeService->isLoggedIn($testUser1->getName()));
		$this->assertFalse($privilegeService->isLoggedIn($testUser2->getName()));
	}

	public function testIsLoggedInByEmailString()
	{
		$testUser1 = new \Szurubooru\Entities\User();
		$testUser1->setName('user1');
		$testUser1->setEmail('dummy');
		$testUser2 = new \Szurubooru\Entities\User();
		$testUser2->setName('user2');
		$testUser2->setEmail('godzilla');
		$this->authServiceMock->expects($this->atLeastOnce())->method('getLoggedInUser')->willReturn($testUser1);

		$privilegeService = $this->getPrivilegeService();
		$this->assertTrue($privilegeService->isLoggedIn($testUser1->getEmail()));
		$this->assertFalse($privilegeService->isLoggedIn($testUser2->getEmail()));
	}

	public function testIsLoggedInByUserId()
	{
		$testUser1 = new \Szurubooru\Entities\User('dummy');
		$testUser2 = new \Szurubooru\Entities\User('godzilla');
		$this->authServiceMock->expects($this->atLeastOnce())->method('getLoggedInUser')->willReturn($testUser1);

		$privilegeService = $this->getPrivilegeService();
		$this->assertTrue($privilegeService->isLoggedIn($testUser1));
		$this->assertFalse($privilegeService->isLoggedIn($testUser2));
	}

	public function testIsLoggedInByUserName()
	{
		$testUser1 = new \Szurubooru\Entities\User();
		$testUser1->setName('dummy');
		$testUser2 = new \Szurubooru\Entities\User();
		$testUser2->setName('godzilla');
		$this->authServiceMock->expects($this->atLeastOnce())->method('getLoggedInUser')->willReturn($testUser1);

		$privilegeService = $this->getPrivilegeService();
		$this->assertFalse($privilegeService->isLoggedIn($testUser1));
		$this->assertFalse($privilegeService->isLoggedIn($testUser2));
	}

	public function testIsLoggedInByInvalidObject()
	{
		$testUser = new \Szurubooru\Entities\User();
		$testUser->setName('dummy');
		$this->authServiceMock->expects($this->atLeastOnce())->method('getLoggedInUser')->willReturn($testUser);

		$rubbish = new \StdClass;
		$privilegeService = $this->getPrivilegeService();
		$this->setExpectedException(\InvalidArgumentException::class);
		$this->assertTrue($privilegeService->isLoggedIn($rubbish));
	}

	private function getPrivilegeService()
	{
		return new \Szurubooru\Services\PrivilegeService(
			$this->configMock,
			$this->authServiceMock);
	}
}