Fix test coverage.

This commit is contained in:
James Cole
2019-08-09 05:58:52 +02:00
parent dcbc2ca0c3
commit fd2f4e1459
6 changed files with 24 additions and 7 deletions

View File

@@ -98,9 +98,6 @@ class ProfileControllerTest extends TestCase
$userRepos->shouldReceive('hasRole')->withArgs([Mockery::any(), 'owner'])->atLeast()->once()->andReturn(true);
$userRepos->shouldReceive('hasRole')->withArgs([Mockery::any(), 'demo'])->atLeast()->once()->andReturn(false);
// set recovery codes.
Preferences::shouldReceive('set')->withArgs(['mfa_recovery', Mockery::any()])->atLeast()->once();
Google2FA::shouldReceive('generateSecretKey')->andReturn('secret');
Google2FA::shouldReceive('getQRCodeInline')->andReturn('long-data-url');
@@ -448,6 +445,16 @@ class ProfileControllerTest extends TestCase
$userRepos->shouldReceive('setMFACode')->withArgs([Mockery::any(), $secret])->atLeast()->once();
// set recovery history
Preferences::shouldReceive('set')->withArgs(['mfa_history', Mockery::any()])->atLeast()->once();
// set recovery codes.
Preferences::shouldReceive('set')->withArgs(['mfa_recovery', null])->atLeast()->once();
$pref = new Preference;
$pref->data= [];
Preferences::shouldReceive('get')->withArgs(['mfa_history', []])->atLeast()->once()->andReturn($pref);
Preferences::shouldReceive('mark')->once();
Google2FA::shouldReceive('verifyKey')->withArgs([$secret, $key])->andReturn(true);