def test_verify_WHEN_called_AND_status_is_validated_THEN_raise_exception(
self):
otp = OTP(
method=self.otp_method,
to=self.to,
hashed_otp=self.hashed_otp,
encoder=self.hash_stub,
status=OTPStatusEnum.VALIDATED.value,
)
with self.assertRaises(OTPVerificationFailedException):
otp.verify(self.invalid_otp_code)
def test_generate_otp_WHEN_called_AND_hashed_otp_already_exists_THEN_raise_exception(
self, ):
otp = OTP(
method=self.otp_method,
to=self.to,
hashed_otp=self.hashed_otp,
encoder=self.hash_stub,
status=OTPStatusEnum.PENDING.value,
)
with self.assertRaises(OTPGenerationFailedException):
otp.generate_otp()
def test_verify_WHEN_called_with_incorrect_otp_AND_status_is_pending_THEN_raise_exception(
self, ):
otp = OTP(
method=self.otp_method,
to=self.to,
hashed_otp=self.hashed_otp,
encoder=self.hash_stub,
status=OTPStatusEnum.PENDING.value,
)
with self.assertRaises(OTPVerificationFailedException):
otp.verify(self.invalid_otp_code)
def test_verify_WHEN_called_AND_status_is_pending_THEN_calls_encoder_compare(
self):
otp = OTP(
method=self.otp_method,
to=self.to,
hashed_otp=self.hashed_otp,
encoder=self.hash_stub,
status=OTPStatusEnum.PENDING.value,
)
otp.verify(self.otp_code)
self.hash_stub.compare.assert_called_once_with(string=self.otp_code,
hashed=self.hashed_otp)
def test_generate_otp_WHEN_called_AND_hashed_otp_is_none_THEN_calls_encoder_to_hash(
self, randint_mock):
randint_mock.return_value = self.otp_code
otp = OTP(
method=self.otp_method,
to=self.to,
encoder=self.hash_stub,
status=OTPStatusEnum.PENDING.value,
)
otp.generate_otp()
self.hash_stub.to_hash.assert_called_once_with(password=self.otp_code)
def test_verify_WHEN_called_with_correct_otp_AND_status_is_pending_THEN_set_status_to_validated(
self, ):
expected_status = OTPStatusEnum.VALIDATED.value
otp = OTP(
method=self.otp_method,
to=self.to,
hashed_otp=self.hashed_otp,
encoder=self.hash_stub,
status=OTPStatusEnum.PENDING.value,
)
otp.verify(self.otp_code)
actual_status = otp.status
self.assertEqual(expected_status, actual_status)
def test_generate_otp_WHEN_called_AND_hashed_otp_is_none_THEN_returns_otp_code(
self, randint_mock):
randint_mock.return_value = self.otp_code
expected_otp_code = self.otp_code
otp = OTP(
method=self.otp_method,
to=self.to,
encoder=self.hash_stub,
status=OTPStatusEnum.PENDING.value,
)
actual_otp_code = otp.generate_otp()
self.assertEqual(expected_otp_code, actual_otp_code)
def __build_otp(self, encoder: HashAbstract, to: str) -> OTP:
return OTP(
method=OTPMethodEnum.EMAIL.value,
to=to,
encoder=encoder,
status=OTPStatusEnum.PENDING.value,
)
def test_generate_otp_WHEN_called_AND_hashed_otp_is_none_THEN_calls_randint(
self, randint_mock):
randint_mock.return_value = self.otp_code
expected_call_arguments = (0, 999999)
otp = OTP(
method=self.otp_method,
to=self.to,
encoder=self.hash_stub,
status=OTPStatusEnum.PENDING.value,
)
otp.generate_otp()
actual_call_arguments = randint_mock.call_args[0]
self.assertEqual(expected_call_arguments, actual_call_arguments)
def setUp(self) -> None:
self.database_sesssion_mock = mock.Mock()
self.encoder = mock.Mock(spec=HashAbstract)
self.otp = OTP(
method=OTPMethodEnum.EMAIL.value,
to=self.to,
encoder=self.encoder,
status=OTPStatusEnum.PENDING.value,
)
self.serialized_otp = 'serialized-otp-entity'
def setUp(self) -> None:
self.repository_mock = mock.Mock(spec=OTPRepositoryAbstract)
self.hash_mock = mock.Mock(spec=HashAbstract)
self.mailer_mock = mock.Mock(spec=MailerAbstract)
self.otp = OTP(
method=OTPMethodEnum.EMAIL.value,
to=self.to,
encoder=self.hash_mock,
status=OTPStatusEnum.PENDING.value,
)
def setUpClass(cls) -> None:
cls.hash_mock = mock.Mock()
cls.repository_mock = mock.Mock()
cls.to = '*****@*****.**'
cls.otp_code = '123456'
cls.hashed_otp = str.encode(cls.otp_code)
cls.otp = OTP(
method=OTPMethodEnum.EMAIL.value,
to=cls.to,
encoder=cls.hash_mock,
status=OTPStatusEnum.PENDING.value,
hashed_otp=cls.hashed_otp,
)
def test_otp_WHEN_created_THEN_set_correct_attributes(self):
expected_otp_dict = {
'method': self.otp_method,
'to': self.to,
'encoder': self.hash_stub,
'status': OTPStatusEnum.PENDING.value,
'hashed_otp': self.hashed_otp,
'ttl': 60,
}
otp = OTP(
method=self.otp_method,
to=self.to,
encoder=self.hash_stub,
status=OTPStatusEnum.PENDING.value,
hashed_otp=self.hashed_otp,
)
actual_otp_dict = otp.__dict__
self.assertDictEqual(expected_otp_dict, actual_otp_dict)