static void test_read_and_write(void) { int status; pid_t pid; ThriftSocket *tsocket = NULL; ThriftTransport *transport = NULL; int port = 51199; guchar buf[10] = TEST_DATA; /* a buffer */ pid = fork (); assert ( pid >= 0 ); if ( pid == 0 ) { /* child listens */ thrift_server (port); exit (0); } else { /* parent connects, wait a bit for the socket to be created */ sleep (1); tsocket = g_object_new (THRIFT_TYPE_SOCKET, "hostname", "localhost", "port", port, NULL); transport = g_object_new (THRIFT_TYPE_BUFFERED_TRANSPORT, "transport", THRIFT_TRANSPORT (tsocket), "w_buf_size", 4, NULL); assert (thrift_buffered_transport_open (transport, NULL) == TRUE); assert (thrift_buffered_transport_is_open (transport)); /* write 10 bytes */ thrift_buffered_transport_write (transport, buf, 10, NULL); /* write 1 byte at a time */ thrift_buffered_transport_write (transport, buf, 1, NULL); thrift_buffered_transport_write (transport, buf, 1, NULL); thrift_buffered_transport_write (transport, buf, 1, NULL); /* overflow the buffer */ thrift_buffered_transport_write (transport, buf, 2, NULL); thrift_buffered_transport_write (transport, buf, 1, NULL); thrift_buffered_transport_flush (transport, NULL); /* write 1 byte and flush */ thrift_buffered_transport_write (transport, buf, 1, NULL); thrift_buffered_transport_flush (transport, NULL); /* write and overflow buffer with 2 system calls */ thrift_buffered_transport_write (transport, buf, 1, NULL); thrift_buffered_transport_write (transport, buf, 3, NULL); /* write 10 bytes */ thrift_buffered_transport_write (transport, buf, 10, NULL); thrift_buffered_transport_write_end (transport, NULL); thrift_buffered_transport_flush (transport, NULL); thrift_buffered_transport_close (transport, NULL); g_object_unref (transport); g_object_unref (tsocket); assert ( wait (&status) == pid ); assert ( status == 0 ); } }
static void test_write_fail(void) { int status; pid_t pid; ThriftSocket *tsocket = NULL; ThriftTransport *transport = NULL; int port = 51198; guchar buf[10] = TEST_DATA; /* a buffer */ /* SIGPIPE when send to disconnected socket */ signal(SIGPIPE, SIG_IGN); pid = fork (); assert ( pid >= 0 ); if ( pid == 0 ) { /* child listens */ ThriftServerTransport *transport = NULL; ThriftTransport *client = NULL; ThriftServerSocket *tsocket = g_object_new (THRIFT_TYPE_SERVER_SOCKET, "port", port, NULL); transport = THRIFT_SERVER_TRANSPORT (tsocket); thrift_server_transport_listen (transport, NULL); /* wrap the client in a BufferedTransport */ client = g_object_new (THRIFT_TYPE_BUFFERED_TRANSPORT, "transport", thrift_server_transport_accept (transport, NULL), "r_buf_size", 5, NULL); assert (client != NULL); /* just close socket */ thrift_buffered_transport_close (client, NULL); g_object_unref (client); g_object_unref (tsocket); exit (0); } else { /* parent connects, wait a bit for the socket to be created */ sleep (1); tsocket = g_object_new (THRIFT_TYPE_SOCKET, "hostname", "localhost", "port", port, NULL); transport = g_object_new (THRIFT_TYPE_BUFFERED_TRANSPORT, "transport", THRIFT_TRANSPORT (tsocket), "w_buf_size", 4, NULL); assert (thrift_buffered_transport_open (transport, NULL) == TRUE); assert (thrift_buffered_transport_is_open (transport)); /* recognize disconnection */ sleep(1); assert (thrift_buffered_transport_write (transport, buf, 10, NULL) == TRUE); assert (thrift_buffered_transport_write (transport, buf, 10, NULL) == FALSE); /* write and overflow buffer */ assert (thrift_buffered_transport_write (transport, buf, 10, NULL) == FALSE); /* write 1 and flush */ assert (thrift_buffered_transport_write (transport, buf, 1, NULL) == TRUE); assert (thrift_buffered_transport_flush (transport, NULL) == FALSE); thrift_buffered_transport_close (transport, NULL); g_object_unref (transport); g_object_unref (tsocket); assert ( wait (&status) == pid ); assert ( status == 0 ); } }