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