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 <limits.h>

#include <cx/string.h>
#include <cx/utils.h>
#include <cx/printf.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);
    
    cxmutstr clf = cx_asprintf("%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) {
    cxmutstr cl = cx_asprintf("%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, const void *buf, size_t size) {
    TestIOStream *st = (TestIOStream*)io;
    if(size > st->max_write) size = st->max_write;
    return cxBufferWrite(buf, 1, size, st->buf);
}

static ssize_t test_io_writev(IOStream *io, struct iovec *iovec, int iovctn) {
    TestIOStream *st = (TestIOStream*)io;
    ssize_t wv = 0;
    for(int i=0;i<iovctn;i++) {
        ssize_t available = st->max_write - wv;
        size_t len = iovec[i].iov_len;
        if(len > available) {
            len = available;
        }
        
        ssize_t w =  test_io_write(io, iovec[i].iov_base, len);
        if(w <= 0) {
            break;
        }
        wv += w;
        if(wv >= st->max_write) {
            break;
        }
    }
    return wv;
}

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 = CX_BUFFER_AUTO_EXTEND|CX_BUFFER_FREE_CONTENTS;
    }
    stream->buf = malloc(sizeof(CxBuffer));
    stream->max_write = INT_MAX;
    cxBufferInit(stream->buf, NULL, size, cxDefaultAllocator, 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;
    stream->io.st.type = IO_STREAM_TYPE_HTTP;
    
    return stream;
}

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