def test_register_ok(self): data = self._get_register_user_data(password='******') request = self.create_post_request(data) with self.assert_one_mail_sent() as sent_emails, self.timer() as timer: response = self.view_func(request) self.assert_valid_response(response, status.HTTP_201_CREATED) user_id = response.data['id'] # Check database state. user = self.user_class.objects.get(id=user_id) self.assertEqual(user.username, data['username']) self.assertTrue(user.check_password(data['password'])) self.assertFalse(user.is_active) # Check verification e-mail. sent_email = sent_emails[0] self.assertEqual(sent_email.from_email, VERIFICATION_FROM_EMAIL) self.assertListEqual(sent_email.to, [data['email']]) url = self.assert_one_url_line_in_text(sent_email.body) verification_data = self.assert_valid_verification_url( url, expected_path=REGISTER_VERIFICATION_URL, expected_fields={'signature', 'user_id', 'timestamp'}, ) url_user_id = int(verification_data['user_id']) self.assertEqual(url_user_id, user_id) url_sig_timestamp = int(verification_data['timestamp']) self.assertGreaterEqual(url_sig_timestamp, timer.start_time) self.assertLessEqual(url_sig_timestamp, timer.end_time) signer = RegisterSigner(verification_data) signer.verify()
def assert_valid_register_verification_email(sent_email, user, timer): assert sent_email.from_email == VERIFICATION_FROM_EMAIL assert sent_email.to == [user.email] url = assert_one_url_line_in_text(sent_email.body) verification_data = assert_valid_verification_url( url, expected_path=REGISTER_VERIFICATION_URL, expected_fields={'signature', 'user_id', 'timestamp'}, ) url_user_id = int(verification_data['user_id']) assert url_user_id == user.pk url_sig_timestamp = int(verification_data['timestamp']) assert timer.start_time <= url_sig_timestamp <= timer.end_time signer = RegisterSigner(verification_data) signer.verify()