def test_bad_url(self): self.assertFalse(is_safe_url('http://thatserver.com/junk', 'http://thisserver.com'))
def test_good_url(self): self.assertTrue(is_safe_url('/junk', 'http://thisserver.com')) self.assertTrue(is_safe_url('http://thisserver.com/junk', 'http://thisserver.com'))