wait_t *wait_create_msg_handler (flux_t *h, flux_msg_handler_t *wh,
const flux_msg_t *msg,
flux_msg_handler_f cb, void *arg)
{
wait_t *w = wait_create (NULL, NULL);
if (w) {
w->hand.cb = cb;
w->hand.arg = arg;
w->hand.h = h;
w->hand.w = wh;
if (msg && !(w->hand.msg = flux_msg_copy (msg, true))) {
wait_destroy (w);
errno = ENOMEM;
return NULL;
}
}
return w;
}
int main (int argc, char *argv[])
{
struct cache *cache;
struct cache_entry *e1, *e2;
json_object *o1;
json_object *o2;
wait_t *w;
int count, i;
plan (NO_PLAN);
cache_destroy (NULL);
cache_entry_destroy (NULL);
diag ("cache_destroy and cache_entry_destroy accept NULL arg");
ok ((cache = cache_create ()) != NULL,
"cache_create works");
ok (cache_count_entries (cache) == 0,
"cache contains 0 entries");
cache_destroy (cache);
/* Play with one entry.
* N.B.: json ref is NOT incremented by create or get_json.
*/
o1 = Jnew ();
Jadd_int (o1, "foo", 42);
ok ((e1 = cache_entry_create (o1)) != NULL,
"cache_entry_create works");
ok (cache_entry_get_valid (e1) == true,
"cache entry initially valid");
ok (cache_entry_get_dirty (e1) == false,
"cache entry initially not dirty");
cache_entry_set_dirty (e1, true);
ok (cache_entry_get_dirty (e1) == true,
"cache entry succcessfully set dirty");
ok ((o2 = cache_entry_get_json (e1)) != NULL,
"json retrieved from cache entry");
ok (Jget_int (o2, "foo", &i) == true && i == 42,
"expected json object found");
cache_entry_destroy (e1); /* destroys o1 */
/* Test cache entry waiters.
* N.B. waiter is destroyed when run.
*/
count = 0;
ok ((w = wait_create (wait_cb, &count)) != NULL,
"wait_create works");
ok ((e1 = cache_entry_create (NULL)) != NULL,
"cache_entry_create created empty object");
ok (cache_entry_get_valid (e1) == false,
"cache entry invalid, adding waiter");
o1 = Jnew ();
Jadd_int (o1, "foo", 42);
cache_entry_wait_valid (e1, w);
cache_entry_set_json (e1, o1);
ok (cache_entry_get_valid (e1) == true,
"cache entry set valid with one waiter");
ok (count == 1,
"waiter callback ran");
count = 0;
ok ((w = wait_create (wait_cb, &count)) != NULL,
"wait_create works");
cache_entry_set_dirty (e1, true);
ok (cache_entry_get_dirty (e1) == true,
"cache entry set dirty, adding waiter");
cache_entry_wait_notdirty (e1, w);
cache_entry_set_dirty (e1, false);
ok (cache_entry_get_dirty (e1) == false,
"cache entry set not dirty with one waiter");
ok (count == 1,
"waiter callback ran");
cache_entry_destroy (e1); /* destroys o1 */
/* Put entry in cache and test lookup, expire
*/
ok ((cache = cache_create ()) != NULL,
"cache_create works");
ok (cache_count_entries (cache) == 0,
"cache contains 0 entries");
o1 = Jnew ();
Jadd_int (o1, "foo", 42);
ok ((e1 = cache_entry_create (o1)) != NULL,
"cache_entry_create works");
cache_insert (cache, "xxx1", e1);
ok (cache_count_entries (cache) == 1,
"cache contains 1 entry after insert");
ok (cache_lookup (cache, "yyy1", 0) == NULL,
"cache_lookup of wrong hash fails");
ok ((e2 = cache_lookup (cache, "xxx1", 42)) != NULL,
"cache_lookup of correct hash works (last use=42)");
i = 0;
ok ((o2 = cache_entry_get_json (e2)) != NULL
&& Jget_int (o2, "foo", &i) == true && i == 42,
"expected json object found");
ok (cache_count_entries (cache) == 1,
"cache contains 1 entry");
ok (cache_expire_entries (cache, 43, 1) == 0,
"cache_expire_entries now=43 thresh=1 expired 0");
ok (cache_count_entries (cache) == 1,
"cache contains 1 entry");
ok (cache_expire_entries (cache, 44, 1) == 1,
"cache_expire_entries now=44 thresh=1 expired 1");
ok (cache_count_entries (cache) == 0,
"cache contains 0 entries");
cache_destroy (cache);
done_testing ();
return (0);
}
int main (int argc, char *argv[])
{
waitqueue_t *q;
waitqueue_t *q2;
wait_t *w;
flux_msg_t *msg;
int count = 0;
int i;
plan (NO_PLAN);
q = wait_queue_create ();
q2 = wait_queue_create ();
ok (q && q2,
"wait_queue_create works");
ok (wait_queue_length (q) == 0 && wait_queue_length (q2) == 0,
"wait_queue_length on brandnew waitqueue_t returns zero");
msg = flux_msg_create (FLUX_MSGTYPE_REQUEST);
ok (msg != NULL,
"flux_msg_create works");
w = wait_create (NULL, NULL, msg, msghand, &count);
ok (w != NULL,
"wait_create works");
flux_msg_destroy (msg);
wait_addqueue (q, w);
wait_addqueue (q2, w);
ok (wait_queue_length (q) == 1 && wait_queue_length (q2) == 1,
"wait_addqueue can add wait_t to a two queues");
wait_runqueue (q);
ok (wait_queue_length (q) == 0 && wait_queue_length (q2) == 1,
"wait_runqueue dequeued wait_t from first queue");
wait_runqueue (q2);
ok (wait_queue_length (q) == 0 && wait_queue_length (q2) == 0,
"wait_runqueue dequeued wait_t from second queue");
ok (count == 1,
"wait_runqueue ran the wait_t once");
for (i = 0; i < 20; i++) {
char *s = xasprintf ("%d", i);
msg = flux_msg_create (FLUX_MSGTYPE_REQUEST);
if (!msg)
break;
if (flux_msg_enable_route (msg) < 0
|| flux_msg_push_route (msg, s) < 0)
break;
w = wait_create (NULL, NULL, msg, msghand, &count);
if (!w)
break;
wait_addqueue (q, w);
free (s);
}
ok (wait_queue_length (q) == 20,
"wait_addqueue 20x works");
wait_destroy_match (q, msgcmp, NULL);
ok (wait_queue_length (q) == 17,
"wait_destroy_match on sender works");
wait_destroy_match (q, msgcmp2, NULL);
ok (wait_queue_length (q) == 0,
"all-match wait_destroy_match works");
wait_queue_destroy (q);
wait_queue_destroy (q2);
done_testing ();
return (0);
}
