def test_check_is_safe_false(self):
PhishUrl.objects.create(phish_id=1, url="http://facebook.com")
short = Shortening(url="http://facebook.com")
short.full_clean()
short.save()
self.assertFalse(short.check_is_safe())
def test_check_is_safe_true(self):
PhishUrl.objects.create(phish_id=1, url="http://is-safe-true.com")
short = Shortening(url="http://opendns.com")
short.full_clean()
short.save()
self.assertTrue(short.check_is_safe())