def test_missing_host_settings(self):
host_settings = {
'schemes': ['udp', 'http'],
'ports': [443, 8080, 1234]
}
expected_urls = []
result_urls = get_urls(host_settings)
self.assertEqual(result_urls, sorted(expected_urls))
def test_missing_ports_settings(self):
host_settings = {
'schemes': ['ftp', 'http', 'https'],
'host': 'facebook.com'
}
expected_urls = [
'https://facebook.com', 'ftp://facebook.com', 'http://facebook.com'
]
result_urls = sorted(get_urls(host_settings))
self.assertEqual(result_urls, sorted(expected_urls))
def test_http(self):
host_settings = {
'host': 'google.com',
'schemes': ['http', 'https'],
'ports': [8080, 9090],
}
expected_urls = [
'http://google.com:8080',
'http://google.com:9090',
'http://google.com',
'https://google.com',
]
result_urls = sorted(get_urls(host_settings))
self.assertEqual(result_urls, sorted(expected_urls))
def test_missing_schemes_settings(self):
host_settings = {
'host': 'yandex.com',
'ports': [443, 8080, 1234, 5555, 80]
}
expected_urls = [
'http://yandex.com:1234',
'http://yandex.com:8080',
'http://yandex.com:5555',
'http://yandex.com',
'https://yandex.com',
]
result_urls = sorted(get_urls(host_settings))
self.assertEqual(result_urls, sorted(expected_urls))
def test_filled_in_settings(self):
host_settings = {
'host': 'google.com',
'schemes': ['udp', 'http', 'https'],
'ports': [443, 8080, 1234]
}
expected_urls = [
'http://google.com:1234',
'http://google.com:8080',
'https://google.com',
'udp://google.com',
'http://google.com',
]
result_urls = sorted(get_urls(host_settings))
self.assertEqual(result_urls, sorted(expected_urls))
def test_additional_host_settings(self):
host_settings = {
'host': 'apple.com',
'schemes': ['udp', 'http', 'https'],
'ports': [443, 8080, 1234],
'ips': ['198.120.0.1', '0.0.0.0']
}
expected_urls = [
'http://apple.com:1234',
'http://apple.com:8080',
'https://apple.com',
'udp://apple.com',
'http://apple.com',
]
result_urls = sorted(get_urls(host_settings))
self.assertEqual(result_urls, sorted(expected_urls))
def get_failed_servers(servers):
for friendly_name in servers:
hostname = servers[friendly_name].get('host')
if not hostname:
continue
for url in get_urls(servers[friendly_name]):
try:
response = requests.get(url, timeout=5)
if response.status_code >= 400:
yield (url, response.status_code)
except requests.exceptions.RequestException as exc:
# dbc.increase_uptime_table(url, percentage)
yield (url, exc.args[0])
except Exception as exc:
yield (url, exc.args[0])
def test_empty_host_settings(self):
host_settings = {}
expected_urls = []
result_urls = sorted(get_urls(host_settings))
self.assertEqual(result_urls, sorted(expected_urls))