def test_file_scan_invalid_api_key(self):
"""Проверка метода 'file_scan' при неверном ключе доступа к API.
"""
vt_invalid_api_key = VirusTotalAPI()
time.sleep(TIME_DELAY)
response = vt_invalid_api_key.file_scan(TEST_FILE_PATH)
time.sleep(TIME_DELAY)
self.assertEqual(response['error_code'], vt_invalid_api_key.ERROR_HTTP)
def test_file_scan_connect_error(self):
"""Проверка метода 'file_scan' при ошибке соединения с сервером.
"""
vt_api = VirusTotalAPI(API_KEY)
time.sleep(TIME_DELAY)
response = vt_api.file_scan(TEST_FILE_PATH, None, TEST_PROXI)
time.sleep(TIME_DELAY)
self.assertEqual(response['error_code'], vt_api.ERROR_CONNECT)
def test_file_scan_file_name_error(self):
"""Проверка метода 'file_scan' при некорректных входных параметрах
(невалидный путь к сканируемому файлу).
"""
vt_api = VirusTotalAPI(API_KEY)
time.sleep(TIME_DELAY)
response = vt_api.file_scan(' ')
time.sleep(TIME_DELAY)
self.assertEqual(response['error_code'], vt_api.ERROR_FILE)
def test_file_scan_file_size_error(self):
"""Проверка метода 'file_scan' при отправке файла размером
более 32 MB.
"""
vt_api = VirusTotalAPI(API_KEY)
time.sleep(TIME_DELAY)
response = vt_api.file_scan('d:/test_file.zip') ## файл более 32 MB
time.sleep(TIME_DELAY)
self.assertEqual(response['error_code'], vt_api.ERROR_HTTP)
def test_file_scan_timeout(self):
"""Проверка метода 'file_scan' при превышении времени ожидания ответа
от сервера.
"""
vt_api = VirusTotalAPI(API_KEY)
time.sleep(TIME_DELAY)
response = vt_api.file_scan(TEST_FILE_PATH, TEST_TIMEOUT)
time.sleep(TIME_DELAY)
self.assertEqual(response['error_code'], vt_api.ERROR_TIMEOUT)
def test_file_scan_correct_param(self):
"""Проверка метода 'file_scan' при корректных входных параметрах
(валидный путь к сканируемому файлу).
"""
vt_api = VirusTotalAPI(API_KEY)
time.sleep(TIME_DELAY)
response = vt_api.file_scan(TEST_FILE_PATH)
time.sleep(TIME_DELAY)
self.assertEqual(response['error_code'], vt_api.ERROR_SUCCESS)
