Как протестировать приложение python / flask со сторонним http lib? - PullRequest
Новая
       16

Как протестировать приложение python / flask со сторонним http lib?

0 голосов
/ 31 января 2020

У меня есть чистейший пакет для моего приложения flask, который прекрасно работает. Тем не менее, я хочу протестировать часть своего кода, который использует стороннюю библиотеку (Qt) для отправки http-запросов. Как это возможно? Я вижу, flask-testing имеет устройство live_server, которое выполняет это вместе с flask.url_for(), но для запуска сервера в устройстве требуется слишком много времени.

Существует ли более быстрый способ отправки http запрос от стороннего http lib на flask приложение?

Спасибо!

1 Ответ

0 голосов
/ 31 января 2020

Оказывается, вы можете сделать это вручную, преобразовав сторонний запрос в запрос FlaskClient, используя monkeypatch для любого метода "send", используемого сторонней библиотекой lib, а затем преобразовать ответ flask.Response обратно к стороннему объекту ответа. Все это происходит без использования порта TCP.

Вот пример, который я написал, чтобы соединить запросы Qt http с приложением flask: 

@pytest.fixture
def qnam(qApp, client, monkeypatch):

    def sendCustomRequest(request, verb, data):
        # Qt -> Flask
        headers = []
        for name in request.rawHeaderList():
            key = bytes(name).decode('utf-8')
            value = bytes(request.rawHeader(name)).decode('utf-8')
            headers.append((key, value))
        query_string = None
        if request.url().hasQuery():
            query_string = request.url().query()
        # method = request.attribute(QNetworkRequest.CustomVerbAttribute).decode('utf-8')
        # send
        response = FlaskClient.open(client,
                                    request.url().path(),
                                    method=verb.decode('utf-8'),
                                    headers=headers,
                                    data=data)
        # Flask -> Qt
        class NetworkReply(QNetworkReply):
            def abort(self):
                pass
        reply = NetworkReply()
        reply.setAttribute(QNetworkRequest.HttpStatusCodeAttribute, response.status_code)
        for key, value in response.headers:
            reply.setRawHeader(key.encode('utf-8'), value.encode('utf-8'))
        reply.open(QIODevice.ReadWrite)
        reply.write(response.data)
        QTimer.singleShot(10, reply.finished.emit) # after return
        return reply

    qnam = QNetworkAccessManager.instance() # or wherever you get your instance

    monkeypatch.setattr(qnam, 'sendCustomRequest', sendCustomRequest)
    return ret
...