def setUp(self): url = ShortUrl() url.default_url = 'http://bbc.com' url.mobile_url = 'http://m.bbc.com' url.tablet_url = 'http://t.bbc.com' url.save() url = ShortUrl() url.default_url = 'http://google.com' url.save()
def test_str_representation(self): ShortUrl.objects.all().delete() url = ShortUrl() url.default_url = 'http://google.com' url.save() u = ShortUrl.objects.get(default_url='http://google.com') self.assertEqual(str(u), 'Short Url:3e9 Default Url:http://google.com')
def test_invalid_url_returns_error(self): with self.assertRaises(Exception) as context: ShortUrl.objects.all().delete() u = ShortUrl() u.default_url = 'http://googl' u.save() self.assertTrue('This is broken' in context.exception) self.assertEqual(ShortUrl.objects.all().count(), 0)