def test_get_links_several(self): doc = '<html><a href="/1"></a><a href="/2"></a></html>' page = webvulnscan.page.Page("http://test/", doc, {}, 0) output = tutil.gen_to_set(page.get_links()) self.assertEqual(output, {"http://test/1", "http://test/2"})
def test_get_links_one_link(self): doc = '<html><a href="/test">click</a></html>' page = webvulnscan.page.Page("http://test/", doc, {}, 0) output = tutil.gen_to_set(page.get_links()) self.assertEqual(output, {"http://test/test"})
def test_get_links_no_links(self): doc = '<html>link</html>' page = webvulnscan.page.Page("http://test/", doc, {}, 0) output = tutil.gen_to_set(page.get_links()) self.assertEqual(output, set())