testengine.cpp

#include "testengine.hpp"

#include "testengine_p.h"

#include <QBuffer>

using namespace Cutelyst;

TestEngine::TestEngine(Application *app, const QVariantMap &opts)
    : Engine{app, 0, opts}
{
}

int TestEngine::workerId() const
{
    return 0;
}

TestEngine::TestResponse TestEngine::createRequest(const QByteArray &method,
                                                   const QByteArray &path,
                                                   const QByteArray &query,
                                                   const Headers &headers,
                                                   QByteArray *body)
{
    QIODevice *bodyDevice = nullptr;
    if (headers.header("Sequential"_qba).isEmpty()) {
        bodyDevice = new QBuffer(body);
    } else {
        bodyDevice = new SequentialBuffer(body);
    }
    bodyDevice->open(QIODevice::ReadOnly);

    Headers headersCL = headers;
    if (bodyDevice->size()) {
        headersCL.setContentLength(bodyDevice->size());
    }

    TestEngineConnection req;
    req.method       = method;
    QByteArray _path = path;
    req.setPath(_path);
    req.query          = query;
    req.protocol       = "HTTP/1.1"_qba;
    req.isSecure       = false;
    req.serverAddress  = "127.0.0.1"_qba;
    req.remoteAddress  = QHostAddress(u"127.0.0.1"_qs);
    req.remotePort     = 3000;
    req.remoteUser     = QString{};
    req.headers        = headersCL;
    req.startOfRequest = std::chrono::steady_clock::now();
    req.body           = bodyDevice;

    Q_EMIT processRequestAsync(&req);

    // Due async requests we create a local event loop
    req.m_eventLoop.exec();

    TestResponse ret;
    ret.body       = req.m_responseData;
    ret.statusCode = req.m_statusCode;
    ret.headers    = req.m_headers;

    return ret;
}

TestEngine::TestResponse TestEngine::createRequest(const QByteArray &method,
                                                   const QString &path,
                                                   const QByteArray &query,
                                                   const Headers &headers,
                                                   QByteArray *body)
{
    return createRequest(method, path.toLatin1(), query, headers, body);
}

bool TestEngine::init()
{
    return initApplication() && postForkApplication();
}

SequentialBuffer::SequentialBuffer(QByteArray *buffer)
    : buf(buffer)
{
}

bool SequentialBuffer::isSequential() const
{
    return true;
}

qint64 SequentialBuffer::bytesAvailable() const
{
    return buf->size() + QIODevice::bytesAvailable();
}

qint64 SequentialBuffer::readData(char *data, qint64 maxlen)
{
    QByteArray mid = buf->mid(pos(), maxlen);
    memcpy(data, mid.data(), mid.size());
    // Sequential devices consume the body
    buf->remove(0, mid.size());
    return mid.size();
}

qint64 SequentialBuffer::writeData(const char *data, qint64 len)
{
    Q_UNUSED(data);
    Q_UNUSED(len);
    return -1;
}

qint64 TestEngineConnection::doWrite(const char *data, qint64 len)
{
    m_responseData.append(data, len);
    return len;
}

bool TestEngineConnection::writeHeaders(quint16 status, const Headers &headers)
{
    m_statusCode = status;
    m_headers    = headers;

    return true;
}

void TestEngineConnection::processingFinished()
{
    m_eventLoop.quit();
}

#include "moc_testengine.cpp"