1 /**

     2  * The main test code entry point

     3  */

     4 #include "sock_test.h"

     5 #include "irc_conn.h"

     6 #include "log.h"

     7 

     8 #include <string.h>

     9 #include <assert.h>

    10 

    11 void assert_sock_read (struct sock_stream *sock, const char *str)

    12 {

    13     char buf[strlen(str)];

    14 

    15     log_debug("read: %p: '%s'", sock, str);

    16     

    17     // read it

    18     assert(sock_stream_read(sock, buf, strlen(str)) == (int) strlen(str));

    19 

    20     // cmp

    21     assert(strncmp(buf, str, strlen(str)) == 0);

    22 }

    23 

    24 void assert_sock_write (struct sock_stream *sock, const char *str)

    25 {

    26     log_debug("write: %p: '%s'", sock, str);

    27 

    28     // write it

    29     assert(sock_stream_write(sock, str, strlen(str)) == (int) strlen(str));

    30 }

    31 

    32 void test_sock_test (void)

    33 {

    34     struct sock_test *sock = sock_test_create();

    35     struct io_vec _read_data[] = {

    36         {   "foo",      3   },

    37         {   "barx",     4   }

    38     };

    39     const char *_write_data = "test data";

    40     

    41     // put the read data

    42     log_debug("set_recv_buffer: %p, %d", _read_data, 2);

    43     sock_test_set_recv_buffer(sock, _read_data, 2);

    44 

    45     // read it out

    46     assert_sock_read(SOCK_TEST_BASE(sock), "foo");

    47     assert_sock_read(SOCK_TEST_BASE(sock), "ba");

    48     assert_sock_read(SOCK_TEST_BASE(sock), "rx");

    49 

    50     // write the data in

    51     assert_sock_write(SOCK_TEST_BASE(sock), "test ");

    52     assert_sock_write(SOCK_TEST_BASE(sock), "data");

    53 

    54     // get the data out

    55     char *data;

    56     size_t len;

    57 

    58     sock_test_get_send_data(sock, &data, &len);

    59     

    60     log_debug("get_send_data: %u: '%s'", len, data);

    61 

    62     // should be the same

    63     assert(len == strlen(_write_data));

    64     assert(strncmp(data, _write_data, len) == 0);

    65 }

    66 

    67 static struct irc_conn_callbacks _conn_callbacks = {

    68     .on_registered      = NULL,

    69 };

    70 

    71 void test_irc_conn (void)

    72 {

    73     struct sock_test *sock;

    74     struct irc_conn *conn;

    75     struct error_info err;

    76 

    77     // create the test socket

    78     assert((sock = sock_test_create()));

    79     

    80     // create the irc_conn

    81     assert(irc_conn_create(&conn, SOCK_TEST_BASE(sock), &_conn_callbacks, NULL, &err) == SUCCESS);

    82 

    83     // destroy it

    84     irc_conn_destroy(conn);

    85 }

    86 

    87 /**

    88  * Test definition

    89  */

    90 static struct test {

    91     /** Test name */

    92     const char *name;

    93 

    94     /** Test func */

    95     void (*func) (void);

    96 

    97 } _tests[] = {

    98     {   "sock_test",    &test_sock_test     },

    99     {   "irc_conn",     &test_irc_conn      },

   100     {   NULL,           NULL                }

   101 };

   102 

   103 int main (int argc, char **argv)

   104 {

   105     struct test *test;

   106 

   107     (void) argv;

   108     

   109     // no arguments

   110     assert(argc == 1);

   111 

   112     // run tests

   113     for (test = _tests; test->name; test++) {

   114         log_info("Running test: %s", test->name);

   115 

   116         test->func();

   117     }

   118 }