index da94f06ec3cfd1bf5b1391e7249b98dd6b9a0f20..43322e02adf00c7c852642e6a685263a44c8f3fd 100644 (file)
#include "frontend/connection.h"
#include "frontend/connection-private.h"
+#include "utils/proto.h"
#include "libsysdb_test.h"
#include "utils/strbuf.h"
return NULL;
} /* mock_client */
+static void
+connection_setup(sdb_conn_t *conn)
+{
+ ssize_t check, expected;
+
+ expected = 2 * sizeof(uint32_t) + strlen("fakeuser");
+ check = sdb_connection_send(conn, CONNECTION_STARTUP,
+ (uint32_t)strlen("fakeuser"), "fakeuser");
+ fail_unless(check == expected,
+ "sdb_connection_send(STARTUP, fakeuser) = %zi; expected: %zi",
+ check, expected);
+
+ mock_conn_rewind(conn);
+ check = sdb_connection_read(conn);
+ fail_unless(check == expected,
+ "On startup: sdb_connection_read() = %zi; expected: %zi",
+ check, expected);
+
+ fail_unless(sdb_strbuf_len(conn->errbuf) == 0,
+ "sdb_connection_read() left %zu bytes in the error "
+ "buffer; expected: 0", sdb_strbuf_len(conn->errbuf));
+
+ mock_conn_truncate(conn);
+} /* connection_setup */
+
/*
* tests
*/
}
END_TEST
+/* test connection setup and very basic commands */
START_TEST(test_conn_setup)
{
sdb_conn_t *conn = mock_conn_create();
const char *msg;
const char *err;
} golden_data[] = {
+ /* code == UINT32_MAX => no data will be sent */
{ UINT32_MAX, NULL, NULL },
{ CONNECTION_IDLE, "fakedata", "Invalid command 0" },
{ CONNECTION_PING, NULL, "Authentication required" },
}
END_TEST
+/* test simple I/O on open connections */
+START_TEST(test_conn_io)
+{
+ sdb_conn_t *conn = mock_conn_create();
+
+ struct {
+ uint32_t code;
+ uint32_t msg_len;
+ const char *msg;
+ size_t buf_len; /* number of bytes we expect in conn->buf */
+ const char *err;
+ } golden_data[] = {
+ /* code == UINT32_MAX => this is a follow-up package */
+ { CONNECTION_PING, 20, "9876543210", 10, NULL },
+ { UINT32_MAX, -1, "9876543210", 0, NULL },
+ { CONNECTION_IDLE, 20, "9876543210", 0, "Invalid command 0" },
+ { UINT32_MAX, -1, "9876543210", 0, "Invalid command 0" },
+ { CONNECTION_IDLE, 20, "9876543210", 0, "Invalid command 0" },
+ { UINT32_MAX, -1, "9876543210", 0, "Invalid command 0" },
+ { CONNECTION_PING, 10, "9876543210", 0, NULL },
+ { CONNECTION_PING, 20, "9876543210", 10, NULL },
+ { UINT32_MAX, -1, "9876543210", 0, NULL },
+ };
+
+ size_t i;
+
+ connection_setup(conn);
+
+ for (i = 0; i < SDB_STATIC_ARRAY_LEN(golden_data); ++i) {
+ size_t msg_len = strlen(golden_data[i].msg);
+ char buffer[2 * sizeof(uint32_t) + msg_len];
+ size_t offset = 0;
+
+ ssize_t check;
+
+ mock_conn_truncate(conn);
+
+ if (golden_data[i].code != UINT32_MAX) {
+ uint32_t tmp;
+
+ tmp = htonl(golden_data[i].code);
+ memcpy(buffer, &tmp, sizeof(tmp));
+ tmp = htonl(golden_data[i].msg_len);
+ memcpy(buffer + sizeof(tmp), &tmp, sizeof(tmp));
+
+ msg_len += 2 * sizeof(uint32_t);
+ offset += 2 * sizeof(uint32_t);
+ }
+
+ memcpy(buffer + offset, golden_data[i].msg,
+ strlen(golden_data[i].msg));
+
+ check = sdb_proto_send(conn->fd, msg_len, buffer);
+ fail_unless(check == (ssize_t)msg_len,
+ "sdb_proto_send(%s) = %zi; expected: %zu",
+ check, msg_len);
+
+ mock_conn_rewind(conn);
+ check = sdb_connection_read(conn);
+ fail_unless(check == (ssize_t)msg_len,
+ "sdb_connection_read() = %zi; expected: %zu",
+ check, msg_len);
+
+ if (golden_data[i].buf_len) {
+ /* partial commands need to be stored in the object */
+ fail_unless(conn->cmd == golden_data[i].code,
+ "sdb_connection_read() set partial command "
+ "to %u; expected: %u", conn->cmd, golden_data[i].code);
+ fail_unless(conn->cmd_len > golden_data[i].buf_len,
+ "sdb_connection_read() set partial command length "
+ "to %u; expected: > %u", conn->cmd_len,
+ golden_data[i].buf_len);
+ }
+ else {
+ fail_unless(conn->cmd == CONNECTION_IDLE,
+ "sdb_connection_read() did not reset command; "
+ "got %u; expected: %u", conn->cmd, CONNECTION_IDLE);
+ fail_unless(conn->cmd_len == 0,
+ "sdb_connection_read() did not reset command length; "
+ "got %u; expected: 0", conn->cmd_len);
+ }
+
+ fail_unless(sdb_strbuf_len(conn->buf) == golden_data[i].buf_len,
+ "sdb_connection_read() left %zu bytes in the buffer; "
+ "expected: %zu", sdb_strbuf_len(conn->buf),
+ golden_data[i].buf_len);
+
+ if (golden_data[i].err) {
+ const char *err = sdb_strbuf_string(conn->errbuf);
+ fail_unless(strcmp(err, golden_data[i].err) == 0,
+ "sdb_connection_read(): got error '%s'; "
+ "expected: '%s'", err, golden_data[i].err);
+ }
+ else
+ fail_unless(sdb_strbuf_len(conn->errbuf) == 0,
+ "sdb_connection_read() left %zu bytes in the error "
+ "buffer; expected: 0", sdb_strbuf_len(conn->errbuf));
+ }
+
+ mock_conn_destroy(conn);
+}
+END_TEST
+
Suite *
fe_conn_suite(void)
{
tc = tcase_create("core");
tcase_add_test(tc, test_conn_accept);
tcase_add_test(tc, test_conn_setup);
+ tcase_add_test(tc, test_conn_io);
suite_add_tcase(s, tc);
return s;