discover-client: interact directly with waitset

[petitboot] / ui / test / discover-test.c

diff --git a/ui/test/discover-test.c b/ui/test/discover-test.c
index f3ef746de2382488fc40c469dcbe0e39c37383da..ae43b1e524b30421e694ed4fee4d04dcb309a8b3 100644 (file)
--- 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;
        }