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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
|
#include "test.h"
#include "tap.h"
#include "mongo-wire.h"
#include <string.h>
void
test_mongo_wire_cmd_query (void)
{
bson *q, *s, *tmp;
mongo_packet *p;
mongo_packet_header hdr;
const guint8 *data;
gint32 data_size;
bson_cursor *c;
gint32 pos;
q = test_bson_generate_full ();
s = bson_new ();
bson_append_boolean (s, "_id", TRUE);
bson_append_boolean (s, "double", TRUE);
bson_finish (s);
tmp = bson_new ();
ok (mongo_wire_cmd_query (1, NULL, 0, 0, 0, q, s) == NULL,
"mongo_wire_cmd_query() fails whith a NULL namespace");
ok (mongo_wire_cmd_query (1, "test.ns", 0, 0, 0, NULL, s) == NULL,
"mongo_wire_cmd_query() fails with a NULL query");
ok (mongo_wire_cmd_query (1, "test.ns", 0, 0, 0, tmp, s) == NULL,
"mongo_wire_cmd_query() fails with an unfinished query");
ok (mongo_wire_cmd_query (1, "test.ns", 0, 0, 0, q, tmp) == NULL,
"mongo_wire_cmd_query() fails with an unfinished selector");
bson_free (tmp);
ok ((p = mongo_wire_cmd_query (1, "test.ns", 0, 0, 10, q, NULL)) != NULL,
"mongo_wire_cmd_query() works with a NULL selector");
mongo_wire_packet_get_header (p, &hdr);
cmp_ok ((data_size = mongo_wire_packet_get_data (p, &data)), "!=", -1,
"Packet data size looks fine");
cmp_ok (hdr.length, "==", sizeof (mongo_packet_header) + data_size,
"Packet header length is OK");
cmp_ok (hdr.id, "==", 1, "Packet request ID is ok");
cmp_ok (hdr.resp_to, "==", 0, "Packet reply ID is ok");
/* pos = zero + collection_name + NULL + skip + ret */
pos = sizeof (gint32) + strlen ("test.ns") + 1 + sizeof (gint32) * 2;
ok ((tmp = bson_new_from_data (data + pos,
bson_stream_doc_size (data, pos) - 1)) != NULL,
"Packet contains a valid BSON query document");
bson_finish (tmp);
ok ((c = bson_find (tmp, "int32")) != NULL,
"BSON contains 'int32'");
cmp_ok (bson_cursor_type (c), "==", BSON_TYPE_INT32,
"int32 has correct type");
bson_cursor_next (c);
cmp_ok (bson_cursor_type (c), "==", BSON_TYPE_INT64,
"next element has correct type too");
ok (bson_cursor_next (c) == FALSE,
"No more data after the update BSON object");
bson_cursor_free (c);
cmp_ok (hdr.length, "==", sizeof (mongo_packet_header) + pos +
bson_size (q),
"Packet header lenght is correct");
bson_free (tmp);
mongo_wire_packet_free (p);
/*
* Test again with a selector document
*/
ok ((p = mongo_wire_cmd_query (1, "test.ns", 0, 0, 10, q, s)) != NULL,
"mongo_wire_cmd_query() works with a NULL selector");
mongo_wire_packet_get_header (p, &hdr);
cmp_ok ((data_size = mongo_wire_packet_get_data (p, &data)), "!=", -1,
"Packet data size looks fine");
cmp_ok (hdr.length, "==", sizeof (mongo_packet_header) + data_size,
"Packet header length is OK");
cmp_ok (hdr.id, "==", 1, "Packet request ID is ok");
cmp_ok (hdr.resp_to, "==", 0, "Packet reply ID is ok");
/* pos = zero + collection_name + NULL + skip + ret */
pos = sizeof (gint32) + strlen ("test.ns") + 1 + sizeof (gint32) * 2;
ok ((tmp = bson_new_from_data (data + pos,
bson_stream_doc_size (data, pos) - 1)) != NULL,
"Packet contains a valid BSON query document");
bson_finish (tmp);
pos += bson_size (tmp);
bson_free (tmp);
bson_free (q);
bson_free (s);
ok ((s = bson_new_from_data (data + pos,
bson_stream_doc_size (data, pos) - 1)) != NULL,
"Packet contains a valid BSON selector document");
bson_finish (s);
ok ((c = bson_find (s, "_id")) != NULL,
"BSON contains '_id'");
cmp_ok (bson_cursor_type (c), "==", BSON_TYPE_BOOLEAN,
"_id has correct type");
bson_cursor_next (c);
cmp_ok (bson_cursor_type (c), "==", BSON_TYPE_BOOLEAN,
"next element has correct type too");
bson_cursor_free (c);
bson_free (s);
mongo_wire_packet_free (p);
}
RUN_TEST (25, mongo_wire_cmd_query);
|
