1 files changed, 58 insertions, 0 deletions
diff --git a/tests/unit/mongo/wire/packet_get_set_header.c b/tests/unit/mongo/wire/packet_get_set_header.c
new file mode 100644
index 0000000..38e0ea7
--- /dev/null
+++ b/tests/unit/mongo/wire/packet_get_set_header.c
@@ -0,0 +1,58 @@
+#include "tap.h"
+#include "test.h"
+#include "mongo-wire.h"
+
+#include <string.h>
+
+void
+test_mongo_wire_packet_get_set_header (void)
+{
+  mongo_packet *p;
+  mongo_packet_header ph1, ph2;
+
+  p = mongo_wire_packet_new ();
+
+  ok (mongo_wire_packet_get_header (NULL, &ph2) == FALSE,
+      "mongo_wire_packet_get_header() should fail with a NULL packet");
+  ok (mongo_wire_packet_get_header (p, NULL) == FALSE,
+      "mongo_wire_packet_get_header() should fail with a NULL header");
+  ok (mongo_wire_packet_set_header (NULL, &ph1) == FALSE,
+      "mongo_wire_packet_set_header() should fail with a NULL packet");
+  ok (mongo_wire_packet_set_header (p, NULL) == FALSE,
+      "mongo_wire_packet_set_header() should fail with a NULL header");
+
+  ok (mongo_wire_packet_get_header (p, &ph2),
+      "mongo_wire_packet_get_header() works on a fresh packet");
+  cmp_ok (ph2.length, "==", sizeof (mongo_packet_header),
+          "Initial packet length is the length of the header");
+
+  ph1.length = sizeof (mongo_packet_header);
+  ph1.id = 1;
+  ph1.resp_to = 0;
+  ph1.opcode = 1000;
+
+  memset (&ph2, 0, sizeof (mongo_packet_header));
+
+  ok (mongo_wire_packet_set_header (p, &ph1),
+      "mongo_wire_packet_set_header() works");
+  ok (mongo_wire_packet_get_header (p, &ph2),
+      "mongo_wire_packet_get_header() works");
+
+  cmp_ok (ph1.length, "==", ph2.length,
+          "Packet lengths match");
+  cmp_ok (ph1.id, "==", ph2.id,
+          "Sequence IDs match");
+  cmp_ok (ph1.resp_to, "==", ph2.resp_to,
+          "Response IDs match");
+  cmp_ok (ph1.opcode, "==", ph2.opcode,
+          "OPCodes match");
+
+  ph1.length = GINT32_TO_LE (1);
+  ok (mongo_wire_packet_set_header (p, &ph1) == FALSE,
+      "Setting a packet with length shorter than the header "
+      "returns an error");
+
+  mongo_wire_packet_free (p);
+}
+
+RUN_TEST (13, mongo_wire_packet_get_set_header);