mirror of
https://github.com/valitydev/thrift.git
synced 2024-11-07 18:58:51 +00:00
96 lines
3.1 KiB
C
Executable File
96 lines
3.1 KiB
C
Executable File
/*
|
|
* Licensed to the Apache Software Foundation (ASF) under one
|
|
* or more contributor license agreements. See the NOTICE file
|
|
* distributed with this work for additional information
|
|
* regarding copyright ownership. The ASF licenses this file
|
|
* to you under the Apache License, Version 2.0 (the
|
|
* "License"); you may not use this file except in compliance
|
|
* with the License. You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing,
|
|
* software distributed under the License is distributed on an
|
|
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
* KIND, either express or implied. See the License for the
|
|
* specific language governing permissions and limitations
|
|
* under the License.
|
|
*/
|
|
|
|
#include <assert.h>
|
|
#include <netdb.h>
|
|
|
|
#include <thrift/c_glib/transport/thrift_transport.h>
|
|
#include <thrift/c_glib/transport/thrift_socket.h>
|
|
#include <thrift/c_glib/transport/thrift_server_transport.h>
|
|
#include <thrift/c_glib/transport/thrift_server_socket.h>
|
|
|
|
#define TEST_DATA { 'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j' }
|
|
|
|
#include "../src/thrift/c_glib/transport/thrift_memory_buffer.c"
|
|
|
|
/* test object creation and destruction */
|
|
static void
|
|
test_create_and_destroy(void)
|
|
{
|
|
GObject *object = NULL;
|
|
object = g_object_new (THRIFT_TYPE_MEMORY_BUFFER, NULL);
|
|
assert (object != NULL);
|
|
g_object_unref (object);
|
|
}
|
|
|
|
static void
|
|
test_open_and_close(void)
|
|
{
|
|
ThriftMemoryBuffer *tbuffer = NULL;
|
|
|
|
/* create a ThriftMemoryBuffer */
|
|
tbuffer = g_object_new (THRIFT_TYPE_MEMORY_BUFFER, NULL);
|
|
|
|
/* this shouldn't work */
|
|
assert (thrift_memory_buffer_open (THRIFT_TRANSPORT (tbuffer), NULL) == TRUE);
|
|
assert (thrift_memory_buffer_is_open (THRIFT_TRANSPORT (tbuffer)) == TRUE);
|
|
assert (thrift_memory_buffer_close (THRIFT_TRANSPORT (tbuffer), NULL) == TRUE);
|
|
g_object_unref (tbuffer);
|
|
}
|
|
|
|
static void
|
|
test_read_and_write(void)
|
|
{
|
|
ThriftMemoryBuffer *tbuffer = NULL;
|
|
guchar buf[10] = TEST_DATA;
|
|
guchar read[10];
|
|
GError *error = NULL;
|
|
|
|
tbuffer = g_object_new (THRIFT_TYPE_MEMORY_BUFFER, "buf_size", 5, NULL);
|
|
assert (thrift_memory_buffer_write (THRIFT_TRANSPORT (tbuffer),
|
|
(gpointer) buf,
|
|
10, &error) == FALSE);
|
|
assert (error != NULL);
|
|
g_error_free (error);
|
|
error = NULL;
|
|
g_object_unref (tbuffer);
|
|
|
|
tbuffer = g_object_new (THRIFT_TYPE_MEMORY_BUFFER, "buf_size", 15, NULL);
|
|
assert (thrift_memory_buffer_write (THRIFT_TRANSPORT (tbuffer),
|
|
(gpointer) buf, 10, &error) == TRUE);
|
|
assert (error == NULL);
|
|
|
|
assert (thrift_memory_buffer_read (THRIFT_TRANSPORT (tbuffer),
|
|
&read, 10, &error) > 0);
|
|
assert (error == NULL);
|
|
}
|
|
|
|
int
|
|
main(int argc, char *argv[])
|
|
{
|
|
g_type_init();
|
|
g_test_init (&argc, &argv, NULL);
|
|
|
|
g_test_add_func ("/testmemorybuffer/CreateAndDestroy", test_create_and_destroy);
|
|
g_test_add_func ("/testmemorybuffer/OpenAndClose", test_open_and_close);
|
|
g_test_add_func ("/testmemorybuffer/ReadAndWrite", test_read_and_write);
|
|
|
|
return g_test_run ();
|
|
}
|