def test_domain_report_limit(self):
"""Проверка метода 'idomain_report' при превышении лмита запросов
(более 4-х запросов в минуту).
"""
vt_api = VirusTotalAPI(API_KEY)
time.sleep(TIME_DELAY)
response = vt_api.domain_report(TEST_DOMAIN)
response = vt_api.domain_report(TEST_DOMAIN)
response = vt_api.domain_report(TEST_DOMAIN)
response = vt_api.domain_report(TEST_DOMAIN)
response = vt_api.domain_report(TEST_DOMAIN)
time.sleep(TIME_DELAY)
self.assertEqual(response['error_code'], vt_api.ERROR_HTTP)
def test_domain_report_invalid_api_key(self):
"""Проверка метода 'domain_report' при неверном ключе доступа к API.
"""
vt_invalid_api_key = VirusTotalAPI()
time.sleep(TIME_DELAY)
response = vt_invalid_api_key.domain_report(TEST_DOMAIN)
time.sleep(TIME_DELAY)
self.assertEqual(response['error_code'], vt_invalid_api_key.ERROR_HTTP)
def test_domain_report_connection_error(self):
"""Проверка метода 'domain_report' при ошибке соединения с сервером.
"""
vt_api = VirusTotalAPI(API_KEY)
time.sleep(TIME_DELAY)
response = vt_api.domain_report(TEST_DOMAIN, None, TEST_PROXI)
time.sleep(TIME_DELAY)
self.assertEqual(response['error_code'], vt_api.ERROR_CONNECT)
def test_domain_report_correct_param(self):
"""Проверка метода 'domain_report' при корректных входных параметрах.
"""
vt_api = VirusTotalAPI(API_KEY)
time.sleep(TIME_DELAY)
response = vt_api.domain_report(TEST_DOMAIN)
time.sleep(TIME_DELAY)
self.assertEqual(response['error_code'], vt_api.ERROR_SUCCESS)
def test_domain_report_timeout(self):
"""Проверка метода 'domain_report' при превышении времени ожидания
ответа от сервера.
"""
vt_api = VirusTotalAPI(API_KEY)
time.sleep(TIME_DELAY)
response = vt_api.domain_report(TEST_DOMAIN, TEST_TIMEOUT)
time.sleep(TIME_DELAY)
self.assertEqual(response['error_code'], vt_api.ERROR_TIMEOUT)
