summaryrefslogtreecommitdiff
path: root/tests/unit/mongo/wire/reply_packet_get_data.c
blob: e22f1420b469f279915b93491b906b58f7b59c6d (plain)
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
#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);