Initial commit

1 files changed, 221 insertions, 0 deletions

diff --git a/tests/t_http.c b/tests/t_http.c
new file mode 100644
index 0000000..5251dc7
--- /dev/null
+++ b/tests/t_http.c
@@ -0,0 +1,221 @@
+#include "http.h"
+
+#include <sqlite3.h>
+
+#include "test.h"
+
+SQLITE_EXTENSION_INIT3
+
+void http_backend_dummy_set_errmsg(const char* zErrMsg);
+void http_backend_dummy_set_response(http_response* response);
+void http_backend_dummy_reset_request();
+const http_request* http_backend_dummy_get_last_request();
+
+int sqlite3_http_init(sqlite3*, char**, const sqlite3_api_routines*);
+
+void new_text_response(http_response* response,
+                       const char* zBody,
+                       const char* zHeaders,
+                       int iStatusCode,
+                       const char* zStatus) {
+    memset(response, 0, sizeof(*response));
+    assert(zStatus != NULL);
+    if (zBody) {
+        response->pBody = sqlite3_mprintf("%s", zBody);
+        response->szBody = strlen(zBody);
+    }
+    if (zHeaders) {
+        response->zHeaders = sqlite3_mprintf("%s", zHeaders);
+        response->szHeaders = strlen(zHeaders);
+    }
+    response->iStatusCode = iStatusCode;
+    response->zStatus = sqlite3_mprintf("%s", zStatus);
+}
+
+static sqlite3* db;
+
+void test_http_get() {
+    sqlite3_stmt* stmt;
+    http_response response;
+    new_text_response(&response, "hello, world!", "Foo: Bar\r\n\r\n", 200, "HTTP/1.0 200 OK");
+    http_backend_dummy_set_response(&response);
+    ASSERT_INT_EQ(
+        sqlite3_prepare_v2(db, "select * from http_get('http://example.com')", -1, &stmt, NULL),
+        SQLITE_OK);
+    ASSERT_INT_EQ(sqlite3_step(stmt), SQLITE_ROW);
+    ASSERT_INT_EQ(sqlite3_column_count(stmt), 4);
+    ASSERT_STR_EQ(sqlite3_column_text(stmt, 0), "HTTP/1.0 200 OK");
+    ASSERT_INT_EQ(sqlite3_column_int(stmt, 1), 200);
+    ASSERT_STR_EQ(sqlite3_column_text(stmt, 2), "Foo: Bar\r\n\r\n");
+    ASSERT_STR_EQ(sqlite3_column_text(stmt, 3), "hello, world!");
+    ASSERT_INT_EQ(sqlite3_finalize(stmt), SQLITE_OK);
+}
+
+void test_http_get_hidden_columns() {
+    sqlite3_stmt* stmt;
+    http_response response;
+    new_text_response(&response, "hello, world!", "Foo: Bar\r\n\r\n", 200, "HTTP/1.0 200 OK");
+    http_backend_dummy_set_response(&response);
+    ASSERT_INT_EQ(
+        sqlite3_prepare_v2(db,
+                           "select response_status, response_status_code, response_headers, "
+                           "response_body, request_method, request_url, request_headers, "
+                           "request_body from http_get('http://example.com/foo')",
+                           -1,
+                           &stmt,
+                           NULL),
+        SQLITE_OK);
+    ASSERT_INT_EQ(sqlite3_step(stmt), SQLITE_ROW);
+    ASSERT_INT_EQ(sqlite3_column_count(stmt), 8);
+    ASSERT_STR_EQ(sqlite3_column_text(stmt, 0), "HTTP/1.0 200 OK");
+    ASSERT_INT_EQ(sqlite3_column_int(stmt, 1), 200);
+    ASSERT_STR_EQ(sqlite3_column_text(stmt, 2), "Foo: Bar\r\n\r\n");
+    ASSERT_STR_EQ(sqlite3_column_text(stmt, 3), "hello, world!");
+    ASSERT_STR_EQ(sqlite3_column_text(stmt, 4), "GET");
+    ASSERT_STR_EQ(sqlite3_column_text(stmt, 5), "http://example.com/foo");
+    ASSERT_NULL(sqlite3_column_text(stmt, 6));
+    ASSERT_NULL(sqlite3_column_text(stmt, 7));
+    ASSERT_INT_EQ(sqlite3_finalize(stmt), SQLITE_OK);
+}
+
+void test_http_get_request_headers() {
+    sqlite3_stmt* stmt;
+    http_response response;
+    new_text_response(&response, "hello, world!", "Foo: Bar\r\n\r\n", 200, "HTTP/1.0 200 OK");
+    http_backend_dummy_set_response(&response);
+    ASSERT_INT_EQ(
+        sqlite3_prepare_v2(db,
+                           "select response_status, response_status_code, response_headers, "
+                           "response_body, request_method, request_url, request_headers, "
+                           "request_body from http_get('http://example.com/foo', "
+                           "http_headers('Req1', 'Val1', 'Req2', 'Val2'))",
+                           -1,
+                           &stmt,
+                           NULL),
+        SQLITE_OK);
+    ASSERT_INT_EQ(sqlite3_step(stmt), SQLITE_ROW);
+    ASSERT_INT_EQ(sqlite3_column_count(stmt), 8);
+    ASSERT_STR_EQ(sqlite3_column_text(stmt, 0), "HTTP/1.0 200 OK");
+    ASSERT_INT_EQ(sqlite3_column_int(stmt, 1), 200);
+    ASSERT_STR_EQ(sqlite3_column_text(stmt, 2), "Foo: Bar\r\n\r\n");
+    ASSERT_STR_EQ(sqlite3_column_text(stmt, 3), "hello, world!");
+    ASSERT_STR_EQ(sqlite3_column_text(stmt, 4), "GET");
+    ASSERT_STR_EQ(sqlite3_column_text(stmt, 5), "http://example.com/foo");
+    ASSERT_STR_EQ(sqlite3_column_text(stmt, 6), "Req1: Val1\r\nReq2: Val2\r\n");
+    ASSERT_NULL(sqlite3_column_text(stmt, 7));
+    ASSERT_INT_EQ(sqlite3_finalize(stmt), SQLITE_OK);
+}
+
+void test_http_post() {
+    sqlite3_stmt* stmt;
+    http_response response;
+    new_text_response(&response, "hello, world!", "Foo: Bar\r\n\r\n", 200, "HTTP/1.0 200 OK");
+    http_backend_dummy_set_response(&response);
+    ASSERT_INT_EQ(
+        sqlite3_prepare_v2(db, "select * from http_post('http://example.com')", -1, &stmt, NULL),
+        SQLITE_OK);
+    ASSERT_INT_EQ(sqlite3_step(stmt), SQLITE_ROW);
+    ASSERT_INT_EQ(sqlite3_column_count(stmt), 4);
+    ASSERT_STR_EQ(sqlite3_column_text(stmt, 0), "HTTP/1.0 200 OK");
+    ASSERT_INT_EQ(sqlite3_column_int(stmt, 1), 200);
+    ASSERT_STR_EQ(sqlite3_column_text(stmt, 2), "Foo: Bar\r\n\r\n");
+    ASSERT_STR_EQ(sqlite3_column_text(stmt, 3), "hello, world!");
+    ASSERT_INT_EQ(sqlite3_finalize(stmt), SQLITE_OK);
+}
+
+void test_http_post_hidden_columns() {
+    sqlite3_stmt* stmt;
+    http_response response;
+    new_text_response(&response, "hello, world!", "Foo: Bar\r\n\r\n", 200, "HTTP/1.0 200 OK");
+    http_backend_dummy_set_response(&response);
+    ASSERT_INT_EQ(
+        sqlite3_prepare_v2(db,
+                           "select response_status, response_status_code, response_headers, "
+                           "response_body, request_method, request_url, request_headers, "
+                           "request_body from http_post('http://example.com/foo')",
+                           -1,
+                           &stmt,
+                           NULL),
+        SQLITE_OK);
+    ASSERT_INT_EQ(sqlite3_step(stmt), SQLITE_ROW);
+    ASSERT_INT_EQ(sqlite3_column_count(stmt), 8);
+    ASSERT_STR_EQ(sqlite3_column_text(stmt, 0), "HTTP/1.0 200 OK");
+    ASSERT_INT_EQ(sqlite3_column_int(stmt, 1), 200);
+    ASSERT_STR_EQ(sqlite3_column_text(stmt, 2), "Foo: Bar\r\n\r\n");
+    ASSERT_STR_EQ(sqlite3_column_text(stmt, 3), "hello, world!");
+    ASSERT_STR_EQ(sqlite3_column_text(stmt, 4), "POST");
+    ASSERT_STR_EQ(sqlite3_column_text(stmt, 5), "http://example.com/foo");
+    ASSERT_NULL(sqlite3_column_text(stmt, 6));
+    ASSERT_NULL(sqlite3_column_text(stmt, 7));
+    ASSERT_INT_EQ(sqlite3_finalize(stmt), SQLITE_OK);
+}
+
+void test_http_post_request_headers() {
+    sqlite3_stmt* stmt;
+    http_response response;
+    new_text_response(&response, "hello, world!", "Foo: Bar\r\n\r\n", 200, "HTTP/1.0 200 OK");
+    http_backend_dummy_set_response(&response);
+    ASSERT_INT_EQ(
+        sqlite3_prepare_v2(db,
+                           "select response_status, response_status_code, response_headers, "
+                           "response_body, request_method, request_url, request_headers, "
+                           "request_body from http_post('http://example.com/foo', "
+                           "http_headers('Req1', 'Val1', 'Req2', 'Val2'))",
+                           -1,
+                           &stmt,
+                           NULL),
+        SQLITE_OK);
+    ASSERT_INT_EQ(sqlite3_step(stmt), SQLITE_ROW);
+    ASSERT_INT_EQ(sqlite3_column_count(stmt), 8);
+    ASSERT_STR_EQ(sqlite3_column_text(stmt, 0), "HTTP/1.0 200 OK");
+    ASSERT_INT_EQ(sqlite3_column_int(stmt, 1), 200);
+    ASSERT_STR_EQ(sqlite3_column_text(stmt, 2), "Foo: Bar\r\n\r\n");
+    ASSERT_STR_EQ(sqlite3_column_text(stmt, 3), "hello, world!");
+    ASSERT_STR_EQ(sqlite3_column_text(stmt, 4), "POST");
+    ASSERT_STR_EQ(sqlite3_column_text(stmt, 5), "http://example.com/foo");
+    ASSERT_STR_EQ(sqlite3_column_text(stmt, 6), "Req1: Val1\r\nReq2: Val2\r\n");
+    ASSERT_NULL(sqlite3_column_text(stmt, 7));
+    ASSERT_INT_EQ(sqlite3_finalize(stmt), SQLITE_OK);
+}
+
+void test_http_post_request_body() {
+    sqlite3_stmt* stmt;
+    http_response response;
+    new_text_response(&response, "hello, world!", "Foo: Bar\r\n\r\n", 200, "HTTP/1.0 200 OK");
+    http_backend_dummy_set_response(&response);
+    ASSERT_INT_EQ(
+        sqlite3_prepare_v2(db,
+                           "select response_status, response_status_code, response_headers, "
+                           "response_body, request_method, request_url, request_headers, "
+                           "request_body from http_post('http://example.com/foo', "
+                           "http_headers('Req1', 'Val1', 'Req2', 'Val2'), 'hello')",
+                           -1,
+                           &stmt,
+                           NULL),
+        SQLITE_OK);
+    ASSERT_INT_EQ(sqlite3_step(stmt), SQLITE_ROW);
+    ASSERT_INT_EQ(sqlite3_column_count(stmt), 8);
+    ASSERT_STR_EQ(sqlite3_column_text(stmt, 0), "HTTP/1.0 200 OK");
+    ASSERT_INT_EQ(sqlite3_column_int(stmt, 1), 200);
+    ASSERT_STR_EQ(sqlite3_column_text(stmt, 2), "Foo: Bar\r\n\r\n");
+    ASSERT_STR_EQ(sqlite3_column_text(stmt, 3), "hello, world!");
+    ASSERT_STR_EQ(sqlite3_column_text(stmt, 4), "POST");
+    ASSERT_STR_EQ(sqlite3_column_text(stmt, 5), "http://example.com/foo");
+    ASSERT_STR_EQ(sqlite3_column_text(stmt, 6), "Req1: Val1\r\nReq2: Val2\r\n");
+    ASSERT_STR_EQ(sqlite3_column_text(stmt, 7), "hello");
+    ASSERT_INT_EQ(sqlite3_finalize(stmt), SQLITE_OK);
+}
+
+int main(int argc, char const* argv[]) {
+    sqlite3_initialize();
+    sqlite3_auto_extension((void (*)(void))sqlite3_http_init);
+    ASSERT_INT_EQ(sqlite3_open(":memory:", &db), SQLITE_OK);
+    test_http_get();
+    test_http_get_hidden_columns();
+    test_http_get_request_headers();
+    test_http_post();
+    test_http_post_hidden_columns();
+    test_http_post_request_headers();
+    test_http_post_request_body();
+    return 0;
+}