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())
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())
