1 files changed, 54 insertions, 0 deletions
diff --git a/tests/unit/mongo/wire/reply_packet_get_header.c b/tests/unit/mongo/wire/reply_packet_get_header.c
new file mode 100644
index 0000000..36b548c
--- /dev/null
+++ b/tests/unit/mongo/wire/reply_packet_get_header.c
@@ -0,0 +1,54 @@
+#include "test.h"
+#include "tap.h"
+#include "mongo-wire.h"
+
+#include <string.h>
+
+void
+test_mongo_wire_reply_packet_get_header (void)
+{
+  mongo_packet *p;
+  mongo_packet_header h;
+  mongo_reply_packet_header rh;
+
+  p = mongo_wire_packet_new ();
+  memset (&h, 0, sizeof (mongo_packet_header));
+  h.opcode = 1;
+  h.length = sizeof (mongo_packet_header);
+
+  mongo_wire_packet_set_header (p, &h);
+
+  ok (mongo_wire_reply_packet_get_header (NULL, &rh) == FALSE,
+      "mongo_wire_reply_packet_get_header() fails with a NULL packet");
+  ok (mongo_wire_reply_packet_get_header (p, NULL) == FALSE,
+      "mongo_wire_reply_packet_get_header() fails with a NULL header");
+
+  ok (mongo_wire_reply_packet_get_header (p, &rh) == FALSE,
+      "mongo_wire_reply_packet_get_header() fails if the packet has "
+      "no reply header");
+
+  h.opcode = 2;
+  mongo_wire_packet_set_header (p, &h);
+  ok (mongo_wire_reply_packet_get_header (p, &rh) == FALSE,
+      "mongo_wire_reply_packet_get_header() fails if the packet is "
+      "not a reply packet");
+
+  mongo_wire_packet_free (p);
+
+  p = test_mongo_wire_generate_reply (TRUE, 0, FALSE);
+
+  ok (mongo_wire_reply_packet_get_header (p, &rh),
+      "mongo_wire_reply_packet_get_header() works");
+  cmp_ok (rh.flags, "==", 0,
+          "Reply flags are correct");
+  ok (rh.cursor_id == (gint64)12345,
+      "Cursor ID is correct");
+  cmp_ok (rh.start, "==", 0,
+          "Reply start document is OK");
+  cmp_ok (rh.returned, "==", 0,
+          "Number of documents returned is OK");
+
+  mongo_wire_packet_free (p);
+}
+
+RUN_TEST (9, mongo_wire_reply_packet_get_header);