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)