4 files changed, 165 insertions, 0 deletions
diff --git a/tests/unit/mongo/sync-cursor/sync_cursor_free.c b/tests/unit/mongo/sync-cursor/sync_cursor_free.c
new file mode 100644
index 0000000..bd01cb5
--- /dev/null
+++ b/tests/unit/mongo/sync-cursor/sync_cursor_free.c
@@ -0,0 +1,34 @@
+#include "test.h"
+#include "mongo.h"
+#include "config.h"
+
+#include "libmongo-private.h"
+
+#include <errno.h>
+
+void
+test_mongo_sync_cursor_free (void)
+{
+  mongo_sync_connection *conn;
+  mongo_packet *p;
+  mongo_sync_cursor *c;
+
+  test_env_setup ();
+
+  p = mongo_wire_packet_new ();
+  conn = test_make_fake_sync_conn (-1, FALSE);
+
+  c = mongo_sync_cursor_new (conn, config.ns, p);
+
+  errno = 0;
+  mongo_sync_cursor_free (NULL);
+  ok (errno == ENOTCONN,
+      "mongo_sync_cursor_free(NULL) sets errno to ENOTCONN");
+  mongo_sync_cursor_free (c);
+  pass ("mongo_sync_cursor_free() works");
+
+  mongo_sync_disconnect (conn);
+  test_env_free ();
+}
+
+RUN_TEST (2, mongo_sync_cursor_free);
diff --git a/tests/unit/mongo/sync-cursor/sync_cursor_get_data.c b/tests/unit/mongo/sync-cursor/sync_cursor_get_data.c
new file mode 100644
index 0000000..0dd391c
--- /dev/null
+++ b/tests/unit/mongo/sync-cursor/sync_cursor_get_data.c
@@ -0,0 +1,51 @@
+#include "test.h"
+#include "mongo.h"
+#include "config.h"
+
+#include "libmongo-private.h"
+
+#include <errno.h>
+
+void
+test_mongo_sync_cursor_get_data (void)
+{
+  mongo_sync_connection *conn;
+  mongo_packet *p;
+  bson *b;
+  mongo_sync_cursor *c;
+
+  test_env_setup ();
+
+  p = test_mongo_wire_generate_reply (TRUE, 4, TRUE);
+  conn = test_make_fake_sync_conn (-1, FALSE);
+
+  c = mongo_sync_cursor_new (conn, config.ns, p);
+
+  errno = 0;
+  b = mongo_sync_cursor_get_data (NULL);
+  ok (b == NULL && errno == EINVAL,
+      "mongo_sync_cursor_get_data(NULL) should fail");
+
+  b = mongo_sync_cursor_get_data (c);
+  ok (b == NULL,
+      "mongo_sync_cursor_get_data() should fail without _cursor_next()");
+
+  mongo_sync_cursor_next (c);
+  b = mongo_sync_cursor_get_data (c);
+  ok (b != NULL,
+      "mongo_sync_cursor_get_data() works");
+
+  c->offset = 5;
+
+  errno = 0;
+  b = mongo_sync_cursor_get_data (c);
+  ok (b == NULL && errno == ERANGE,
+      "mongo_sync_cursor_get_data() should fail if the cursor is "
+      "out of range");
+
+  mongo_sync_cursor_free (c);
+  mongo_sync_disconnect (conn);
+  test_env_free ();
+}
+
+RUN_TEST (4, mongo_sync_cursor_get_data);
diff --git a/tests/unit/mongo/sync-cursor/sync_cursor_new.c b/tests/unit/mongo/sync-cursor/sync_cursor_new.c
new file mode 100644
index 0000000..642d826
--- /dev/null
+++ b/tests/unit/mongo/sync-cursor/sync_cursor_new.c
@@ -0,0 +1,40 @@
+#include "test.h"
+#include "mongo.h"
+#include "config.h"
+
+#include "libmongo-private.h"
+
+#include <errno.h>
+
+void
+test_mongo_sync_cursor_new (void)
+{
+  mongo_sync_connection *conn;
+  mongo_packet *p;
+  mongo_sync_cursor *c;
+
+  test_env_setup ();
+
+  p = mongo_wire_packet_new ();
+  conn = test_make_fake_sync_conn (-1, FALSE);
+
+  c = mongo_sync_cursor_new (conn, config.ns, NULL);
+  ok (c == NULL,
+      "mongo_sync_cursor_new() fails with a NULL packet");
+  c = mongo_sync_cursor_new (conn, NULL, p);
+  ok (c == NULL,
+      "mongo_sync_cursor_new() fails with a NULL namespace");
+  c = mongo_sync_cursor_new (NULL, config.ns, p);
+  ok (c == NULL,
+      "mongo_sync_cursor_new() fails with a NULL connection");
+
+  c = mongo_sync_cursor_new (conn, config.ns, p);
+  ok (c != NULL,
+      "mongo_sync_cursor_new() works");
+
+  mongo_sync_cursor_free (c);
+  mongo_sync_disconnect (conn);
+  test_env_free ();
+}
+
+RUN_TEST (4, mongo_sync_cursor_new);
diff --git a/tests/unit/mongo/sync-cursor/sync_cursor_next.c b/tests/unit/mongo/sync-cursor/sync_cursor_next.c
new file mode 100644
index 0000000..442df96
--- /dev/null
+++ b/tests/unit/mongo/sync-cursor/sync_cursor_next.c
@@ -0,0 +1,40 @@
+#include "test.h"
+#include "mongo.h"
+#include "config.h"
+
+#include "libmongo-private.h"
+
+#include <errno.h>
+
+void
+test_mongo_sync_cursor_next (void)
+{
+  mongo_sync_connection *conn;
+  mongo_packet *p;
+  mongo_sync_cursor *c;
+  gboolean r = TRUE;
+  gint i;
+
+  test_env_setup ();
+
+  p = test_mongo_wire_generate_reply (TRUE, 2, TRUE);
+  conn = test_make_fake_sync_conn (-1, FALSE);
+
+  c = mongo_sync_cursor_new (conn, config.ns, p);
+
+  ok (mongo_sync_cursor_next (NULL) == FALSE,
+      "mongo_sync_cursor_next() should fail with a NULL cursor");
+  for (i = 0; i < 2; i++)
+    r &= mongo_sync_cursor_next (c);
+
+  ok (r == TRUE,
+      "mongo_sync_cursor_next() works");
+  ok (mongo_sync_cursor_next (c) == FALSE,
+      "mongo_sync_cursor_next() should fail past the end of the resultset");
+
+  mongo_sync_cursor_free (c);
+  mongo_sync_disconnect (conn);
+  test_env_free ();
+}
+
+RUN_TEST (3, mongo_sync_cursor_next);