def test_mailto(self): self.assertIsNone(main.clean_link("mailto:[email protected]"))
def test_absolute_link(self): self.assertEqual(main.clean_link("https://test.com/testing_link", "https://test.com/"), "https://test.com/testing_link") self.assertEqual(main.clean_link("http://test.com/testing_link", "http://test.com/"), "http://test.com/testing_link")
def test_none(self): self.assertIsNone(main.clean_link(None))
def test_relative_link(self): valid_result = "https://test.com/testing_link" res = main.clean_link("/testing_link", "https://test.com/") self.assertEqual(res, valid_result)