1 files changed, 52 insertions, 0 deletions
diff --git a/tests/unit/mongo/wire/reply_packet_get_data.c b/tests/unit/mongo/wire/reply_packet_get_data.c
new file mode 100644
index 0000000..e22f142
--- /dev/null
+++ b/tests/unit/mongo/wire/reply_packet_get_data.c
@@ -0,0 +1,52 @@
+#include "test.h"
+#include "tap.h"
+#include "bson.h"
+#include "mongo-wire.h"
+
+#include <string.h>
+
+void
+test_mongo_wire_reply_packet_get_data (void)
+{
+  mongo_packet *p;
+  mongo_packet_header h;
+  const guint8 *data;
+  bson *b;
+
+  p = mongo_wire_packet_new ();
+  memset (&h, 0, sizeof (mongo_packet_header));
+  h.opcode = 0;
+  h.length = sizeof (mongo_packet_header);
+  mongo_wire_packet_set_header (p, &h);
+
+  ok (mongo_wire_reply_packet_get_data (NULL, &data) == FALSE,
+      "mongo_wire_reply_packet_get_data() fails with a NULL packet");
+  ok (mongo_wire_reply_packet_get_data (p, NULL) == FALSE,
+      "mongo_wire_reply_packet_get_data() fails with a NULL destination");
+  ok (mongo_wire_reply_packet_get_data (p, &data) == FALSE,
+      "mongo_wire_reply_packet_get_data() fails with a non-reply packet");
+
+  h.opcode = 1;
+  mongo_wire_packet_set_header (p, &h);
+
+  ok (mongo_wire_reply_packet_get_data (p, &data) == FALSE,
+      "mongo_wire_reply_packet_get_data() fails if the packet has "
+      "no data");
+
+  mongo_wire_packet_free (p);
+
+  p = test_mongo_wire_generate_reply (TRUE, 2, TRUE);
+
+  ok (mongo_wire_reply_packet_get_data (p, &data),
+      "mongo_wire_reply_packet_get_data() works");
+
+  b = test_bson_generate_full ();
+
+  ok (memcmp (data, bson_data (b), bson_size (b)) == 0,
+      "The returned data is correct");
+
+  bson_free (b);
+  mongo_wire_packet_free (p);
+}
+
+RUN_TEST (6, mongo_wire_reply_packet_get_data);