Robot Framework: отправка двоичных данных в теле запроса POST с - PullRequest
Новая
       21

Robot Framework: отправка двоичных данных в теле запроса POST с

0 голосов
/ 27 февраля 2019

У меня проблема с запуском теста с использованием Robot Framework и robotframework-запросы .Мне нужно отправить запрос POST и двоичные данные в теле.Я уже посмотрел на этот вопрос , но он не получил ответа.Вот как выглядит мой тестовый пример: 

Upload ${filename} file
    Create Session  mysession     http://${ADDRESS}
    ${data} =   Get Binary File     ${filename}
    &{headers} =    Create Dictionary   Content-Type=application/octet-stream   Accept=application/octet-stream
    ${resp} =   Post Request    mysession     ${CGIPath}  data=${data}    headers=&{headers}
    [Return]    ${resp.status_code}     ${resp.text}

Проблема в том, что мои двоичные данные занимают около 250 МБ.Когда данные читаются с Get Binary File, я вижу, что потребление памяти увеличивается до 2x ГБ.Несколько секунд спустя, когда сработал Post Request, мой тест был убит OOM.Я уже посмотрел на параметр files, но, похоже, он использует загрузку многочастного кодирования, а это не то, что мне нужно.

Другая моя мысль была о передаче обработчика открытого файла непосредственно в библиотеку базовых запросов, но я предполагаю, чтопотребует модификации запроса роботизированной конструкции.Другая идея - вернуться к скручиванию только для этого теста.

Я что-то упустил в своем тесте?Как лучше решить эту проблему?

1 Ответ

0 голосов
/ 28 февраля 2019

Я приступил к идее модификации запроса робота-фреймворка и добавил этот метод 

def post_request_binary(                                                                                          
        self,                                                                                                     
        alias,
        uri,
        path=None,
        params=None,
        headers=None,
        allow_redirects=None,                                                                                     
        timeout=None):      

    session = self._cache.switch(alias)                                                                           
    redir = True if allow_redirects is None else allow_redirects                                                  
    self._capture_output()  

    method_name = "post"    
    method = getattr(session, method_name)                                                                        

    with open(path, 'rb') as f:                                                                                   
        resp = method(self._get_url(session, uri),                                                                
                      data=f,                                                                                     
                      params=self._utf8_urlencode(params),                                                        
                      headers=headers,                                                                            
                      allow_redirects=allow_redirects,                                                            
                      timeout=self._get_timeout(timeout),                                                         
                      cookies=self.cookies,                                                                       
                      verify=self.verify)                                                                         

    self._print_debug()                                                                                           

    # Store the last session object                                                                               
    session.last_resp = resp

    self.builtin.log(method_name + ' response: ' + resp.text, 'DEBUG')                                            

    return resp

Полагаю, я могу немного его улучшить и создать запрос на извлечение.

...