Add more exception tests
This commit is contained in:
parent
038e5e1b7d
commit
a6bca34677
1 changed files with 30 additions and 0 deletions
|
@ -98,6 +98,36 @@ class SecurityControllerTest extends ControllerTestCase
|
|||
$this->assertResponseJsonEquals(['qrImage' => 'qrCodeImage'], $response);
|
||||
}
|
||||
|
||||
/**
|
||||
* Test TOTP setting controller when no exception is thrown by the service.
|
||||
*/
|
||||
public function testSetTotpControllerSuccess()
|
||||
{
|
||||
$model = $this->generateRequestUserModel();
|
||||
|
||||
$this->request->shouldReceive('input')->with('token')->once()->andReturn('testToken');
|
||||
$this->toggleTwoFactorService->shouldReceive('handle')->with($model, 'testToken')->once();
|
||||
|
||||
$response = $this->getController()->setTotp($this->request);
|
||||
$this->assertIsResponse($response);
|
||||
$this->assertSame('true', $response->getContent());
|
||||
}
|
||||
|
||||
/**
|
||||
* Test TOTP setting controller when an exception is thrown by the service.
|
||||
*/
|
||||
public function testSetTotpControllerWhenExceptionIsThrown()
|
||||
{
|
||||
$model = $this->generateRequestUserModel();
|
||||
|
||||
$this->request->shouldReceive('input')->with('token')->once()->andReturn('testToken');
|
||||
$this->toggleTwoFactorService->shouldReceive('handle')->with($model, 'testToken')->once()->andThrow(new TwoFactorAuthenticationTokenInvalid());
|
||||
|
||||
$response = $this->getController()->setTotp($this->request);
|
||||
$this->assertIsResponse($response);
|
||||
$this->assertSame('false', $response->getContent());
|
||||
}
|
||||
|
||||
/**
|
||||
* Test the disable totp controller when no exception is thrown by the service.
|
||||
*/
|
||||
|
|
Loading…
Reference in a new issue