tests: fix segfault in fetch_one test when connection to ctdbd fails

[sahlberg/ctdb.git] / tests / src / ctdb_fetch_one.c

diff --git a/tests/src/ctdb_fetch_one.c b/tests/src/ctdb_fetch_one.c
index 648aa183056a6a201b00a5102bf09397e71d0313..7348aa698696408f293caa8d6fa89a448d4992bc 100644 (file)
--- a/tests/src/ctdb_fetch_one.c
+++ b/tests/src/ctdb_fetch_one.c
@@ -19,7 +19,7 @@
 */
 
 #include "includes.h"
-#include "lib/events/events.h"
+#include "lib/tevent/tevent.h"
 #include "system/filesys.h"
 #include "popt.h"
 #include "cmdline.h"
@@ -116,6 +116,11 @@ int main(int argc, const char *argv[])
 
        ctdb = ctdb_cmdline_client(ev);
 
+       if (ctdb == NULL) {
+               printf("failed to connect to ctdb deamon.\n");
+               exit(1);
+       }
+
        /* attach to a specific database */
        ctdb_db = ctdb_attach(ctdb, "test.tdb", false, 0);
        if (!ctdb_db) {