Skip to content

Commit

Permalink
To allow tests to be run in context with a custom user model, call ge…
Browse files Browse the repository at this point in the history
…t_user_model in tests. (#90)

Co-authored-by: Trevor Cox <trevor.cox@appazur.com>
  • Loading branch information
trevoriancox and Trevor Cox authored Aug 5, 2020
1 parent ce5a2b9 commit 347f0ed
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion shibboleth/tests/test_shib.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,12 +5,14 @@
from django.contrib import auth
from django.contrib.auth.middleware import AuthenticationMiddleware
from django.contrib.auth.middleware import RemoteUserMiddleware
from django.contrib.auth.models import User, Group
from django.contrib.auth.models import Group
from django.contrib.auth import get_user_model
from django.contrib.sessions.middleware import SessionMiddleware
from django.db.utils import IntegrityError
from django.test import TestCase, RequestFactory
from django import VERSION

User = get_user_model()

SAMPLE_HEADERS = {
"REMOTE_USER": 'sampledeveloper@school.edu',
Expand Down

0 comments on commit 347f0ed

Please sign in to comment.