def test_verify_totp_valid(self, monkeypatch):
verify_totp = pretend.call_recorder(lambda *a: True)
monkeypatch.setattr(otp, "verify_totp", verify_totp)
form = forms.ProvisionTOTPForm(data={"totp_value": "123456"},
totp_secret=pretend.stub())
assert form.validate()
def test_verify_totp_invalid(self, monkeypatch):
verify_totp = pretend.call_recorder(lambda *a: False)
monkeypatch.setattr(otp, "verify_totp", verify_totp)
form = forms.ProvisionTOTPForm(data={"totp_value": "123456"},
totp_secret=pretend.stub())
assert not form.validate()
assert form.totp_value.errors.pop() == "Invalid TOTP code. Try again?"
def test_creation(self):
totp_secret = pretend.stub()
form = forms.ProvisionTOTPForm(totp_secret=totp_secret)
assert form.totp_secret is totp_secret
