static void
test_find_and_modify_opts (void)
{
mock_server_t *server;
mongoc_client_t *client;
mongoc_collection_t *collection;
bson_error_t error;
mongoc_find_and_modify_opts_t *opts;
future_t *future;
request_t *request;
server = mock_server_with_autoismaster (0);
mock_server_run (server);
client = mongoc_client_new_from_uri (mock_server_get_uri (server));
collection = mongoc_client_get_collection (client, "db", "collection");
opts = mongoc_find_and_modify_opts_new ();
assert (mongoc_find_and_modify_opts_set_max_time_ms (opts, 42));
assert (mongoc_find_and_modify_opts_append (opts, tmp_bson ("{'foo': 1}")));
future = future_collection_find_and_modify_with_opts (
collection, tmp_bson ("{}"), opts, NULL, &error);
request = mock_server_receives_command (
server, "db", MONGOC_QUERY_NONE,
"{'findAndModify': 'collection', 'maxTimeMS': 42, 'foo': 1}");
mock_server_replies_ok_and_destroys (request);
ASSERT_OR_PRINT (future_get_bool (future), error);
future_destroy (future);
mongoc_find_and_modify_opts_destroy (opts);
mongoc_collection_destroy (collection);
mongoc_client_destroy (client);
mock_server_destroy (server);
}
static void
test_find_and_modify_write_concern_wire_32_failure (void *context)
{
mongoc_collection_t *collection;
mongoc_client_t *client;
bson_error_t error;
mongoc_find_and_modify_opts_t *opts;
bson_t reply;
bson_t query = BSON_INITIALIZER;
bson_t *update;
bool success;
mongoc_write_concern_t *wc;
client = test_framework_client_new ();
collection = get_test_collection (client, "writeFailure");
wc = mongoc_write_concern_new ();
mongoc_write_concern_set_w (wc, 42);
mongoc_collection_set_write_concern (collection, wc);
/* Find Zlatan Ibrahimovic, the striker */
BSON_APPEND_UTF8 (&query, "firstname", "Zlatan");
BSON_APPEND_UTF8 (&query, "lastname", "Ibrahimovic");
BSON_APPEND_UTF8 (&query, "profession", "Football player");
BSON_APPEND_INT32 (&query, "age", 34);
BSON_APPEND_INT32 (
&query, "goals", (16 + 35 + 23 + 57 + 16 + 14 + 28 + 84) + (1 + 6 + 62));
/* Add his football position */
update = BCON_NEW ("$set", "{", "position", BCON_UTF8 ("striker"), "}");
opts = mongoc_find_and_modify_opts_new ();
mongoc_find_and_modify_opts_set_update (opts, update);
/* Create the document if it didn't exist, and return the updated document */
mongoc_find_and_modify_opts_set_flags (
opts, MONGOC_FIND_AND_MODIFY_UPSERT | MONGOC_FIND_AND_MODIFY_RETURN_NEW);
success = mongoc_collection_find_and_modify_with_opts (
collection, &query, opts, &reply, &error);
ASSERT (!success);
ASSERT_ERROR_CONTAINS (
error, MONGOC_ERROR_WRITE_CONCERN, 100, "Write Concern error:");
bson_destroy (&reply);
bson_destroy (update);
bson_destroy (&query);
mongoc_find_and_modify_opts_destroy (opts);
mongoc_collection_drop (collection, NULL);
mongoc_collection_destroy (collection);
mongoc_client_destroy (client);
}
static future_t *
find_and_modify (func_ctx_t *ctx, bson_t *cmd)
{
mongoc_find_and_modify_opts_t *fam;
BSON_APPEND_UTF8 (cmd, "findAndModify", "collection");
fam = mongoc_find_and_modify_opts_new ();
bson_concat (&fam->extra, ctx->opts);
/* destroy the mongoc_find_and_modify_opts_t later */
ctx->data = fam;
ctx->destructor = find_and_modify_cleanup;
return future_collection_find_and_modify_with_opts (
ctx->collection, tmp_bson ("{}"), fam, NULL, &ctx->error);
}
void fam_flags(mongoc_collection_t *collection)
{
mongoc_find_and_modify_opts_t *opts;
bson_t reply;
bson_error_t error;
bson_t query = BSON_INITIALIZER;
bson_t *update;
bool success;
/* Find Zlatan Ibrahimovic, the striker */
BSON_APPEND_UTF8 (&query, "firstname", "Zlatan");
BSON_APPEND_UTF8 (&query, "lastname", "Ibrahimovic");
BSON_APPEND_UTF8 (&query, "profession", "Football player");
BSON_APPEND_INT32 (&query, "age", 34);
BSON_APPEND_INT32 (&query, "goals", (16+35+23+57+16+14+28+84)+(1+6+62));
/* Add his football position */
update = BCON_NEW ("$set", "{",
"position", BCON_UTF8 ("striker"),
"}");
opts = mongoc_find_and_modify_opts_new ();
mongoc_find_and_modify_opts_set_update (opts, update);
/* Create the document if it didn't exist, and return the updated document */
mongoc_find_and_modify_opts_set_flags (opts, MONGOC_FIND_AND_MODIFY_UPSERT|MONGOC_FIND_AND_MODIFY_RETURN_NEW);
success = mongoc_collection_find_and_modify_with_opts (collection, &query, opts, &reply, &error);
if (success) {
char *str;
str = bson_as_json (&reply, NULL);
printf ("%s\n", str);
bson_free (str);
} else {
fprintf(stderr, "Got error: \"%s\" on line %d\n", error.message, __LINE__);
}
bson_destroy (&reply);
bson_destroy (update);
bson_destroy (&query);
mongoc_find_and_modify_opts_destroy (opts);
}
static void
test_find_and_modify_bypass (bool bypass)
{
mongoc_collection_t *collection;
mongoc_client_t *client;
mock_server_t *server;
request_t *request;
future_t *future;
bson_error_t error;
bson_t *update;
bson_t doc = BSON_INITIALIZER;
bson_t reply;
mongoc_find_and_modify_opts_t *opts;
server = mock_server_new ();
mock_server_run (server);
client = mongoc_client_new_from_uri (mock_server_get_uri (server));
ASSERT (client);
collection =
mongoc_client_get_collection (client, "test", "test_find_and_modify");
auto_ismaster (server,
WIRE_VERSION_FAM_WRITE_CONCERN, /* max_wire_version */
48000000, /* max_message_size */
16777216, /* max_bson_size */
1000); /* max_write_batch_size */
BSON_APPEND_INT32 (&doc, "superduper", 77889);
update = BCON_NEW ("$set", "{", "superduper", BCON_INT32 (1234), "}");
opts = mongoc_find_and_modify_opts_new ();
mongoc_find_and_modify_opts_set_bypass_document_validation (opts, bypass);
mongoc_find_and_modify_opts_set_update (opts, update);
mongoc_find_and_modify_opts_set_flags (opts,
MONGOC_FIND_AND_MODIFY_RETURN_NEW);
future = future_collection_find_and_modify_with_opts (
collection, &doc, opts, &reply, &error);
if (bypass) {
request = mock_server_receives_command (
server,
"test",
MONGOC_QUERY_NONE,
"{ 'findAndModify' : 'test_find_and_modify', "
"'query' : { 'superduper' : 77889 },"
"'update' : { '$set' : { 'superduper' : 1234 } },"
"'new' : true,"
"'bypassDocumentValidation' : true }");
} else {
request = mock_server_receives_command (
server,
"test",
MONGOC_QUERY_NONE,
"{ 'findAndModify' : 'test_find_and_modify', "
"'query' : { 'superduper' : 77889 },"
"'update' : { '$set' : { 'superduper' : 1234 } },"
"'new' : true,"
"'bypassDocumentValidation' : false }");
}
mock_server_replies_simple (request, "{ 'value' : null, 'ok' : 1 }");
ASSERT_OR_PRINT (future_get_bool (future), error);
future_destroy (future);
mongoc_find_and_modify_opts_destroy (opts);
bson_destroy (&reply);
bson_destroy (update);
mongoc_collection_destroy (collection);
mongoc_client_destroy (client);
mock_server_destroy (server);
bson_destroy (&doc);
}
static void
test_find_and_modify_collation (int wire)
{
mock_server_t *server;
mongoc_client_t *client;
mongoc_collection_t *collection;
bson_error_t error;
mongoc_find_and_modify_opts_t *opts;
future_t *future;
request_t *request;
bson_t *collation;
server = mock_server_with_autoismaster (wire);
mock_server_run (server);
client = mongoc_client_new_from_uri (mock_server_get_uri (server));
collection = mongoc_client_get_collection (client, "db", "collection");
collation = BCON_NEW ("collation",
"{",
"locale",
BCON_UTF8 ("en_US"),
"caseFirst",
BCON_UTF8 ("lower"),
"}");
opts = mongoc_find_and_modify_opts_new ();
mongoc_find_and_modify_opts_append (opts, collation);
if (wire >= WIRE_VERSION_COLLATION) {
future = future_collection_find_and_modify_with_opts (
collection, tmp_bson ("{}"), opts, NULL, &error);
request = mock_server_receives_command (
server,
"db",
MONGOC_QUERY_NONE,
"{'findAndModify': 'collection',"
" 'collation': { 'locale': 'en_US', 'caseFirst': 'lower'}"
"}");
mock_server_replies_ok_and_destroys (request);
ASSERT_OR_PRINT (future_get_bool (future), error);
future_destroy (future);
} else {
bool ok = mongoc_collection_find_and_modify_with_opts (
collection, tmp_bson ("{}"), opts, NULL, &error);
ASSERT_ERROR_CONTAINS (error,
MONGOC_ERROR_COMMAND,
MONGOC_ERROR_PROTOCOL_BAD_WIRE_VERSION,
"The selected server does not support collation");
ASSERT (!ok);
}
bson_destroy (collation);
mongoc_find_and_modify_opts_destroy (opts);
mongoc_collection_destroy (collection);
mongoc_client_destroy (client);
mock_server_destroy (server);
}
static void
test_find_and_modify (void)
{
mongoc_collection_t *collection;
mongoc_client_t *client;
bson_error_t error;
bson_iter_t iter;
bson_iter_t citer;
bson_t *update;
bson_t doc = BSON_INITIALIZER;
bson_t reply;
mongoc_find_and_modify_opts_t *opts;
client = test_framework_client_new ();
ASSERT (client);
collection = get_test_collection (client, "test_find_and_modify");
ASSERT (collection);
BSON_APPEND_INT32 (&doc, "superduper", 77889);
ASSERT_OR_PRINT (mongoc_collection_insert (
collection, MONGOC_INSERT_NONE, &doc, NULL, &error),
error);
update = BCON_NEW ("$set", "{", "superduper", BCON_INT32 (1234), "}");
opts = mongoc_find_and_modify_opts_new ();
mongoc_find_and_modify_opts_set_update (opts, update);
mongoc_find_and_modify_opts_set_fields (opts,
tmp_bson ("{'superduper': 1}"));
mongoc_find_and_modify_opts_set_sort (opts, tmp_bson ("{'superduper': 1}"));
mongoc_find_and_modify_opts_set_flags (opts,
MONGOC_FIND_AND_MODIFY_RETURN_NEW);
ASSERT_OR_PRINT (mongoc_collection_find_and_modify_with_opts (
collection, &doc, opts, &reply, &error),
error);
assert (bson_iter_init_find (&iter, &reply, "value"));
assert (BSON_ITER_HOLDS_DOCUMENT (&iter));
assert (bson_iter_recurse (&iter, &citer));
assert (bson_iter_find (&citer, "superduper"));
assert (BSON_ITER_HOLDS_INT32 (&citer));
assert (bson_iter_int32 (&citer) == 1234);
assert (bson_iter_init_find (&iter, &reply, "lastErrorObject"));
assert (BSON_ITER_HOLDS_DOCUMENT (&iter));
assert (bson_iter_recurse (&iter, &citer));
assert (bson_iter_find (&citer, "updatedExisting"));
assert (BSON_ITER_HOLDS_BOOL (&citer));
assert (bson_iter_bool (&citer));
bson_destroy (&reply);
bson_destroy (update);
ASSERT_OR_PRINT (mongoc_collection_drop (collection, &error), error);
mongoc_find_and_modify_opts_destroy (opts);
mongoc_collection_destroy (collection);
mongoc_client_destroy (client);
bson_destroy (&doc);
}
