tests: accept view name with login() (#29162)

This commit is contained in:
Benjamin Dauvergne 2018-12-18 15:55:00 +01:00
parent 002dff7acf
commit a26417bb76
1 changed files with 5 additions and 1 deletions

View File

@ -9,6 +9,7 @@ from django.test import TestCase
from django.core.urlresolvers import reverse
from django.conf import settings
from django.utils.encoding import iri_to_uri
from django.shortcuts import resolve_url
from authentic2 import utils
@ -17,8 +18,11 @@ skipif_sqlite = pytest.mark.skipif('sqlite' in settings.DATABASES['default']['EN
reason='this test does not work with sqlite')
def login(app, user, path=None, password=None, remember_me=None):
def login(app, user, path=None, password=None, remember_me=None, args=None, kwargs=None):
if path:
args = args or []
kwargs = kwargs or {}
path = resolve_url(path, *args, **kwargs)
login_page = app.get(path, status=302).maybe_follow()
else:
login_page = app.get(reverse('auth_login'))