fixed up unit test

dgrant · Mar 11, 2014 · 2faa81a · 2faa81a
1 parent c21ac6e
commit 2faa81a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/links/tests/test_models.py b/links/tests/test_models.py
@@ -138,9 +138,9 @@ def test_owned_by(self):
 
 class LinkTest(TestCase):
     def test_unicode(self):
-        link = mommy.make('Link')
-        self.assertEquals(unicode(link), link.title)
-
+        url = mommy.make('Url')
+        link = mommy.make('Link', url=url)
+        self.assertEquals(unicode(link), link.title + ', ' + unicode(url))
 
 class MagicTokenTest(TestCase):
     def test_create(self):