Cleaned up hard coded User model.

django_mesh/tests/test_managers.py

@@ -20,7 +20,6 @@
 # Test imports
 from .util import BaseTestCase
 
-from django.contrib.auth import get_user_model
 
 class PostQuerySetTestCase(BaseTestCase):
     def test_active(self):
@@ -267,9 +266,8 @@ def test_only_published_posts_shows_up(self):
         self.assertNotIn(self.t2, viewable)
 
     def test_tags_only_show_up_if_user_has_access_to_that_Channel(self):
+        user = self.user
 
-        new_user = get_user_model()
-        new_user.id = 1
         self.c1.save()
         self.c3.save() # user does not have access to this channel
 
@@ -285,7 +283,7 @@ def test_tags_only_show_up_if_user_has_access_to_that_Channel(self):
         self.p1.tags.add(self.t1)
         self.p2.tags.add(self.t2)
 
-        viewable = Tag.objects.get_for_user(new_user)
+        viewable = Tag.objects.get_for_user(user)
 
         self.assertIn(self.t1, viewable)
-        self.assertNotIn(self.t2, viewable)
+        self.assertNotIn(self.t2, viewable)

django_mesh/tests/util.py

@@ -20,7 +20,7 @@
 
 #Django imports
 from django.test import TestCase
-from django.contrib.auth.models import User
+from django.contrib.auth import get_user_model
 from django.contrib.sites.models import Site
 from django.core.cache import cache
 from django.utils import timezone
@@ -29,6 +29,10 @@
 from .. import models
 from ..models import Channel, Post, Tag
 
+
+User = get_user_model()
+
+
 class BaseTestCase(TestCase):
     def setUp(self):