1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 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);
|