def test_file_report_limit(self):
"""Проверка метода 'file_report' при превышении лмита запросов
(более 4-х запросов в минуту).
"""
vt_api = VirusTotalAPI(API_KEY)
time.sleep(TIME_DELAY)
response = vt_api.file_report(TEST_HASH)
response = vt_api.file_report(TEST_HASH)
response = vt_api.file_report(TEST_HASH)
response = vt_api.file_report(TEST_HASH)
response = vt_api.file_report(TEST_HASH)
time.sleep(TIME_DELAY)
self.assertEqual(response['error_code'], vt_api.ERROR_HTTP)
def test_file_report_connection_error(self):
"""Проверка метода 'file_report' при ошибке соединения с сервером.
"""
vt_api = VirusTotalAPI(API_KEY)
time.sleep(TIME_DELAY)
response = vt_api.file_report(TEST_HASH, None, TEST_PROXI)
time.sleep(TIME_DELAY)
self.assertEqual(response['error_code'], vt_api.ERROR_CONNECT)
def test_file_report_correct_param(self):
"""Проверка метода 'file_report' с корректными входными параметрами.
"""
vt_api = VirusTotalAPI(API_KEY)
time.sleep(TIME_DELAY)
response = vt_api.file_report(TEST_HASH)
time.sleep(TIME_DELAY)
self.assertEqual(response['error_code'], vt_api.ERROR_SUCCESS)
def test_file_report_timeout(self):
"""Проверка метода 'file_report' при превышении времени ожидании
ответа от сервера.
"""
vt_api = VirusTotalAPI(API_KEY)
time.sleep(TIME_DELAY)
response = vt_api.file_report(TEST_HASH, TEST_TIMEOUT)
time.sleep(TIME_DELAY)
self.assertEqual(response['error_code'], vt_api.ERROR_TIMEOUT)
def test_file_report_invalid_api_key(self):
"""Проверка метода 'file_report' с использованием неверного
ключа доступа к API.
"""
vt_invalid_api_key = VirusTotalAPI()
time.sleep(TIME_DELAY)
response = vt_invalid_api_key.file_report(TEST_HASH)
time.sleep(TIME_DELAY)
self.assertEqual(response['error_code'], vt_invalid_api_key.ERROR_HTTP)
