Free and open source ticket system written in python
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129from django.test import TestCasefrom django.contrib.auth.models import Groupfrom django.urls import reversefrom .utils.initial_data import populate_groupsfrom .utils.general import sanitize_usernamefrom .models import PawUserfrom django.conf import settings
class PopulateGroupTestCase(TestCase): def setUp(self): populate_groups(None, None)
def test_groups_created(self): self.assertEqual(Group.objects.filter(name__in=["Client", "Supporter"]).count(), 2) # Run again and confirm no duplicates populate_groups(None, None) self.assertEqual(Group.objects.filter(name__in=["Client", "Supporter"]).count(), 2)
class UsernameSanitizationTestCase(TestCase): def test_allows_safe_chars(self): self.assertEqual(sanitize_username("Alice-01_@corp"), "Alice-01_@corp")
def test_strips_spaces_and_symbols(self): self.assertEqual(sanitize_username(" alice.smith "), "alicesmith") self.assertEqual(sanitize_username("a/b\\c"), "abc") self.assertEqual(sanitize_username("a%b^c&d*e"), "abcde")
def test_empty_and_only_illegal(self): self.assertEqual(sanitize_username(""), "") self.assertEqual(sanitize_username("!#$()., +="), "")
def test_unicode_behavior(self): # Decide policy: currently regex drops non ASCII letters. self.assertEqual(sanitize_username("álïçé"), "l")
class LoginViewTestCase(TestCase): def setUp(self): self.username = "testuser" self.password = "testpassword" self.user = PawUser.objects.create_user(username=self.username, password=self.password)
def test_login_view(self): url = reverse("login") response = self.client.post(url, {"username": self.username, "password": self.password}) self.assertEqual(response.status_code, 302) self.assertEqual(response.url, reverse("home"))
def test_login_view_invalid(self): url = reverse("login") response = self.client.post(url, {"username": self.username, "password": "invalid"}) self.assertEqual(response.status_code, 200) self.assertContains(response, "Please enter a correct username and password") def test_user_language(self): url = reverse("home") self.client.force_login(self.user) response = self.client.get(url) self.assertEqual(response.status_code, 302) self.client.cookies.load(response.cookies) self.assertEqual(self.client.cookies[settings.LANGUAGE_COOKIE_NAME].value, self.user.language) self.assertEqual(response.url, reverse("all_tickets")) def test_login_sets_session(self): resp = self.client.post(reverse("login"), {"username": self.username, "password": self.password}) self.assertEqual(resp.status_code, 302) # Access a protected view to ensure session auth holds resp2 = self.client.get(reverse("home")) self.assertIn("_auth_user_id", self.client.session)
class RegisterViewTestCase(TestCase):
def test_register_view(self): url = reverse("register") response = self.client.post(url, {"username": "test", "email": "test@example.com", "password": "testtesttesttest", "password_confirm": "testtesttesttest"}) self.assertEqual(response.status_code, 302) self.assertEqual(response.url, reverse("login")) self.assertEqual(PawUser.objects.count(), 1) user = PawUser.objects.first() self.assertEqual(user.username, "test") self.assertEqual(user.email, "test@example.com") self.assertTrue(user.check_password("testtesttesttest")) self.assertEqual(user.groups.count(), 0) # No group assigned, might want to give Client group by default
def test_register_view_password_too_short(self): url = reverse("register") response = self.client.post(url, {"username": "test", "email": "test@example.com", "password": "123456789", "password_confirm": "123456789"}) self.assertEqual(response.status_code, 200) self.assertContains(response, "Password must be at least 10 characters long.") def test_register_view_password_mismatch(self): url = reverse("register") response = self.client.post(url, {"username": "test", "email": "test@example.com", "password": "1234567890", "password_confirm": "123456789"}) self.assertEqual(response.status_code, 200) self.assertContains(response, "Password and Confirm Password do not match.")
class SettingsViewTestCase(TestCase):
def test_settings_view(self): url = reverse("settings") response = self.client.get(url) self.assertEqual(response.status_code, 302) self.assertEqual(response.url, "%s?next=%s" % (reverse("login"), reverse("settings")))
user = PawUser.objects.create_user(username="test", password="testtesttesttest", email="test@example.com") self.client.force_login(user) response = self.client.get(url) self.assertEqual(response.status_code, 200) self.assertContains(response, "Settings")
response = self.client.post(url, {"language": "fr", "email": "test@example.com"}) self.assertEqual(response.status_code, 200) user.refresh_from_db() self.assertEqual(user.language, "fr") self.client.cookies.load(response.cookies) self.assertEqual(self.client.cookies[settings.LANGUAGE_COOKIE_NAME].value, "fr")
response = self.client.post(url, {"language": "invalid", "email": "test@example.com"}) self.assertEqual(response.status_code, 200) self.client.cookies.load(response.cookies) self.assertEqual(self.client.cookies[settings.LANGUAGE_COOKIE_NAME].value, "fr")
response = self.client.post(url, {"language": "en", "email": "test@example.com"}) self.assertEqual(response.status_code, 200) user.refresh_from_db() self.assertEqual(user.language, "en") self.client.cookies.load(response.cookies) self.assertEqual(self.client.cookies[settings.LANGUAGE_COOKIE_NAME].value, "en")