def test_print_subdomain_result_both(self): url = 'example.com' ip = '192.168.1.0' http_connection_headers = {'HTTP HEADER': 'value'} nearby = {'192.168.1.1': 'nearby.com'} with io.StringIO() as stream: fierce.print_subdomain_result( url, ip, http_connection_headers=http_connection_headers, nearby=nearby, stream=stream ) result = stream.getvalue() expected = textwrap.dedent(''' Found: example.com (192.168.1.0) HTTP connected: {'HTTP HEADER': 'value'} Nearby: {'192.168.1.1': 'nearby.com'} ''').lstrip() self.assertEqual(expected, result)
def test_print_subdomain_result_basic(self): url = 'example.com' ip = '192.168.1.0' with io.StringIO() as stream: fierce.print_subdomain_result(url, ip, stream=stream) result = stream.getvalue() expected = 'Found: example.com (192.168.1.0)\n' self.assertEqual(expected, result)
def test_print_subdomain_result_nearby(self): url = 'example.com' ip = '192.168.1.0' nearby = {'192.168.1.1': 'nearby.com'} with io.StringIO() as stream: fierce.print_subdomain_result(url, ip, nearby=nearby, stream=stream) result = stream.getvalue() expected = textwrap.dedent(''' Found: example.com (192.168.1.0) Nearby: {'192.168.1.1': 'nearby.com'} ''').lstrip() self.assertEqual(expected, result)
def test_print_subdomain_result_http_header(self): url = 'example.com' ip = '192.168.1.0' http_connection_headers = {'HTTP HEADER': 'value'} with io.StringIO() as stream: fierce.print_subdomain_result( url, ip, http_connection_headers=http_connection_headers, stream=stream) result = stream.getvalue() expected = textwrap.dedent(''' Found: example.com (192.168.1.0) HTTP connected: {'HTTP HEADER': 'value'} ''').lstrip() assert expected == result