Rooba
/
odyssey
mirror of https://github.com/yandex/odyssey.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

machinarium: add read timeout and cancel tests

This commit is contained in:
Dmitry Simonenko 2017-05-16 12:48:16 +03:00
parent c520c508db
commit ffff1d7606
5 changed files with 192 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
test-suite/machinarium_test.c
View File

@ -31,6 +31,8 @@ extern void test_accept_cancel(void);
extern void test_client_server(void);
extern void test_client_server_readahead(void);
extern void test_read_timeout(void);
extern void test_read_cancel(void);
int
main(int argc, char *argv[])
@ -54,5 +56,7 @@ main(int argc, char *argv[])
machinarium_test(test_accept_cancel);
machinarium_test(test_client_server);
machinarium_test(test_client_server_readahead);
machinarium_test(test_read_timeout);
machinarium_test(test_read_cancel);
return 0;
}

8
test-suite/machinarium_test.h
View File

@ -9,15 +9,17 @@
#define machinarium_test(function) \
do { \
fprintf(stdout, "%s: ", #function); \
fflush(stdout); \
(function)(); \
fprintf(stdout, "%s: ok\n", #function); \
fprintf(stdout, "ok\n"); \
} while (0);
#define test(expression) \
do { \
if (! (expression)) { \
fprintf(stdout, "%s: fail (%s:%d) %s\n", \
__func__, __FILE__, __LINE__, #expression); \
fprintf(stdout, "fail (%s:%d) %s\n", \
__FILE__, __LINE__, #expression); \
fflush(stdout); \
abort(); \
} \

4
test-suite/makefile
View File

@ -23,7 +23,9 @@ OBJECTS = machinarium_test.o \
test_accept_timeout.o \
test_accept_cancel.o \
test_client_server.o \
test_client_server_readahead.o
test_client_server_readahead.o \
test_read_timeout.o \
test_read_cancel.o
all: clean $(OBJECTS) machinarium_test
machinarium_test:
$(CC) $(OBJECTS) $(LFLAGS) -o machinarium_test

94
test-suite/test_read_cancel.c Normal file
View File

@ -0,0 +1,94 @@
/*
* machinarium.
*
* Cooperative multitasking engine.
*/
#include <machinarium.h>
#include <machinarium_test.h>
#include <string.h>
#include <arpa/inet.h>
static int client_id;
static void
server(void *arg)
{
machine_t machine = arg;
machine_io_t server = machine_create_io(machine);
test(server != NULL);
struct sockaddr_in sa;
sa.sin_family = AF_INET;
sa.sin_addr.s_addr = inet_addr("127.0.0.1");
sa.sin_port = htons(7778);
int rc;
rc = machine_bind(server, (struct sockaddr*)&sa);
test(rc == 0);
machine_io_t client;
rc = machine_accept(server, &client, 16, INT_MAX);
test(rc == 0);
machine_sleep(machine, 0);
rc = machine_cancel(machine, client_id);
test(rc == 0);
rc = machine_close(client);
test(rc == 0);
machine_free_io(client);
rc = machine_close(server);
test(rc == 0);
machine_free_io(server);
}
static void
client(void *arg)
{
machine_t machine = arg;
machine_io_t client = machine_create_io(machine);
test(client != NULL);
struct sockaddr_in sa;
sa.sin_family = AF_INET;
sa.sin_addr.s_addr = inet_addr("127.0.0.1");
sa.sin_port = htons(7778);
int rc;
rc = machine_connect(client, (struct sockaddr*)&sa, INT_MAX);
test(rc == 0);
char buf[16];
rc = machine_read(client, buf, 12, INT_MAX);
test(rc == -1);
test(! machine_read_timedout(client));
test(machine_cancelled(machine));
rc = machine_close(client);
test(rc == 0);
machine_free_io(client);
machine_stop(machine);
}
void
test_read_cancel(void)
{
machine_t machine = machine_create();
test(machine != NULL);
int rc;
rc = machine_create_fiber(machine, server, machine);
test(rc != -1);
rc = machine_create_fiber(machine, client, machine);
test(rc != -1);
client_id = rc;
machine_start(machine);
rc = machine_free(machine);
test(rc != -1);
}

86
test-suite/test_read_timeout.c Normal file
View File

@ -0,0 +1,86 @@
/*
* machinarium.
*
* Cooperative multitasking engine.
*/
#include <machinarium.h>
#include <machinarium_test.h>
#include <string.h>
#include <arpa/inet.h>
static void
server(void *arg)
{
machine_t machine = arg;
machine_io_t server = machine_create_io(machine);
test(server != NULL);
struct sockaddr_in sa;
sa.sin_family = AF_INET;
sa.sin_addr.s_addr = inet_addr("127.0.0.1");
sa.sin_port = htons(7778);
int rc;
rc = machine_bind(server, (struct sockaddr*)&sa);
test(rc == 0);
machine_io_t client;
rc = machine_accept(server, &client, 16, INT_MAX);
test(rc == 0);
rc = machine_close(client);
test(rc == 0);
machine_free_io(client);
rc = machine_close(server);
test(rc == 0);
machine_free_io(server);
}
static void
client(void *arg)
{
machine_t machine = arg;
machine_io_t client = machine_create_io(machine);
test(client != NULL);
struct sockaddr_in sa;
sa.sin_family = AF_INET;
sa.sin_addr.s_addr = inet_addr("127.0.0.1");
sa.sin_port = htons(7778);
int rc;
rc = machine_connect(client, (struct sockaddr*)&sa, INT_MAX);
test(rc == 0);
char buf[16];
rc = machine_read(client, buf, 12, 0);
test(rc == -1);
test(machine_read_timedout(client));
rc = machine_close(client);
test(rc == 0);
machine_free_io(client);
machine_stop(machine);
}
void
test_read_timeout(void)
{
machine_t machine = machine_create();
test(machine != NULL);
int rc;
rc = machine_create_fiber(machine, server, machine);
test(rc != -1);
rc = machine_create_fiber(machine, client, machine);
test(rc != -1);
machine_start(machine);
rc = machine_free(machine);
test(rc != -1);
}