def test_ip_report_limit(self):
"""Проверка метода 'ip_report' при превышении лмита запросов
(более 4-х запросов в минуту).
"""
vt_api = VirusTotalAPI(API_KEY)
time.sleep(TIME_DELAY)
response = vt_api.ip_report(TEST_IP)
response = vt_api.ip_report(TEST_IP)
response = vt_api.ip_report(TEST_IP)
response = vt_api.ip_report(TEST_IP)
response = vt_api.ip_report(TEST_IP)
time.sleep(TIME_DELAY)
self.assertEqual(response['error_code'], vt_api.ERROR_HTTP)
def test_ip_report_incorrect_param(self):
"""Проверка метода 'url_report' при некорректных входных параметрах.
"""
vt_api = VirusTotalAPI(API_KEY)
time.sleep(TIME_DELAY)
response = vt_api.ip_report('This is an invalid IP value')
time.sleep(TIME_DELAY)
self.assertEqual(response['error_code'], vt_api.ERROR_SUCCESS)
def test_ip_report_invalid_api_key(self):
"""Проверка метода 'ip_report' при неверном ключе доступа к API.
"""
vt_invalid_api_key = VirusTotalAPI()
time.sleep(TIME_DELAY)
response = vt_invalid_api_key.ip_report(TEST_IP)
time.sleep(TIME_DELAY)
self.assertEqual(response['error_code'], vt_invalid_api_key.ERROR_HTTP)
def test_ip_report_connect_error(self):
"""Проверка метода 'ip_report' при ошибке соединения с сервером.
"""
vt_api = VirusTotalAPI(API_KEY)
time.sleep(TIME_DELAY)
response = vt_api.ip_report(TEST_IP, None, TEST_PROXI)
time.sleep(TIME_DELAY)
self.assertEqual(response['error_code'], vt_api.ERROR_CONNECT)
def test_ip_report_correct_param(self):
"""Проверка метода 'ip_report' при корректных входных параметрах.
"""
vt_api = VirusTotalAPI(API_KEY)
time.sleep(TIME_DELAY)
response = vt_api.ip_report(TEST_IP)
time.sleep(TIME_DELAY)
self.assertEqual(response['error_code'], vt_api.ERROR_SUCCESS)
def test_ip_report_timeout(self):
"""Проверка метода 'ip_report' при превышении времени ожидания
ответа от сервера.
"""
vt_api = VirusTotalAPI(API_KEY)
time.sleep(TIME_DELAY)
response = vt_api.ip_report(TEST_IP, TEST_TIMEOUT)
time.sleep(TIME_DELAY)
self.assertEqual(response['error_code'], vt_api.ERROR_TIMEOUT)
