123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410 |
- <?php
- declare(strict_types=1);
- namespace Test\Authentication\Token;
- use Doctrine\DBAL\Exception\UniqueConstraintViolationException;
- use OC\Authentication\Exceptions\InvalidTokenException;
- use OC\Authentication\Token\IToken;
- use OC\Authentication\Token\Manager;
- use OC\Authentication\Token\PublicKeyToken;
- use OC\Authentication\Token\PublicKeyTokenProvider;
- use PHPUnit\Framework\MockObject\MockObject;
- use Test\TestCase;
- class ManagerTest extends TestCase {
-
- private $publicKeyTokenProvider;
-
- private $manager;
- protected function setUp(): void {
- parent::setUp();
- $this->publicKeyTokenProvider = $this->createMock(PublicKeyTokenProvider::class);
- $this->manager = new Manager(
- $this->publicKeyTokenProvider
- );
- }
- public function testGenerateToken() {
- $token = new PublicKeyToken();
- $this->publicKeyTokenProvider->expects($this->once())
- ->method('generateToken')
- ->with(
- 'token',
- 'uid',
- 'loginName',
- 'password',
- 'name',
- IToken::TEMPORARY_TOKEN,
- IToken::REMEMBER
- )->willReturn($token);
- $actual = $this->manager->generateToken(
- 'token',
- 'uid',
- 'loginName',
- 'password',
- 'name',
- IToken::TEMPORARY_TOKEN,
- IToken::REMEMBER
- );
- $this->assertSame($token, $actual);
- }
- public function testGenerateConflictingToken() {
-
- $exception = $this->createMock(UniqueConstraintViolationException::class);
- $token = new PublicKeyToken();
- $token->setUid('uid');
- $this->publicKeyTokenProvider->expects($this->once())
- ->method('generateToken')
- ->with(
- 'token',
- 'uid',
- 'loginName',
- 'password',
- 'name',
- IToken::TEMPORARY_TOKEN,
- IToken::REMEMBER
- )->willThrowException($exception);
- $this->publicKeyTokenProvider->expects($this->once())
- ->method('getToken')
- ->with('token')
- ->willReturn($token);
- $actual = $this->manager->generateToken(
- 'token',
- 'uid',
- 'loginName',
- 'password',
- 'name',
- IToken::TEMPORARY_TOKEN,
- IToken::REMEMBER
- );
- $this->assertSame($token, $actual);
- }
- public function testGenerateTokenTooLongName() {
- $token = $this->createMock(IToken::class);
- $token->method('getName')
- ->willReturn(str_repeat('a', 120) . '…');
- $this->publicKeyTokenProvider->expects($this->once())
- ->method('generateToken')
- ->with(
- 'token',
- 'uid',
- 'loginName',
- 'password',
- str_repeat('a', 120) . '…',
- IToken::TEMPORARY_TOKEN,
- IToken::REMEMBER
- )->willReturn($token);
- $actual = $this->manager->generateToken(
- 'token',
- 'uid',
- 'loginName',
- 'password',
- str_repeat('a', 200),
- IToken::TEMPORARY_TOKEN,
- IToken::REMEMBER
- );
- $this->assertSame(121, mb_strlen($actual->getName()));
- }
- public function tokenData(): array {
- return [
- [new PublicKeyToken()],
- [$this->createMock(IToken::class)],
- ];
- }
- protected function setNoCall(IToken $token) {
- if (!($token instanceof PublicKeyToken)) {
- $this->publicKeyTokenProvider->expects($this->never())
- ->method($this->anything());
- }
- }
- protected function setCall(IToken $token, string $function, $return = null) {
- if ($token instanceof PublicKeyToken) {
- $this->publicKeyTokenProvider->expects($this->once())
- ->method($function)
- ->with($token)
- ->willReturn($return);
- }
- }
- protected function setException(IToken $token) {
- if (!($token instanceof PublicKeyToken)) {
- $this->expectException(InvalidTokenException::class);
- }
- }
-
- public function testUpdateToken(IToken $token) {
- $this->setNoCall($token);
- $this->setCall($token, 'updateToken');
- $this->setException($token);
- $this->manager->updateToken($token);
- }
-
- public function testUpdateTokenActivity(IToken $token) {
- $this->setNoCall($token);
- $this->setCall($token, 'updateTokenActivity');
- $this->setException($token);
- $this->manager->updateTokenActivity($token);
- }
-
- public function testGetPassword(IToken $token) {
- $this->setNoCall($token);
- $this->setCall($token, 'getPassword', 'password');
- $this->setException($token);
- $result = $this->manager->getPassword($token, 'tokenId', 'password');
- $this->assertSame('password', $result);
- }
-
- public function testSetPassword(IToken $token) {
- $this->setNoCall($token);
- $this->setCall($token, 'setPassword');
- $this->setException($token);
- $this->manager->setPassword($token, 'tokenId', 'password');
- }
- public function testInvalidateTokens() {
- $this->publicKeyTokenProvider->expects($this->once())
- ->method('invalidateToken')
- ->with('token');
- $this->manager->invalidateToken('token');
- }
- public function testInvalidateTokenById() {
- $this->publicKeyTokenProvider->expects($this->once())
- ->method('invalidateTokenById')
- ->with('uid', 42);
- $this->manager->invalidateTokenById('uid', 42);
- }
- public function testInvalidateOldTokens() {
- $this->publicKeyTokenProvider->expects($this->once())
- ->method('invalidateOldTokens');
- $this->manager->invalidateOldTokens();
- }
- public function testInvalidateLastUsedBefore() {
- $this->publicKeyTokenProvider->expects($this->once())
- ->method('invalidateLastUsedBefore')
- ->with('user', 946684800);
- $this->manager->invalidateLastUsedBefore('user', 946684800);
- }
- public function testGetTokenByUser() {
- $t1 = new PublicKeyToken();
- $t2 = new PublicKeyToken();
- $this->publicKeyTokenProvider
- ->method('getTokenByUser')
- ->willReturn([$t1, $t2]);
- $result = $this->manager->getTokenByUser('uid');
- $this->assertEquals([$t1, $t2], $result);
- }
- public function testRenewSessionTokenPublicKey() {
- $this->publicKeyTokenProvider->expects($this->once())
- ->method('renewSessionToken')
- ->with('oldId', 'newId');
- $this->manager->renewSessionToken('oldId', 'newId');
- }
- public function testRenewSessionInvalid() {
- $this->publicKeyTokenProvider->expects($this->once())
- ->method('renewSessionToken')
- ->with('oldId', 'newId')
- ->willThrowException(new InvalidTokenException());
- $this->expectException(InvalidTokenException::class);
- $this->manager->renewSessionToken('oldId', 'newId');
- }
- public function testGetTokenByIdPublicKey() {
- $token = $this->createMock(IToken::class);
- $this->publicKeyTokenProvider->expects($this->once())
- ->method('getTokenById')
- ->with(42)
- ->willReturn($token);
- $this->assertSame($token, $this->manager->getTokenById(42));
- }
- public function testGetTokenByIdInvalid() {
- $this->publicKeyTokenProvider->expects($this->once())
- ->method('getTokenById')
- ->with(42)
- ->willThrowException(new InvalidTokenException());
- $this->expectException(InvalidTokenException::class);
- $this->manager->getTokenById(42);
- }
- public function testGetTokenPublicKey() {
- $token = new PublicKeyToken();
- $this->publicKeyTokenProvider
- ->method('getToken')
- ->with('tokenId')
- ->willReturn($token);
- $this->assertSame($token, $this->manager->getToken('tokenId'));
- }
- public function testGetTokenInvalid() {
- $this->publicKeyTokenProvider
- ->method('getToken')
- ->with('tokenId')
- ->willThrowException(new InvalidTokenException());
- $this->expectException(InvalidTokenException::class);
- $this->manager->getToken('tokenId');
- }
- public function testRotateInvalid() {
- $this->expectException(InvalidTokenException::class);
- $this->manager->rotate($this->createMock(IToken::class), 'oldId', 'newId');
- }
- public function testRotatePublicKey() {
- $token = new PublicKeyToken();
- $this->publicKeyTokenProvider
- ->method('rotate')
- ->with($token, 'oldId', 'newId')
- ->willReturn($token);
- $this->assertSame($token, $this->manager->rotate($token, 'oldId', 'newId'));
- }
- public function testMarkPasswordInvalidPublicKey() {
- $token = $this->createMock(PublicKeyToken::class);
- $this->publicKeyTokenProvider->expects($this->once())
- ->method('markPasswordInvalid')
- ->with($token, 'tokenId');
- $this->manager->markPasswordInvalid($token, 'tokenId');
- }
- public function testMarkPasswordInvalidInvalidToken() {
- $this->expectException(InvalidTokenException::class);
- $this->manager->markPasswordInvalid($this->createMock(IToken::class), 'tokenId');
- }
- public function testUpdatePasswords() {
- $this->publicKeyTokenProvider->expects($this->once())
- ->method('updatePasswords')
- ->with('uid', 'pass');
- $this->manager->updatePasswords('uid', 'pass');
- }
- public function testInvalidateTokensOfUserNoClientName() {
- $t1 = new PublicKeyToken();
- $t2 = new PublicKeyToken();
- $t1->setId(123);
- $t2->setId(456);
- $this->publicKeyTokenProvider
- ->expects($this->once())
- ->method('getTokenByUser')
- ->with('theUser')
- ->willReturn([$t1, $t2]);
- $this->publicKeyTokenProvider
- ->expects($this->exactly(2))
- ->method('invalidateTokenById')
- ->withConsecutive(
- ['theUser', 123],
- ['theUser', 456],
- );
- $this->manager->invalidateTokensOfUser('theUser', null);
- }
- public function testInvalidateTokensOfUserClientNameGiven() {
- $t1 = new PublicKeyToken();
- $t2 = new PublicKeyToken();
- $t3 = new PublicKeyToken();
- $t1->setId(123);
- $t1->setName('Firefox session');
- $t2->setId(456);
- $t2->setName('My Client Name');
- $t3->setId(789);
- $t3->setName('mobile client');
- $this->publicKeyTokenProvider
- ->expects($this->once())
- ->method('getTokenByUser')
- ->with('theUser')
- ->willReturn([$t1, $t2, $t3]);
- $this->publicKeyTokenProvider
- ->expects($this->once())
- ->method('invalidateTokenById')
- ->with('theUser', 456);
- $this->manager->invalidateTokensOfUser('theUser', 'My Client Name');
- }
- }
|