appManager = $this->createMock(IAppManager::class); $this->user = $this->createMock(IUser::class); $this->registrationContext = $this->createMock(RegistrationContext::class); $coordinator = $this->createMock(Coordinator::class); $coordinator->method('getRegistrationContext') ->willReturn($this->registrationContext); $this->loader = new ProviderLoader($this->appManager, $coordinator); } public function testFailHardIfProviderCanNotBeLoaded(): void { $this->expectException(\Exception::class); $this->expectExceptionMessage('Could not load two-factor auth provider \\OCA\\MyFaulty2faApp\\DoesNotExist'); $this->appManager->expects($this->once()) ->method('getEnabledAppsForUser') ->with($this->user) ->willReturn(['mail', 'twofactor_totp']); $this->appManager ->method('getAppInfo') ->willReturnMap([ ['mail', false, null, []], ['twofactor_totp', false, null, [ 'two-factor-providers' => [ '\\OCA\\MyFaulty2faApp\\DoesNotExist', ], ]], ]); $this->loader->getProviders($this->user); } public function testGetProviders(): void { $provider = $this->createMock(IProvider::class); $provider->method('getId')->willReturn('test'); \OC::$server->registerService('\\OCA\\TwoFactorTest\\Provider', function () use ($provider) { return $provider; }); $this->appManager->expects($this->once()) ->method('getEnabledAppsForUser') ->with($this->user) ->willReturn(['twofactor_test']); $this->appManager ->method('getAppInfo') ->with('twofactor_test') ->willReturn(['two-factor-providers' => [ '\\OCA\\TwoFactorTest\\Provider', ]]); $providers = $this->loader->getProviders($this->user); $this->assertCount(1, $providers); $this->assertArrayHasKey('test', $providers); $this->assertSame($provider, $providers['test']); } public function testGetProvidersBootstrap(): void { $provider = $this->createMock(IProvider::class); $provider->method('getId')->willReturn('test'); \OC::$server->registerService('\\OCA\\TwoFactorTest\\Provider', function () use ($provider) { return $provider; }); $this->appManager->expects($this->once()) ->method('getEnabledAppsForUser') ->with($this->user) ->willReturn([]); $this->registrationContext->method('getTwoFactorProviders') ->willReturn([ new ServiceRegistration('twofactor_test', '\\OCA\\TwoFactorTest\\Provider') ]); $providers = $this->loader->getProviders($this->user); $this->assertCount(1, $providers); $this->assertArrayHasKey('test', $providers); $this->assertSame($provider, $providers['test']); } }