X-Git-Url: http://git.ozlabs.org/?p=petitboot;a=blobdiff_plain;f=ui%2Ftest%2Fdiscover-test.c;h=ae43b1e524b30421e694ed4fee4d04dcb309a8b3;hp=f3ef746de2382488fc40c469dcbe0e39c37383da;hb=9939af2652ce479645eaa78e891ee06f33845a99;hpb=69c459db80abece18b5557d9b8a8098a88329c28

diff --git a/ui/test/discover-test.c b/ui/test/discover-test.c
index f3ef746..ae43b1e 100644
--- a/ui/test/discover-test.c
+++ b/ui/test/discover-test.c
@@ -44,15 +44,18 @@ static struct discover_client_ops client_ops = {
 int main(void)
 {
 	struct discover_client *client;
+	struct waitset *waitset;
 
-	client = discover_client_init(&client_ops, NULL);
+	waitset = waitset_create(NULL);
+
+	client = discover_client_init(waitset, &client_ops, NULL);
 	if (!client)
 		return -1;
 
 	for (;;) {
 		int rc;
 
-		rc = discover_client_process(client);
+		rc = waiter_poll(waitset);
 		if (rc)
 			break;
 	}