123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186 |
- <?php
- namespace OCA\OAuth2\Tests\Controller;
- use OCA\OAuth2\Controller\SettingsController;
- use OCA\OAuth2\Db\AccessTokenMapper;
- use OCA\OAuth2\Db\Client;
- use OCA\OAuth2\Db\ClientMapper;
- use OCP\AppFramework\Http;
- use OCP\AppFramework\Http\JSONResponse;
- use OCP\Authentication\Token\IProvider as IAuthTokenProvider;
- use OCP\IL10N;
- use OCP\IRequest;
- use OCP\IUser;
- use OCP\IUserManager;
- use OCP\Security\ICrypto;
- use OCP\Security\ISecureRandom;
- use Test\TestCase;
- class SettingsControllerTest extends TestCase {
-
- private $request;
-
- private $clientMapper;
-
- private $secureRandom;
-
- private $accessTokenMapper;
-
- private $authTokenProvider;
-
- private $userManager;
-
- private $settingsController;
-
- private $l;
-
- private $crypto;
- protected function setUp(): void {
- parent::setUp();
- $this->request = $this->createMock(IRequest::class);
- $this->clientMapper = $this->createMock(ClientMapper::class);
- $this->secureRandom = $this->createMock(ISecureRandom::class);
- $this->accessTokenMapper = $this->createMock(AccessTokenMapper::class);
- $this->authTokenProvider = $this->createMock(IAuthTokenProvider::class);
- $this->userManager = $this->createMock(IUserManager::class);
- $this->crypto = $this->createMock(ICrypto::class);
- $this->l = $this->createMock(IL10N::class);
- $this->l->method('t')
- ->willReturnArgument(0);
- $this->settingsController = new SettingsController(
- 'oauth2',
- $this->request,
- $this->clientMapper,
- $this->secureRandom,
- $this->accessTokenMapper,
- $this->l,
- $this->authTokenProvider,
- $this->userManager,
- $this->crypto
- );
- }
- public function testAddClient(): void {
- $this->secureRandom
- ->expects($this->exactly(2))
- ->method('generate')
- ->with(64, 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789')
- ->willReturnOnConsecutiveCalls(
- 'MySecret',
- 'MyClientIdentifier');
- $this->crypto
- ->expects($this->once())
- ->method('calculateHMAC')
- ->willReturn('MyHashedSecret');
- $client = new Client();
- $client->setName('My Client Name');
- $client->setRedirectUri('https://example.com/');
- $client->setSecret(bin2hex('MyHashedSecret'));
- $client->setClientIdentifier('MyClientIdentifier');
- $this->clientMapper
- ->expects($this->once())
- ->method('insert')
- ->with($this->callback(function (Client $c) {
- return $c->getName() === 'My Client Name' &&
- $c->getRedirectUri() === 'https://example.com/' &&
- $c->getSecret() === bin2hex('MyHashedSecret') &&
- $c->getClientIdentifier() === 'MyClientIdentifier';
- }))->willReturnCallback(function (Client $c) {
- $c->setId(42);
- return $c;
- });
- $result = $this->settingsController->addClient('My Client Name', 'https://example.com/');
- $this->assertInstanceOf(JSONResponse::class, $result);
- $data = $result->getData();
- $this->assertEquals([
- 'id' => 42,
- 'name' => 'My Client Name',
- 'redirectUri' => 'https://example.com/',
- 'clientId' => 'MyClientIdentifier',
- 'clientSecret' => 'MySecret',
- ], $data);
- }
- public function testDeleteClient(): void {
- $userManager = \OC::$server->getUserManager();
-
- $count = 0;
- $function = function (IUser $user) use (&$count): void {
- if ($user->getLastLogin() > 0) {
- $count++;
- }
- };
- $userManager->callForAllUsers($function);
- $user1 = $userManager->createUser('test101', 'test101');
- $user1->updateLastLoginTimestamp();
- $tokenProviderMock = $this->getMockBuilder(IAuthTokenProvider::class)->getMock();
-
- $tokenProviderMock
- ->expects($this->exactly($count + 1))
- ->method('invalidateTokensOfUser')
- ->with($this->isType('string'), 'My Client Name');
- $client = new Client();
- $client->setId(123);
- $client->setName('My Client Name');
- $client->setRedirectUri('https://example.com/');
- $client->setSecret(bin2hex('MyHashedSecret'));
- $client->setClientIdentifier('MyClientIdentifier');
- $this->clientMapper
- ->method('getByUid')
- ->with(123)
- ->willReturn($client);
- $this->accessTokenMapper
- ->expects($this->once())
- ->method('deleteByClientId')
- ->with(123);
- $this->clientMapper
- ->expects($this->once())
- ->method('delete')
- ->with($client);
- $settingsController = new SettingsController(
- 'oauth2',
- $this->request,
- $this->clientMapper,
- $this->secureRandom,
- $this->accessTokenMapper,
- $this->l,
- $tokenProviderMock,
- $userManager,
- $this->crypto
- );
- $result = $settingsController->deleteClient(123);
- $this->assertInstanceOf(JSONResponse::class, $result);
- $this->assertEquals([], $result->getData());
- $user1->delete();
- }
- public function testInvalidRedirectUri(): void {
- $result = $this->settingsController->addClient('test', 'invalidurl');
- $this->assertEquals(Http::STATUS_BAD_REQUEST, $result->getStatus());
- $this->assertSame(['message' => 'Your redirect URL needs to be a full URL for example: https://yourdomain.com/path'], $result->getData());
- }
- }
|