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(int severity, char *message)
   7 {
   8   printf("ERROR(%d) %s\n", severity, message);
   9   if(severity == TCP_FATAL) { exit(1); }
  10 }
  11
  12 GString *unit_test_input;
  13
  14 void unit_test_read_cb(tcp_client *client, char *buffer, int length)
  15 {
  16   GString *reversed = g_string_new(g_utf8_strreverse(buffer, length));
  17
  18   //g_string_append_c(reversed, '\n');
  19
  20   g_string_append_len(unit_test_input, buffer, length);
  21
  22   tcp_client_write(client, reversed->str, reversed->len);
  23 }
  24
  25 void unit_test_accept(tcp_server *server, tcp_client *client)
  26 {
  27   fprintf(stdout, "Connection!\n");
  28   tcp_client_set_read_cb(client, unit_test_read_cb);
  29   //tcp_client_close(client);
  30   //tcp_server_close(server);
  31 }
  32
  33 int main(void)
  34 {
  35   tcp_server *server;
  36   unit_test_input = g_string_new(NULL);
  37   server = tcp_server_new(unit_test_error);
  38
  39   fprintf(stdout, "Starting server at 0.0.0.0 31337\n");
  40
  41   tcp_server_listen(server, "localhost", 31337, 1024, unit_test_accept);
  42
  43   tcp_server_close(server);
  44
  45   return 0; // success
  46 }