mirror of
https://github.com/vale981/ray
synced 2025-03-12 14:16:39 -04:00
084220b0e7
* Allow reading/writing generic message types, not just tasks. * Allow messages of length 0 to be read/written, and handle closed sockets. * Address comments. * Simplify accept_client. * Allow ports to be reused in bind_ipc_sock.
96 lines
2.6 KiB
C
96 lines
2.6 KiB
C
#include "greatest.h"
|
|
|
|
#include <unistd.h>
|
|
#include <sys/types.h>
|
|
#include <sys/socket.h>
|
|
|
|
#include "common.h"
|
|
#include "test/example_task.h"
|
|
#include "task.h"
|
|
#include "io.h"
|
|
|
|
SUITE(task_tests);
|
|
|
|
TEST task_test(void) {
|
|
function_id func_id = globally_unique_id();
|
|
task_spec *task = alloc_task_spec(func_id, 4, 2, 10);
|
|
ASSERT(task_num_args(task) == 4);
|
|
ASSERT(task_num_returns(task) == 2);
|
|
|
|
unique_id arg1 = globally_unique_id();
|
|
ASSERT(task_args_add_ref(task, arg1) == 0);
|
|
ASSERT(task_args_add_val(task, (uint8_t*) "hello", 5) == 1);
|
|
unique_id arg2 = globally_unique_id();
|
|
ASSERT(task_args_add_ref(task, arg2) == 2);
|
|
ASSERT(task_args_add_val(task, (uint8_t*) "world", 5) == 3);
|
|
|
|
unique_id ret0 = globally_unique_id();
|
|
unique_id ret1 = globally_unique_id();
|
|
memcpy(task_return(task, 0), &ret0, sizeof(ret0));
|
|
memcpy(task_return(task, 1), &ret1, sizeof(ret1));
|
|
|
|
ASSERT(memcmp(task_arg_id(task, 0), &arg1, sizeof(arg1)) == 0);
|
|
ASSERT(memcmp(task_arg_val(task, 1), (uint8_t*) "hello",
|
|
task_arg_length(task, 1)) == 0);
|
|
ASSERT(memcmp(task_arg_id(task, 2), &arg2, sizeof(arg2)) == 0);
|
|
ASSERT(memcmp(task_arg_val(task, 3), (uint8_t*) "world",
|
|
task_arg_length(task, 3)) == 0);
|
|
|
|
ASSERT(memcmp(task_return(task, 0), &ret0, sizeof(unique_id)) == 0);
|
|
ASSERT(memcmp(task_return(task, 1), &ret1, sizeof(unique_id)) == 0);
|
|
|
|
free_task_spec(task);
|
|
PASS();
|
|
}
|
|
|
|
TEST send_task(void) {
|
|
function_id func_id = globally_unique_id();
|
|
task_spec *task = alloc_task_spec(func_id, 4, 2, 10);
|
|
*task_return(task, 1) = globally_unique_id();
|
|
int fd[2];
|
|
socketpair(AF_UNIX, SOCK_STREAM, 0, fd);
|
|
write_message(fd[0], SUBMIT_TASK, task_size(task), task);
|
|
int64_t type;
|
|
int64_t length;
|
|
uint8_t *message;
|
|
read_message(fd[1], &type, &length, &message);
|
|
task_spec *result = (task_spec *) message;
|
|
ASSERT(type == SUBMIT_TASK);
|
|
ASSERT(memcmp(task, result, task_size(task)) == 0);
|
|
ASSERT(memcmp(task, result, task_size(result)) == 0);
|
|
free(task);
|
|
free(result);
|
|
PASS();
|
|
}
|
|
|
|
TEST print_and_parse_task(void) {
|
|
task_spec *task = example_task();
|
|
|
|
UT_string *output;
|
|
utstring_new(output);
|
|
print_task(task, output);
|
|
task_spec *result = parse_task(utstring_body(output), utstring_len(output));
|
|
utstring_free(output);
|
|
|
|
ASSERT_EQ(task_size(task), task_size(result));
|
|
ASSERT(memcmp(task, result, task_size(task)) == 0);
|
|
|
|
free_task_spec(task);
|
|
free_task_spec(result);
|
|
|
|
PASS();
|
|
}
|
|
|
|
SUITE(task_tests) {
|
|
RUN_TEST(task_test);
|
|
RUN_TEST(send_task);
|
|
RUN_TEST(print_and_parse_task);
|
|
}
|
|
|
|
GREATEST_MAIN_DEFS();
|
|
|
|
int main(int argc, char **argv) {
|
|
GREATEST_MAIN_BEGIN();
|
|
RUN_SUITE(task_tests);
|
|
GREATEST_MAIN_END();
|
|
}
|