X-Git-Url: http://git.ozlabs.org/?a=blobdiff_plain;f=ccan%2Fhtable%2Ftest%2Frun.c;fp=ccan%2Fhtable%2Ftest%2Frun.c;h=ada85f95a93d803d3db678e292bb2e33a23666cc;hb=209a81909942a07cb334d6dcae7626a3ecde141d;hp=3608941d97a9de42fde9a96d60d761e00662fc8e;hpb=c910bdce167ff42aa6d9e4f1b8f905a76f0b9e75;p=ccan

diff --git a/ccan/htable/test/run.c b/ccan/htable/test/run.c
index 3608941d..ada85f95 100644
--- a/ccan/htable/test/run.c
+++ b/ccan/htable/test/run.c
@@ -105,7 +105,7 @@ int main(void)
 	void *p;
 	struct htable_iter iter;
 
-	plan_tests(38);
+	plan_tests(43);
 	for (i = 0; i < NUM_VALS; i++)
 		val[i] = i;
 	dne = i;
@@ -134,6 +134,12 @@ int main(void)
 	/* Mask should be set. */
 	ok1(check_mask(&ht, val, 1));
 
+	/* htable_pick should always return that value */
+	ok1(htable_pick(&ht, 0, NULL) == val);
+	ok1(htable_pick(&ht, 1, NULL) == val);
+	ok1(htable_pick(&ht, 0, &iter) == val);
+	ok1(get_raw_ptr(&ht, ht.table[iter.off]) == val);
+	
 	/* This should increase it again. */
 	add_vals(&ht, val, 1, 1);
 	ok1(ht.bits == 2);
@@ -210,6 +216,7 @@ int main(void)
 	htable_clear(&ht);
 
 	ok1(htable_count(&ht) == 0);
+	ok1(htable_pick(&ht, 0, NULL) == NULL);
 
 	return exit_status();
 }