forked from mirror/Part-DB.Part-DB-server
Added an service for generating Backup codes and added some tests.
This commit is contained in:
parent
452fc3e78a
commit
fba5f9794f
16 changed files with 245 additions and 7 deletions
|
@ -26,6 +26,7 @@ use Symfony\Component\Security\Core\Exception\AccessDeniedException;
|
|||
|
||||
/**
|
||||
* @group slow
|
||||
* @group DB
|
||||
*/
|
||||
abstract class AbstractAdminControllerTest extends WebTestCase
|
||||
{
|
||||
|
|
|
@ -25,6 +25,7 @@ use App\Entity\Attachments\AttachmentType;
|
|||
|
||||
/**
|
||||
* @group slow
|
||||
* @group DB
|
||||
*/
|
||||
class AttachmentTypeControllerTest extends AbstractAdminControllerTest
|
||||
{
|
||||
|
|
|
@ -25,6 +25,7 @@ use App\Entity\Parts\Category;
|
|||
|
||||
/**
|
||||
* @group slow
|
||||
* @group DB
|
||||
*/
|
||||
class CategoryControllerTest extends AbstractAdminControllerTest
|
||||
{
|
||||
|
|
|
@ -25,6 +25,7 @@ use App\Entity\Devices\Device;
|
|||
|
||||
/**
|
||||
* @group slow
|
||||
* @group DB
|
||||
*/
|
||||
class DeviceControllerTest extends AbstractAdminControllerTest
|
||||
{
|
||||
|
|
|
@ -25,6 +25,7 @@ use App\Entity\Parts\Footprint;
|
|||
|
||||
/**
|
||||
* @group slow
|
||||
* @group DB
|
||||
*/
|
||||
class FootprintControllerTest extends AbstractAdminControllerTest
|
||||
{
|
||||
|
|
|
@ -25,6 +25,7 @@ use App\Entity\Parts\Manufacturer;
|
|||
|
||||
/**
|
||||
* @group slow
|
||||
* @group DB
|
||||
*/
|
||||
class ManufacturerControllerTest extends AbstractAdminControllerTest
|
||||
{
|
||||
|
|
|
@ -25,6 +25,7 @@ use App\Entity\Parts\MeasurementUnit;
|
|||
|
||||
/**
|
||||
* @group slow
|
||||
* @group DB
|
||||
*/
|
||||
class MeasurementUnitControllerTest extends AbstractAdminControllerTest
|
||||
{
|
||||
|
|
|
@ -25,6 +25,7 @@ use App\Entity\Parts\Storelocation;
|
|||
|
||||
/**
|
||||
* @group slow
|
||||
* @group DB
|
||||
*/
|
||||
class StorelocationControllerTest extends AbstractAdminControllerTest
|
||||
{
|
||||
|
|
|
@ -25,6 +25,7 @@ use App\Entity\Parts\Supplier;
|
|||
|
||||
/**
|
||||
* @group slow
|
||||
* @group DB
|
||||
*/
|
||||
class SupplierControllerTest extends AbstractAdminControllerTest
|
||||
{
|
||||
|
|
|
@ -27,6 +27,7 @@ use Symfony\Bundle\FrameworkBundle\Test\WebTestCase;
|
|||
|
||||
/**
|
||||
* @group slow
|
||||
* @group DB
|
||||
*/
|
||||
class RedirectControllerTest extends WebTestCase
|
||||
{
|
||||
|
|
|
@ -21,6 +21,7 @@
|
|||
|
||||
namespace App\Tests\Entity\UserSystem;
|
||||
|
||||
use App\Entity\UserSystem\U2FKey;
|
||||
use App\Entity\UserSystem\User;
|
||||
use PHPUnit\Framework\TestCase;
|
||||
|
||||
|
@ -36,4 +37,88 @@ class UserTest extends TestCase
|
|||
$this->assertEquals('John Doe', $user->getFullName(false));
|
||||
$this->assertEquals('John Doe (username)', $user->getFullName(true));
|
||||
}
|
||||
|
||||
public function googleAuthenticatorEnabledDataProvider() : array
|
||||
{
|
||||
return [
|
||||
[null, false],
|
||||
['', false],
|
||||
['SSSk38498', true]
|
||||
];
|
||||
}
|
||||
|
||||
/**
|
||||
* @dataProvider googleAuthenticatorEnabledDataProvider
|
||||
*/
|
||||
public function testIsGoogleAuthenticatorEnabled(?string $secret, bool $expected)
|
||||
{
|
||||
$user = new User();
|
||||
$user->setGoogleAuthenticatorSecret($secret);
|
||||
$this->assertSame($expected ,$user->isGoogleAuthenticatorEnabled());
|
||||
}
|
||||
|
||||
public function testSetBackupCodes()
|
||||
{
|
||||
$user = new User();
|
||||
$codes = ["test", "invalid", "test"];
|
||||
$user->setBackupCodes($codes);
|
||||
// Backup Codes generation date must be changed!
|
||||
$this->assertEquals(new \DateTime(), $user->getBackupCodesGenerationDate(), '', 0.1);
|
||||
$this->assertEquals($codes, $user->getBackupCodes());
|
||||
|
||||
//Test what happens if we delete the backup keys
|
||||
$user->setBackupCodes([]);
|
||||
$this->assertEmpty($user->getBackupCodes());
|
||||
$this->assertNull($user->getBackupCodesGenerationDate());
|
||||
}
|
||||
|
||||
public function testIsBackupCode()
|
||||
{
|
||||
$user = new User();
|
||||
$codes = ['aaaa', 'bbbb', 'cccc', 'dddd'];
|
||||
$user->setBackupCodes($codes);
|
||||
|
||||
$this->assertTrue($user->isBackupCode('aaaa'));
|
||||
$this->assertTrue($user->isBackupCode('cccc'));
|
||||
|
||||
$this->assertFalse($user->isBackupCode(''));
|
||||
$this->assertFalse($user->isBackupCode('zzzz'));
|
||||
}
|
||||
|
||||
public function testInvalidateBackupCode()
|
||||
{
|
||||
$user = new User();
|
||||
$codes = ['aaaa', 'bbbb', 'cccc', 'dddd'];
|
||||
$user->setBackupCodes($codes);
|
||||
|
||||
//Ensure the code is valid
|
||||
$this->assertTrue($user->isBackupCode('aaaa'));
|
||||
$this->assertTrue($user->isBackupCode('bbbb'));
|
||||
//Invalidate code, afterwards the code has to be invalid!
|
||||
$user->invalidateBackupCode('bbbb');
|
||||
$this->assertFalse($user->isBackupCode('bbbb'));
|
||||
$this->assertTrue($user->isBackupCode('aaaa'));
|
||||
|
||||
//No exception must happen, when we try to invalidate an not existing backup key!
|
||||
$user->invalidateBackupCode('zzzz');
|
||||
}
|
||||
|
||||
public function testInvalidateTrustedDeviceTokens()
|
||||
{
|
||||
$user = new User();
|
||||
$old_value = $user->getTrustedTokenVersion();
|
||||
//To invalidate the token, the new value must be bigger than the old value
|
||||
$user->invalidateTrustedDeviceTokens();
|
||||
$this->assertGreaterThan($old_value, $user->getTrustedTokenVersion());
|
||||
}
|
||||
|
||||
public function testIsU2fEnabled()
|
||||
{
|
||||
$user = new User();
|
||||
$user->addU2FKey(new U2FKey());
|
||||
$this->assertTrue($user->isU2FAuthEnabled());
|
||||
|
||||
$user->getU2FKeys()->clear();
|
||||
$this->assertFalse($user->isU2FAuthEnabled());
|
||||
}
|
||||
}
|
||||
|
|
|
@ -28,6 +28,9 @@ use App\Services\ElementTypeNameGenerator;
|
|||
use App\Services\EntityImporter;
|
||||
use Symfony\Bundle\FrameworkBundle\Test\WebTestCase;
|
||||
|
||||
/**
|
||||
* @group DB
|
||||
*/
|
||||
class EntityImporterTest extends WebTestCase
|
||||
{
|
||||
/**
|
||||
|
|
56
tests/Services/TFA/BackupCodeGeneratorTest.php
Normal file
56
tests/Services/TFA/BackupCodeGeneratorTest.php
Normal file
|
@ -0,0 +1,56 @@
|
|||
<?php
|
||||
|
||||
namespace App\Tests\Services\TFA;
|
||||
|
||||
use App\Services\TFA\BackupCodeGenerator;
|
||||
use PHPUnit\Framework\TestCase;
|
||||
|
||||
class BackupCodeGeneratorTest extends TestCase
|
||||
{
|
||||
/**
|
||||
* Test if an exception is thrown if you are using a too high code length
|
||||
*/
|
||||
public function testLengthUpperLimit()
|
||||
{
|
||||
$this->expectException(\RuntimeException::class);
|
||||
new BackupCodeGenerator(33, 10);
|
||||
}
|
||||
|
||||
/**
|
||||
* Test if an exception is thrown if you are using a too high code length
|
||||
*/
|
||||
public function testLengthLowerLimit()
|
||||
{
|
||||
$this->expectException(\RuntimeException::class);
|
||||
new BackupCodeGenerator(4, 10);
|
||||
}
|
||||
|
||||
|
||||
public function codeLengthDataProvider()
|
||||
{
|
||||
return [[6], [8], [10], [16]];
|
||||
}
|
||||
|
||||
/**
|
||||
* @dataProvider codeLengthDataProvider
|
||||
*/
|
||||
public function testGenerateSingleCode(int $code_length)
|
||||
{
|
||||
$generator = new BackupCodeGenerator($code_length, 10);
|
||||
$this->assertRegExp("/^([a-f0-9]){{$code_length}}\$/", $generator->generateSingleCode());
|
||||
}
|
||||
|
||||
public function codeCountDataProvider()
|
||||
{
|
||||
return [[2], [8], [10]];
|
||||
}
|
||||
|
||||
/**
|
||||
* @dataProvider codeCountDataProvider
|
||||
*/
|
||||
public function testGenerateCodeSet(int $code_count)
|
||||
{
|
||||
$generator = new BackupCodeGenerator(8, $code_count);
|
||||
$this->assertCount($code_count, $generator->generateCodeSet());
|
||||
}
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue