mirror of
https://github.com/wassname/ray.git
synced 2026-07-06 05:16:30 +08:00
7237ec4124
* Add Python and Redis submodules, and remove old third-party modules
* Update VS projects (WARNING: references files that do not exist yet)
* Update code & add shims for APIs except AF_UNIX/{send,recv}msg()
* Minor style changes.
41 lines
1.1 KiB
C
41 lines
1.1 KiB
C
#ifndef TEST_COMMON_H
|
|
#define TEST_COMMON_H
|
|
|
|
#include "hiredis/hiredis.h"
|
|
|
|
#include "task.h"
|
|
|
|
static task_spec *example_task_spec(void) {
|
|
task_id parent_task_id = globally_unique_id();
|
|
function_id func_id = globally_unique_id();
|
|
task_spec *task =
|
|
start_construct_task_spec(parent_task_id, 0, func_id, 2, 1, 0);
|
|
task_args_add_ref(task, globally_unique_id());
|
|
task_args_add_ref(task, globally_unique_id());
|
|
finish_construct_task_spec(task);
|
|
return task;
|
|
}
|
|
|
|
static task *example_task(void) {
|
|
task_spec *spec = example_task_spec();
|
|
task *instance = alloc_task(spec, TASK_STATUS_WAITING, NIL_ID);
|
|
free_task_spec(spec);
|
|
return instance;
|
|
}
|
|
|
|
/* Flush redis. */
|
|
static void flushall_redis() {
|
|
redisContext *context = redisConnect("127.0.0.1", 6379);
|
|
freeReplyObject(redisCommand(context, "FLUSHALL"));
|
|
redisFree(context);
|
|
}
|
|
|
|
/* Cleanup method for running tests with the greatest library.
|
|
* Runs the test, then clears the Redis database. */
|
|
#define RUN_REDIS_TEST(test) \
|
|
flushall_redis(); \
|
|
RUN_TEST(test); \
|
|
flushall_redis();
|
|
|
|
#endif /* TEST_COMMON */
|