mirror of
https://github.com/TrentSPalmer/trentpalmerdotorg.git
synced 2024-10-31 18:18:45 -07:00
54 lines
1.7 KiB
Python
54 lines
1.7 KiB
Python
from django.test import SimpleTestCase
|
|
from django.urls import reverse, resolve
|
|
from accounts.views import log_out, password_change, register, edit_profile
|
|
from accounts.login import log_in, two_factor_input
|
|
from accounts.enable_totp import enable_totp, disable_totp
|
|
from accounts.urls import urlpatterns
|
|
|
|
|
|
class TestNumUrls(SimpleTestCase):
|
|
|
|
def test_num_urls(self):
|
|
self.assertEquals(len(urlpatterns), 12)
|
|
|
|
|
|
class TestViewsUrls(SimpleTestCase):
|
|
|
|
def test_password_change_url_is_resolved(self):
|
|
url = reverse('accounts:password_change')
|
|
self.assertEquals(resolve(url).func, password_change)
|
|
|
|
def test_register_url_is_resolved(self):
|
|
url = reverse('accounts:register')
|
|
self.assertEquals(resolve(url).func, register)
|
|
|
|
def test_logout_url_is_resolved(self):
|
|
url = reverse('accounts:logout')
|
|
self.assertEquals(resolve(url).func, log_out)
|
|
|
|
def test_edit_profile_url_is_resolved(self):
|
|
url = reverse('accounts:edit_profile')
|
|
self.assertEquals(resolve(url).func, edit_profile)
|
|
|
|
|
|
class TestLoginViewsUrls(SimpleTestCase):
|
|
|
|
def test_login_url_is_resolved(self):
|
|
url = reverse('accounts:login')
|
|
self.assertEquals(resolve(url).func, log_in)
|
|
|
|
def test_2fa_url_is_resolved(self):
|
|
url = reverse('accounts:two_factor_input')
|
|
self.assertEquals(resolve(url).func, two_factor_input)
|
|
|
|
|
|
class TestEnableTotpUrls(SimpleTestCase):
|
|
|
|
def test_enable_totp_url_is_resolved(self):
|
|
url = reverse('accounts:enable_totp')
|
|
self.assertEquals(resolve(url).func, enable_totp)
|
|
|
|
def test_disable_totp_url_is_resolved(self):
|
|
url = reverse('accounts:disable_totp')
|
|
self.assertEquals(resolve(url).func, disable_totp)
|