hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-06 10:31:39 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix failing task table test (#924)

Browse source
This commit is contained in:
Stephanie Wang 2017-09-03 22:41:38 -07:00 • committed by Robert Nishihara
parent 2c19ae97a3
commit ae0212b399
1 changed files with 13 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
src/common/test/task_table_tests.cc
View file

@ -93,6 +93,17 @@ void add_success_callback(TaskID task_id, void *context) {
(void *) add_lookup_context);
}
void subscribe_success_callback(TaskID task_id, void *context) {
DBHandle *db = (DBHandle *) context;
RetryInfo retry = {
.num_retries = 5,
.timeout = 1000,
.fail_callback = add_lookup_fail_callback,
};
task_table_add_task(db, Task_copy(add_lookup_task), &retry,
add_success_callback, (void *) db);
}
TEST add_lookup_test(void) {
add_lookup_task = example_task(1, 1, TASK_STATUS_WAITING);
g_loop = event_loop_create();
@ -104,10 +115,9 @@ TEST add_lookup_test(void) {
.timeout = 1000,
.fail_callback = add_lookup_fail_callback,
};
/* Wait for subscription to succeed before adding the task. */
task_table_subscribe(db, NIL_ID, TASK_STATUS_WAITING, NULL, NULL, &retry,
NULL, NULL);
task_table_add_task(db, Task_copy(add_lookup_task), &retry,
add_success_callback, (void *) db);
subscribe_success_callback, (void *) db);
/* Disconnect the database to see if the lookup times out. */
event_loop_run(g_loop);
db_disconnect(db);