tcp_server_test.c

   1 #include "tcp_server.h"
   2 #include <stdio.h>
   3 #include <stdlib.h>
   4
   5 /* Unit tests */
   6 void unit_test_error( const gchar *log_domain
   7                     , GLogLevelFlags log_level
   8                     , const gchar *message
   9                     , gpointer user_data
  10                     )
  11 {
  12   printf("ERROR(%d) %s\n", log_level, message);
  13   if(log_level < G_LOG_LEVEL_ERROR) { exit(1); }
  14 }
  15
  16 GString *unit_test_input;
  17
  18 void unit_test_read_cb(char *buffer, int length, void *data)
  19 {
  20   tcp_client *client = (tcp_client*)(data);
  21   GString *reversed = g_string_new(g_utf8_strreverse(buffer, length));
  22
  23   //g_string_append_c(reversed, '\n');
  24
  25   g_string_append_len(unit_test_input, buffer, length);
  26
  27   tcp_client_write(client, reversed->str, reversed->len);
  28 }
  29
  30 void unit_test_accept(tcp_server *server, tcp_client *client, void *data)
  31 {
  32   fprintf(stdout, "Connection\n");
  33   client->read_cb = unit_test_read_cb;
  34   client->read_cb_data = client;
  35
  36   //tcp_client_close(client);
  37   //tcp_server_close(server);
  38 }
  39
  40 int main(void)
  41 {
  42   tcp_server *server;
  43   unit_test_input = g_string_new(NULL);
  44   server = tcp_server_new(unit_test_error);
  45
  46   g_log_set_handler (TCP_LOG_DOMAIN, G_LOG_LEVEL_MASK | G_LOG_FLAG_FATAL
  47                       | G_LOG_FLAG_RECURSION, unit_test_error, NULL);
  48
  49   fprintf(stdout, "Starting server at 0.0.0.0 31337\n");
  50
  51   tcp_server_listen(server, "localhost", 31337, 1024, unit_test_accept, NULL);
  52
  53   tcp_server_close(server);
  54
  55   return 0; // success
  56 }