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_heartbeat_events (bool pooled,
bool succeeded)
{
context_t context;
mock_server_t *server;
mongoc_uri_t *uri;
mongoc_client_t *client;
mongoc_client_pool_t *pool = NULL;
future_t *future;
request_t *request;
char *expected_json;
bson_error_t error;
context_init (&context);
/* auto-respond to "foo" command */
server = mock_server_new ();
mock_server_run (server);
mock_server_autoresponds (server, responder, NULL, NULL);
uri = mongoc_uri_copy (mock_server_get_uri (server));
mongoc_uri_set_option_as_int32 (uri, "serverSelectionTimeoutMS", 400);
if (pooled) {
pool = mongoc_client_pool_new (uri);
pool_set_heartbeat_event_callbacks (pool, &context);
client = mongoc_client_pool_pop (pool);
} else {
client = mongoc_client_new_from_uri (uri);
client_set_heartbeat_event_callbacks (client, &context);
}
/* trigger "ismaster" handshake */
future = future_client_command_simple (client, "admin",
tmp_bson ("{'foo': 1}"),
NULL, NULL, &error);
/* topology scanner calls ismaster once */
request = mock_server_receives_ismaster (server);
if (succeeded) {
mock_server_replies_ok_and_destroys (request);
} else {
mock_server_hangs_up (request);
request_destroy (request);
}
/* pooled client opens new socket, handshakes it by calling ismaster again */
if (pooled && succeeded) {
request = mock_server_receives_ismaster (server);
mock_server_replies_ok_and_destroys (request);
}
if (succeeded) {
/* "foo" command succeeds */
ASSERT_OR_PRINT (future_get_bool (future), error);
} else {
ASSERT (!future_get_bool (future));
}
if (pooled) {
mongoc_client_pool_push (pool, client);
mongoc_client_pool_destroy (pool);
} else {
mongoc_client_destroy (client);
}
/* even if pooled, only topology scanner sends events, so we get one pair */
if (succeeded) {
expected_json = bson_strdup_printf (
"{'0': {'heartbeat_started_event': {'host': '%s'}},"
" '1': {'heartbeat_succeeded_event': {'host': '%s'}}}",
mock_server_get_host_and_port (server),
mock_server_get_host_and_port (server));
} else {
expected_json = bson_strdup_printf (
"{'0': {'heartbeat_started_event': {'host': '%s'}},"
" '1': {'heartbeat_failed_event': {'host': '%s'}}}",
mock_server_get_host_and_port (server),
mock_server_get_host_and_port (server));
}
check_json_apm_events (&context.events, tmp_bson (expected_json));
future_destroy (future);
bson_free (expected_json);
mongoc_uri_destroy (uri);
mock_server_destroy (server);
context_destroy (&context);
}
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);
}
