diff --git a/tests/utils.py b/tests/utils.py index 5f3a5d633..277edf318 100644 --- a/tests/utils.py +++ b/tests/utils.py @@ -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'))