src/server/test/testutils.c

/*
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
 *
 * Copyright 2019 Olaf Wintermann. All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 *   1. Redistributions of source code must retain the above copyright
 *      notice, this list of conditions and the following disclaimer.
 *
 *   2. Redistributions in binary form must reproduce the above copyright
 *      notice, this list of conditions and the following disclaimer in the
 *      documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 */

#include <stdio.h>
#include <stdlib.h>

#include <ucx/string.h>
#include <ucx/utils.h>

#include "../util/pblock.h"

#include "../util/io.h"

#include "testutils.h"

Session* testutil_session(void) {
    pool_handle_t *pool = pool_create();
    NSAPISession *sn = nsapisession_create(pool);
    sn->connection = testutil_dummy_connection(pool);
    
    return &sn->sn;
}

Request* testutil_request(pool_handle_t *pool, const char *method, const char *uri) {
    NSAPIRequest *rq = pool_malloc(pool, sizeof(NSAPIRequest));
    ZERO(rq, sizeof(NSAPIRequest));
    
    HTTPRequest httprequest;
    ZERO(&httprequest, sizeof(HTTPRequest));
    request_initialize(pool, &httprequest, rq);
    
    sstr_t clf = ucx_sprintf("%s %s HTTP/1.1", method, uri);
    pblock_kvinsert(
            pb_key_clf_request,
            clf.ptr,
            clf.length,
            rq->rq.reqpb);
    free(clf.ptr);
    
    pblock_nvinsert(
            "method",
            method,
            rq->rq.reqpb);
    
    pblock_nvinsert(
            "protocol",
            "HTTP/1.1",
            rq->rq.reqpb);
    
    pblock_nvinsert("uri", uri, rq->rq.reqpb);
    
    return &rq->rq;
}

static int dummyconn_read(Connection *conn, void *buf, int len) {
    return len;
}

static int dummyconn_write(Connection *conn, const void *buf, int len) {
    return len;
}

static void dummyconn_close(Connection *conn) {
    
}


Connection* testutil_dummy_connection(pool_handle_t *pool) {
    Connection *conn = pool_malloc(pool, sizeof(Connection));
    ZERO(conn, sizeof(Connection));
    conn->read = dummyconn_read;
    conn->write = dummyconn_write;
    conn->close = dummyconn_close; 
    return conn;
}

void testutil_request_body(Session *sn, Request *rq, const char *body, size_t len) {
    sstr_t cl = ucx_sprintf("%d", (int)len);
    pblock_nvreplace("content-length", cl.ptr, rq->headers);
    free(cl.ptr);
    
    netbuf *inbuf = pool_malloc(sn->pool, sizeof(netbuf));
    inbuf->sd = NULL;
    inbuf->inbuf = pool_malloc(sn->pool, len);
    inbuf->pos = 0;
    inbuf->maxsize = len;
    inbuf->cursize = len;
    sn->inbuf = inbuf;
    
    memcpy(inbuf->inbuf, body, len);
}

void testutil_destroy_session(Session *sn) {
    pool_destroy(sn->pool);
}


static ssize_t test_io_write(IOStream *io, void *buf, size_t size) {
    TestIOStream *st = (TestIOStream*)io;
    return ucx_buffer_write(buf, 1, size, st->buf);
}

static ssize_t test_io_writev(IOStream *io, struct iovec *iovec, int iovctn) {
    return -1;
}

static ssize_t test_io_read(IOStream *io, void *buf, size_t size) {
    return -1;
}

static void test_io_close(IOStream *io) {
    
}

static void test_io_finish(IOStream *io) {
    
}

static void test_io_setmode(IOStream *io, int mode) {
    
}

static int test_io_poll(IOStream *io, EventHandler *ev, int events , Event *event) {
    return 1;
}

TestIOStream* testutil_iostream(size_t size, int autoextend) {
    TestIOStream *stream = calloc(1, sizeof(TestIOStream));
    int flags = 0;
    if(autoextend) {
        flags = UCX_BUFFER_AUTOEXTEND;
    }
    stream->buf = ucx_buffer_new(NULL, size, flags);
    
    stream->io.st.write = test_io_write;
    stream->io.st.writev = test_io_writev;
    stream->io.st.close = test_io_close;
    stream->io.st.finish = test_io_finish;
    
    return stream;
}

void testutil_iostream_destroy(TestIOStream *stream) {
    ucx_buffer_free(stream->buf);
    free(stream);
}