# Create your tests here. # Copyright © 2022 Aravinth Manivannan # # This program is free software: you can redistribute it and/or modify # it under the terms of the GNU Affero General Public License as # published by the Free Software Foundation, either version 3 of the # License, or (at your option) any later version. # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU Affero General Public License for more details. # # You should have received a copy of the GNU Affero General Public License # along with this program. If not, see . from django.contrib.auth import get_user_model from django.urls import reverse from django.test import TestCase, Client, override_settings from django.contrib.auth import authenticate from .models import AccountConfirmChallenge class LoginTest(TestCase): """ Tests create new app view """ def setUp(self): self.password = "password121231" self.username = "create_new_app_tests" self.email = f"{self.username}@example.org" self.user = get_user_model().objects.create( username=self.username, email=self.email, ) self.user.set_password(self.password) self.user.save() def test_login_template_works(self): """ Tests if login template renders """ resp = self.client.get(reverse("accounts.login")) self.assertEqual(b"Free Forge Ecosystem" in resp.content, True) def test_login_works(self): """ Tests if login template renders """ c = Client() # username login works payload = { "login": self.username, "password": self.password, } resp = c.post(reverse("accounts.login"), payload) self.assertEqual(resp.status_code, 302) self.assertEqual(resp.headers["location"], reverse("accounts.home")) # email login works paylaod = { "login": self.email, "password": self.password, } resp = c.post(reverse("accounts.login"), payload) self.assertEqual(resp.status_code, 302) self.assertEqual(resp.headers["location"], reverse("accounts.home")) # authentication failure when wrong credentials are provided payload = { "login": self.email, "password": self.user.email, } resp = self.client.post(reverse("accounts.login"), paylaod) self.assertEqual(resp.status_code, 401) self.assertEqual(b"Login Failed" in resp.content, True) class RegistrationTest(TestCase): def test_register_template_works(self): """ Tests if register template renders """ resp = self.client.get(reverse("accounts.register")) self.assertEqual(b"Free Forge Ecosystem" in resp.content, True) def test_register_works(self): """ Tests if register works """ c = Client() # passwords don't match msg = { "username": "register_user", "password": "password", "email": "register_user@example.com", "confirm_password": "foo@example.com", } resp = c.post(reverse("accounts.register"), msg) self.assertEqual(resp.status_code, 400) # register user msg["confirm_password"] = msg["password"] resp = c.post(reverse("accounts.register"), msg) self.assertEqual(resp.status_code, 302) user = get_user_model().objects.get(username=msg["username"]) self.assertEqual(user.is_active, False) challenge = AccountConfirmChallenge.objects.get(owned_by=user) pending_url = challenge.pending_url() self.assertEqual(resp.headers["location"], pending_url) resp = c.post(reverse("accounts.verify.resend", args=(challenge.public_ref,))) self.assertEqual(resp.status_code, 302) self.assertEqual(resp.headers["location"], pending_url) resp = c.post(challenge.verification_link()) self.assertEqual(resp.status_code, 302) self.assertEqual(resp.headers["location"], reverse("accounts.login")) user.refresh_from_db() self.assertEqual(user.is_active, True) old_email = msg["email"] # duplicate username msg["email"] = "new_register_user_email@example.com" resp = c.post(reverse("accounts.register"), msg) self.assertEqual(resp.status_code, 400) self.assertEqual(b"Username is already registered" in resp.content, True) # duplicate email msg["username"] = "new_register_user_email" msg["email"] = old_email resp = c.post(reverse("accounts.register"), msg) self.assertEqual(resp.status_code, 400) self.assertEqual(b"Email is already registered" in resp.content, True)