request_t *
mock_server_receives_query (mock_server_t *server,
const char *ns,
mongoc_query_flags_t flags,
uint32_t skip,
uint32_t n_return,
const char *query_json,
const char *fields_json)
{
request_t *request;
request = mock_server_receives_request (server);
if (request && !request_matches_query (request,
ns,
flags,
skip,
n_return,
query_json,
fields_json,
false)) {
request_destroy (request);
return NULL;
}
return request;
}
request_t *mock_server_receives_kill_cursors (mock_server_t *server,
int64_t cursor_id)
{
request_t *request;
request = mock_server_receives_request (server);
if (request && !request_matches_kill_cursors (request, cursor_id)) {
request_destroy (request);
return NULL;
}
return request;
}
static void
test_exhaust (void)
{
mock_server_t *server;
mongoc_client_t *client;
mongoc_collection_t *collection;
mongoc_cursor_t *cursor;
request_t *request;
future_t *future;
const bson_t *doc;
bson_error_t error;
server = mock_server_with_autoismaster (WIRE_VERSION_FIND_CMD);
mock_server_run (server);
client = mongoc_client_new_from_uri (mock_server_get_uri (server));
collection = mongoc_client_get_collection (client, "db", "collection");
cursor = mongoc_collection_find_with_opts (collection,
tmp_bson (NULL),
NULL,
tmp_bson ("{'exhaust': true}"));
future = future_cursor_next (cursor, &doc);
/* Find, getMore and killCursors commands spec: "The find command does not
* support the exhaust flag from OP_QUERY. Drivers that support exhaust MUST
* fallback to existing OP_QUERY wire protocol messages."
*/
request = mock_server_receives_request (server);
mock_server_replies_to_find (
request, MONGOC_QUERY_SLAVE_OK | MONGOC_QUERY_EXHAUST,
0, 0, "db.collection", "{}", false /* is_command */);
ASSERT (future_get_bool (future));
ASSERT_OR_PRINT (!mongoc_cursor_error (cursor, &error), error);
request_destroy (request);
future_destroy (future);
mongoc_cursor_destroy (cursor);
mongoc_collection_destroy (collection);
mongoc_client_destroy (client);
mock_server_destroy (server);
}
request_t *
mock_server_receives_bulk_insert (mock_server_t *server,
const char *ns,
mongoc_insert_flags_t flags,
int n)
{
request_t *request;
request = mock_server_receives_request (server);
if (request && !request_matches_bulk_insert (request,
ns,
flags,
n)) {
request_destroy (request);
return NULL;
}
return request;
}
request_t *
mock_server_receives_getmore (mock_server_t *server,
const char *ns,
uint32_t n_return,
int64_t cursor_id)
{
request_t *request;
request = mock_server_receives_request (server);
if (request && !request_matches_getmore (request,
ns,
n_return,
cursor_id)) {
request_destroy (request);
return NULL;
}
return request;
}
request_t *
mock_server_receives_delete (mock_server_t *server,
const char *ns,
mongoc_remove_flags_t flags,
const char *selector_json)
{
request_t *request;
request = mock_server_receives_request (server);
if (request && !request_matches_delete (request,
ns,
flags,
selector_json)) {
request_destroy (request);
return NULL;
}
return request;
}
request_t *
mock_server_receives_command (mock_server_t *server,
const char *database_name,
mongoc_query_flags_t flags,
const char *command_json,
...)
{
va_list args;
char *formatted_command_json = NULL;
char *ns;
request_t *request;
va_start (args, command_json);
if (command_json) {
formatted_command_json = bson_strdupv_printf (command_json, args);
}
va_end (args);
ns = bson_strdup_printf ("%s.$cmd", database_name);
request = mock_server_receives_request (server);
if (request && !request_matches_query (request,
ns,
flags,
0,
1,
formatted_command_json,
NULL,
true)) {
request_destroy (request);
request = NULL;
}
bson_free (formatted_command_json);
bson_free (ns);
return request;
}
