X-Git-Url: http://git.indexdata.com/?a=blobdiff_plain;f=test%2Ftestclient.c;h=1e9762f4f705b2c67797e763f83da9603be26326;hb=83574dcb462d530758e21aea0a8a843cd0dce60f;hp=0afa9143108eb72dc608dce0219d336d0208827d;hpb=e868e46c150fd38e72a1d087ce5306bb7c715847;p=idzebra-moved-to-github.git

diff --git a/test/testclient.c b/test/testclient.c
index 0afa914..1e9762f 100644
--- a/test/testclient.c
+++ b/test/testclient.c
@@ -1,41 +1,122 @@
 /*
- * $Id: testclient.c,v 1.1 2002-10-22 10:47:48 adam Exp $
+ * $Id: testclient.c,v 1.9 2003-01-13 22:28:18 adam Exp $
  *
- * Synchronous single-target client doing search (but no yet retrieval)
+ * Z39.50 client specifically for Zebra testing.
  */
 
 #include <stdlib.h>
 #include <stdio.h>
 #include <yaz/xmalloc.h>
+#include <yaz/options.h>
 #include <yaz/zoom.h>
 
+char *prog = "testclient";
+
 int main(int argc, char **argv)
 {
     ZOOM_connection z;
     ZOOM_resultset r;
     int error;
     const char *errmsg, *addinfo;
+    char *query = 0;
+    char *target = 0;
+    char *arg;
+    int delay_sec = 0;
+    int expected_hits=-1;
+    int ret;
+    int retrieve_number = 0;
+    int retrieve_offset = 0;
+    char *format = 0;
+    int pos;
+    int check_count = -1;
+    int exit_code = 0;
 
-    if (argc != 3)
+    while ((ret = options("d:n:o:f:c:", argv, argc, &arg)) != -2)
     {
-        fprintf (stderr, "usage:\n%s target query\n", *argv);
-        fprintf (stderr, " eg.  bagel.indexdata.dk/gils computer\n");
-        exit (1);
+        switch (ret)
+        {
+        case 0:
+            if (!target)
+                target = xstrdup(arg);
+            else if (!query)
+                query = xstrdup(arg);
+            break;
+        case 'd':
+            delay_sec = atoi(arg);
+            break;
+        case 'n':
+            retrieve_number = atoi(arg);
+            break;
+        case 'o':
+            retrieve_offset = atoi(arg);
+            break;
+        case 'f':
+            format = xstrdup(arg);
+            break;
+        case 'c':
+	    check_count = atoi(arg);
+	    break;
+        default:
+            printf ("%s: unknown option %s\n", prog, arg);
+            printf ("usage:\n%s [options] target query \n", prog);
+            printf (" eg.  indexdata.dk/gils computer\n");
+            exit (1);
+        }
     }
-    z = ZOOM_connection_new (argv[1], 0);
+
+    if (!target || !target)
+    {
+        printf ("%s: missing target/query\n", prog);
+        printf ("usage:\n%s [options] target query \n", prog);
+        printf (" eg.  bagel.indexdata.dk/gils computer\n");
+        printf ("Options:\n");
+        printf (" -n num       number of records to fetch. Default: 0.\n");
+        printf (" -o off       offset for records - counting from 0.\n");
+        printf (" -f format    set record syntax. Default: none\n");
+        printf (" -d sec       delay a number of seconds before exit.\n");
+        exit (3);
+    }
+    z = ZOOM_connection_new (target, 0);
     
     if ((error = ZOOM_connection_error(z, &errmsg, &addinfo)))
     {
-	fprintf (stderr, "Error: %s (%d) %s\n", errmsg, error, addinfo);
+	printf ("Error: %s (%d) %s\n", errmsg, error, addinfo);
 	exit (2);
     }
 
-    r = ZOOM_connection_search_pqf (z, argv[2]);
+    r = ZOOM_connection_search_pqf (z, query);
     if ((error = ZOOM_connection_error(z, &errmsg, &addinfo)))
-	fprintf (stderr, "Error: %s (%d) %s\n", errmsg, error, addinfo);
+    {
+	printf ("Error: %s (%d) %s\n", errmsg, error, addinfo);
+	if (check_count != -1)
+            exit_code = 10;
+    }
     else
+    {
 	printf ("Result count: %d\n", ZOOM_resultset_size(r));
+	if (check_count != -1 && check_count != ZOOM_resultset_size(r))
+	{
+	    printf("Wrong number of hits, expected %d, got %d\n",
+			    expected_hits, ZOOM_resultset_size(r) );
+	    exit(3);
+	}
+    }
+    if (format)
+        ZOOM_resultset_option_set(r, "preferredRecordSyntax", format);
+    for (pos = 0; pos < retrieve_number; pos++)
+    {
+        int len;
+        const char *rec =
+            ZOOM_record_get(
+                ZOOM_resultset_record(r, pos + retrieve_offset),
+                "render", &len);
+        
+        if (rec)
+            fwrite (rec, 1, len, stdout);
+    }
+    if (delay_sec > 0)
+	sleep(delay_sec);
     ZOOM_resultset_destroy (r);
     ZOOM_connection_destroy (z);
-    exit (0);
+    exit (exit_code);
 }