src/server/test/testutils.c

changeset 385
a1f4cb076d2f
parent 321
9bf375c4b5bd
child 415
d938228c382e
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/src/server/test/testutils.c	Sat Sep 24 16:26:10 2022 +0200
@@ -0,0 +1,171 @@
+/*
+ * 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);
+}

mercurial