def test_url_report_limit(self):
"""Проверка метода 'url_report' при превышении лмита запросов
(более 4-х запросов в минуту).
"""
vt_api = VirusTotalAPI(API_KEY)
time.sleep(TIME_DELAY)
response = vt_api.url_report(TEST_URL)
response = vt_api.url_report(TEST_URL)
response = vt_api.url_report(TEST_URL)
response = vt_api.url_report(TEST_URL)
response = vt_api.url_report(TEST_URL)
time.sleep(TIME_DELAY)
self.assertEqual(response['error_code'], vt_api.ERROR_HTTP)
def test_url_report_connect_error(self):
"""Проверка метода 'url_report' при ошибке соединения с сервером.
"""
vt_api = VirusTotalAPI(API_KEY)
time.sleep(TIME_DELAY)
response = vt_api.url_report(TEST_URL, 0, None, TEST_PROXI)
time.sleep(TIME_DELAY)
self.assertEqual(response['error_code'], vt_api.ERROR_CONNECT)
def test_url_report_incorrect_param(self):
"""Проверка метода 'url_report' при некорректных входных параметрах.
"""
vt_api = VirusTotalAPI(API_KEY)
time.sleep(TIME_DELAY)
response = vt_api.url_report('This is an invalid URL value')
time.sleep(TIME_DELAY)
self.assertEqual(response['error_code'], vt_api.ERROR_SUCCESS)
def test_url_report_invalid_api_key(self):
"""Проверка метода 'url_report' при неверном ключе доступа к API.
"""
vt_api_invalid_api_key = VirusTotalAPI()
time.sleep(TIME_DELAY)
response = vt_api_invalid_api_key.url_report(TEST_URL)
time.sleep(TIME_DELAY)
self.assertEqual(response['error_code'],
vt_api_invalid_api_key.ERROR_HTTP)
def test_url_report_timeout(self):
"""Проверка метода 'url_report' при превышении времени ожидания от
сервера.
"""
vt_api = VirusTotalAPI(API_KEY)
time.sleep(TIME_DELAY)
response = vt_api.url_report(TEST_URL, 0, TEST_TIMEOUT)
time.sleep(TIME_DELAY)
self.assertEqual(response['error_code'], vt_api.ERROR_TIMEOUT)
def test_url_report_correct_param_list(self):
"""Проверка метода 'url_report' при корректных входных параметрах
в виде списка из 4-х URL-адресов.
"""
vt_api = VirusTotalAPI(API_KEY)
time.sleep(TIME_DELAY)
response = vt_api.url_report(TEST_URL_LIST)
time.sleep(TIME_DELAY)
self.assertEqual(response['error_code'], vt_api.ERROR_SUCCESS)
def test_url_report_correct_param_scan(self):
"""Проверка метода 'url_report' при корректных входных параметрах
с использованием параметра 'scan'.
"""
vt_api = VirusTotalAPI(API_KEY)
time.sleep(TIME_DELAY)
response = vt_api.url_report(TEST_URL, '1')
time.sleep(TIME_DELAY)
self.assertEqual(response['error_code'], vt_api.ERROR_SUCCESS)
from virustotalapi import VirusTotalAPI
vt = VirusTotalAPI(
'1a9ab77640f8f5a1c955319c646f0f6138c110cce550676f8f7fdd9a9dd4c571')
print vt.url_report("https://askldaskljdlaskjd.com/test/est")
