CDRIVER-721 additional tests

Author: ajdavis

tests/mock-server.c

@@ -52,6 +52,9 @@ struct _mock_server_t
    int                    maxWireVersion;
    int                    maxBsonObjectSize;
    int                    maxMessageSizeBytes;
+
+   const char                  *setName;
+   const mongoc_host_list_t    *hosts;
 };
 
 
@@ -130,6 +133,11 @@ handle_ismaster (mock_server_t   *server,
 {
    bson_t reply_doc = BSON_INITIALIZER;
    time_t now = time (NULL);
+   bson_t hosts_array;
+   const char *hosts_array_key;
+   char buf[32];
+   uint32_t i;
+   const mongoc_host_list_t *host;
 
    BSON_ASSERT (server);
    BSON_ASSERT (client);
@@ -147,6 +155,21 @@ handle_ismaster (mock_server_t   *server,
                       server->maxWireVersion);
    bson_append_double (&reply_doc, "ok", -1, 1.0);
    bson_append_time_t (&reply_doc, "localtime", -1, now);
+   if (server->setName) {
+      BSON_APPEND_UTF8 (&reply_doc, "setName", server->setName);
+      host = server->hosts;
+      i = 0;
+      bson_append_array_begin (&reply_doc, "hosts", -1, &hosts_array);
+
+      while (host) {
+         bson_uint32_to_string (i, &hosts_array_key, buf, sizeof buf);
+         BSON_APPEND_UTF8 (&hosts_array, hosts_array_key, host->host_and_port);
+         i++;
+         host = host->next;
+      }
+      
+      bson_append_array_end (&reply_doc, &hosts_array);
+   }
 
    mock_server_reply_simple (server, client, rpc, MONGOC_REPLY_NONE, &reply_doc);
 
@@ -280,10 +303,12 @@ dummy_handler (mock_server_t   *server,
 
 
 mock_server_t *
-mock_server_new (const char            *address,
-                 uint16_t          port,
-                 mock_server_handler_t  handler,
-                 void                  *handler_data)
+mock_server_new_rs (const char                  *address,
+                    uint16_t                     port,
+                    mock_server_handler_t        handler,
+                    void                        *handler_data,
+                    const char                  *setName,
+                    const mongoc_host_list_t    *hosts)
 {
    mock_server_t *server;
 
@@ -308,12 +333,24 @@ mock_server_new (const char            *address,
    server->maxBsonObjectSize = 16777216;
    server->maxMessageSizeBytes = 48000000;
 
+   server->setName = setName;
+   server->hosts = hosts;
+
    mongoc_mutex_init (&server->mutex);
    mongoc_cond_init (&server->cond);
 
    return server;
 }
 
+mock_server_t *
+mock_server_new (const char            *address,
+                 uint16_t               port,
+                 mock_server_handler_t  handler,
+                 void                  *handler_data)
+{
+   return mock_server_new_rs (address, port, handler, handler_data, NULL, NULL);
+}
+
 
 int
 mock_server_run (mock_server_t *server)

tests/mock-server.h

@@ -42,6 +42,12 @@ mock_server_t *mock_server_new              (const char            *address,
                                              uint16_t          port,
                                              mock_server_handler_t  handler,
                                              void                  *handler_data);
+mock_server_t *mock_server_new_rs           (const char            *address,
+                                             uint16_t                     port,
+                                             mock_server_handler_t        handler,
+                                             void                        *handler_data,
+                                             const char                  *setName,
+                                             const mongoc_host_list_t    *hosts);
 void           mock_server_set_wire_version (mock_server_t         *server,
                                              int32_t           min_wire_version,
                                              int32_t           max_wire_version);

tests/test-mongoc-client.c

@@ -503,6 +503,45 @@ test_unavailable_seeds(void)
 }
 
 
+/* CDRIVER-721 catch errors in _mongoc_cluster_destroy */
+static void
+test_large_seed_list(void)
+{
+   uint16_t port;
+   char *uri_str;
+   mongoc_uri_t *uri;
+   mock_server_t *server;
+   int i;
+   const mongoc_host_list_t *hosts;
+   mongoc_client_t *client;
+   bson_error_t error;
+
+   port = 20000 + (rand () % 1000);
+   uri_str = bson_strdup_printf ("mongodb://localhost:%hu,a,b,c/?replicaSet=rs",
+                                 port);
+   uri = mongoc_uri_new (uri_str);
+   hosts = mongoc_uri_get_hosts (uri);
+   server = mock_server_new_rs ("127.0.0.1", port, NULL, NULL,
+                                "rs", hosts);
+
+   mock_server_run_in_thread (server);
+
+   for (i = 0; i < 10; i++) {
+      suppress_one_message ();
+   }
+
+   client = mongoc_client_new_from_uri (uri);
+   assert (client);
+   assert (_mongoc_client_warm_up (client, &error));
+
+   mongoc_client_destroy (client);
+   mock_server_quit (server, 0);
+   mock_server_destroy (server);
+   mongoc_uri_destroy (uri);
+   bson_free (uri_str);
+}
+
+
 static void
 test_exhaust_cursor (void)
 {
@@ -742,6 +781,7 @@ test_client_install (TestSuite *suite)
    TestSuite_Add (suite, "/Client/command_secondary", test_mongoc_client_command_secondary);
    TestSuite_Add (suite, "/Client/preselect", test_mongoc_client_preselect);
    TestSuite_Add (suite, "/Client/unavailable_seeds", test_unavailable_seeds);
+   TestSuite_Add (suite, "/Client/large_seed_list", test_large_seed_list);
    TestSuite_Add (suite, "/Client/exhaust_cursor", test_exhaust_cursor);
    TestSuite_Add (suite, "/Client/server_status", test_server_status);
 }