def activate(request, uidb64, token):
"""Set user activation status to true."""
user = UserCreationForm().get_user(uidb64)
if user is not None and account_activation_token.check_token(user, token):
user.is_active = True
user.save()
return redirect('/dashboard')
return render(request, 'registration/signup_activation_invalid.html')
def test_get_user_wrong_value(self):
'''
Test that when you provide the wrong value to the get_user method:
- That a None object is returned.
'''
user_get = UserCreationForm().get_user("example")
self.assertEqual(user_get, None)
def activate_new_email(request, uidb64, token):
"""
Activate the new email provided by the user and update the database.
"""
user = UserCreationForm().get_user(uidb64)
if user is not None and email_activation_token.check_token(user, token):
user.former_email = user.email
user.email = user.change_email
user.change_email = None
user.is_change_allowed = True
user.change_email_tracker = None
user.save()
return redirect('/dashboard')
return render(request, 'registration/signup_activation_invalid.html')
def signup_view(request):
"""
Register a new user to the system.
"""
(phone_categories, social_media) = various_caches()
if request.method == "POST":
form = UserCreationForm(request.POST)
if form.is_valid():
user = form.save(commit=False)
user.is_active = False
user.save()
form.send_email(request, user)
args = {
'user_email': user.email,
'provider': get_user_email_provider(user.email)
}
return render(request, 'registration/signup_email_sent.html', args)
args = {
'form': form,
'social_media': social_media,
'categories': phone_categories
}
return render(request, 'front/signup.html', args)
else:
args = {
'form': UserCreationForm(),
'social_media': social_media,
'categories': phone_categories
}
return render(request, 'front/signup.html', args)
def test_get_user_not_exist(self):
'''
Test that when a user is provided that does not exist to teh
get_user method:
- That a None object is returned.
'''
uid = urlsafe_base64_encode(force_bytes(1111111)).decode()
user_get = UserCreationForm().get_user(uid)
self.assertEqual(user_get, None)
def test_get_user(self):
'''
Test that when you provide a uid for a user that exists to the
get_user method:
- That the correct user is returned.
'''
country_code_k = CountryCode.objects.get(country="Kenya")
User.objects.create_user(email="*****@*****.**",
first_name="Test",
last_name="User_2",
country_code=country_code_k,
phone_number=72200000)
user = User.objects.get(email="*****@*****.**")
uid = urlsafe_base64_encode(force_bytes(user.pk)).decode()
user_get = UserCreationForm().get_user(uid)
self.assertEqual(user_get.email, "*****@*****.**")
def test_send_email(self):
'''
Test that the send_email method when given data to send:
- That it send the data it is expected to send to the recepient.
Test that when you click the activation link sent once and twice:
- That on the first click it redirects you to a login page.
- That on the second click it informs you that the activation
link is invalid
'''
country_code_k = CountryCode.objects.get(country="Kenya")
user_data = {
"email": "*****@*****.**",
"first_name": "Test",
"last_name": "User",
"country_code": country_code_k.pk,
"phone_number": 722000000,
"password1": "*&#@&!*($)lp",
"password2": "*&#@&!*($)lp"
}
request = self.factory.post('/signup', data=user_data)
form = UserCreationForm(request.POST)
self.assertEqual(form.is_valid(), True)
user = form.save(commit=False)
user.is_active = False
user.save()
self.assertEqual(str(user), "Test User")
current_site = get_current_site(request)
context = {
'user': user,
'domain': current_site.domain,
'uid': urlsafe_base64_encode(force_bytes(user.pk)).decode(),
'token': account_activation_token.make_token(user),
'protocol': 'https' if request.is_secure() else 'http',
}
subject = loader.render_to_string(
"registration/signup_activation_subject.txt", context)
subject = ''.join(subject.splitlines())
body = loader.render_to_string(
"registration/signup_activation_email.html", context)
form.send_email(request, user)
self.assertEqual(len(mail.outbox), 1)
self.assertEqual(mail.outbox[0].to, ['*****@*****.**'])
self.assertEqual(mail.outbox[0].subject, subject)
self.assertEqual(mail.outbox[0].from_email,
settings.DEFAULT_FROM_EMAIL)
self.assertEqual(mail.outbox[0].body, body)
email = loader.render_to_string("test/test_email.html", context)
response = self.client.get(email, follow=True)
self.assertRedirects(response, "/login?next=/dashboard", 302)
response_2 = self.client.get(email, follow=True)
self.assertContains(response_2, "The activation link is invalid!")
self.assertEqual(response.status_code, 200)
def test_resend_email(self):
'''
Test that the resend_email method when given data to send:
- That it send the data it is expected to send to the recepient.
'''
country_code_k = CountryCode.objects.get(country="Kenya")
user_data = {
"email": "*****@*****.**",
"first_name": "Test",
"last_name": "User",
"country_code": country_code_k.pk,
"phone_number": 722000000,
"password1": "*&#@&!*($)lp",
"password2": "*&#@&!*($)lp"
}
request = self.factory.post('/signup', data=user_data)
form = UserCreationForm(request.POST)
self.assertEqual(form.is_valid(), True)
user = form.save(commit=False)
user.is_active = False
user.save()
self.assertEqual(str(user), "Test User")
current_site = get_current_site(request)
context = {
'user': user,
'domain': current_site.domain,
'uid': urlsafe_base64_encode(force_bytes(user.pk)).decode(),
'token': account_activation_token.make_token(user),
'protocol': 'https' if request.is_secure() else 'http',
}
subject = loader.render_to_string(
"registration/signup_activation_subject.txt", context)
subject = ''.join(subject.splitlines())
form.send_email(request, user)
self.assertEqual(len(mail.outbox), 1)
self.assertEqual(mail.outbox[0].to, ['*****@*****.**'])
resend_email(request, user, user_data['email'])
self.assertEqual(len(mail.outbox), 2)
self.assertEqual(mail.outbox[0].to, ['*****@*****.**'])