def test_confirm_verification_code(mock_confirm_code): helpers.confirm_verification_code( email='*****@*****.**', verification_code='1234', ) assert mock_confirm_code.call_count == 1 assert mock_confirm_code.call_args == mock.call({ 'email': '*****@*****.**', 'code': '1234' })
def clean_code(self): try: response = helpers.confirm_verification_code( email=self.cleaned_data['email'], verification_code=self.cleaned_data['code'], ) except HTTPError as error: if error.response.status_code == 400: self.add_error('code', self.MESSAGE_INVALID_CODE) else: raise else: self.cleaned_data['cookies'] = helpers.parse_set_cookie_header( response.headers['set-cookie']) return None
def clean_code(self): try: response = helpers.confirm_verification_code( email=self.cleaned_data['email'], verification_code=self.cleaned_data['code'], ) except HTTPError as error: if error.response.status_code == 400: self.add_error('code', self.MESSAGE_INVALID_CODE) else: raise else: self.cleaned_data['cookies'] = helpers.cookiekjar_to_simple_cookie( response.cookies) return None
def validate_code(self, form, response): try: upstream_response = helpers.confirm_verification_code( email=form.cleaned_data['email'], verification_code=form.cleaned_data['code'], ) except HTTPError as error: if error.response.status_code in [400, 404]: self.storage.set_step_data( constants.VERIFICATION, { form.add_prefix('email'): [form.cleaned_data['email']], form.add_prefix('code'): [None] }) return self.render_revalidation_failure( failed_step=constants.VERIFICATION, form=form) else: raise else: cookies = helpers.parse_set_cookie_header( upstream_response.headers['set-cookie']) response.cookies.update(cookies) self.storage.extra_data['is_new_enrollment'] = True return response