def test_url_scan_limit(self):
"""Проверка метода 'url_scan' при превышении лмита запросов
(более 4-х запросов в минуту).
"""
vt_api = VirusTotalAPI(API_KEY)
time.sleep(TIME_DELAY)
response = vt_api.url_scan(TEST_URL)
response = vt_api.url_scan(TEST_URL)
response = vt_api.url_scan(TEST_URL)
response = vt_api.url_scan(TEST_URL)
response = vt_api.url_scan(TEST_URL)
time.sleep(TIME_DELAY)
self.assertEqual(response['error_code'], vt_api.ERROR_HTTP)
def test_url_scan_incorrect_param(self):
"""Проверка метода 'url_scan' при некорректных входных параметрах.
"""
vt_api = VirusTotalAPI(API_KEY)
time.sleep(TIME_DELAY)
response = vt_api.url_scan('This is an invalid URL value')
time.sleep(TIME_DELAY)
self.assertEqual(response['error_code'], vt_api.ERROR_SUCCESS)
def test_url_scan_invalid_api_key(self):
"""Проверка метода 'url_scan' при неверном ключе доступа к API.
"""
vt_invalid_api_key = VirusTotalAPI()
time.sleep(TIME_DELAY)
response = vt_invalid_api_key.url_scan(TEST_URL)
time.sleep(TIME_DELAY)
self.assertEqual(response['error_code'], vt_invalid_api_key.ERROR_HTTP)
def test_url_scan_connect_error(self):
"""Проверка метода 'url_scan' при ошибке соединения с сервером.
"""
vt_api = VirusTotalAPI(API_KEY)
time.sleep(TIME_DELAY)
response = vt_api.url_scan(TEST_URL, None, TEST_PROXI)
time.sleep(TIME_DELAY)
self.assertEqual(response['error_code'], vt_api.ERROR_CONNECT)
def test_url_scan_timeout(self):
"""Проверка метода 'url_scan' при превышении времени ожидания
ответа от сервера.
"""
vt_api = VirusTotalAPI(API_KEY)
time.sleep(TIME_DELAY)
response = vt_api.url_scan(TEST_URL, TEST_TIMEOUT)
time.sleep(TIME_DELAY)
self.assertEqual(response['error_code'], vt_api.ERROR_TIMEOUT)
def test_url_scan_correct_param_list(self):
"""Проверка метода 'url_scan' при корректных входных параметрах
в виде списка из 4-х URL-адресов.
"""
vt_api = VirusTotalAPI(API_KEY)
time.sleep(TIME_DELAY)
response = vt_api.url_scan(TEST_URL_LIST)
time.sleep(TIME_DELAY)
self.assertEqual(response['error_code'], vt_api.ERROR_SUCCESS)