def test_shortening_via_model(self): lsi = LinkShortenedItem() lsi.content_object = self.test_a lsi.save() self.assertUrlsEqual('http://example.com/a/', lsi.shortened_url) self.assertEqual(lsi.url(), lsi.shortened_url)
def test_shortening_no_abs_url(self): no_absolute_url = Site.objects.get_current() self.assertFalse(hasattr(no_absolute_url, 'get_absolute_url')) lsi = LinkShortenedItem() lsi.content_object = no_absolute_url lsi.save() self.assertEqual(lsi.shortened_url, '') self.assertEqual(lsi.url(), '')