def test_get_emails_filter(self): resp = build_http_response(self.url, '') p = SGMLParser(resp) p._emails = {'*****@*****.**', '*****@*****.**'} self.assertEqual(p.get_emails(), {'*****@*****.**', '*****@*****.**'}) self.assertEqual(p.get_emails(domain='w3af.com'), ['*****@*****.**']) self.assertEqual(p.get_emails(domain='not.com'), ['*****@*****.**'])
def test_extract_emails_mailto(self): body = u'<a href="mailto:[email protected]">test</a>' resp = build_http_response(self.url, body) p = SGMLParser(resp) p.parse() expected_res = {u'*****@*****.**'} self.assertEqual(p.get_emails(), expected_res)
def test_mailto_subject_body(self): body = u'<a href="mailto:[email protected]?subject=testing out mailto'\ u'&body=Just testing">test</a>' resp = build_http_response(self.url, body) p = SGMLParser(resp) p.parse() expected_res = {u'*****@*****.**'} self.assertEqual(p.get_emails(), expected_res)
def test_extract_emails_blank(self): resp = build_http_response(self.url, '') p = SGMLParser(resp) self.assertEqual(p.get_emails(), set())