CDRIVER-939: Test with domain socket connection

Author: bjori

tests/test-libmongoc.c

@@ -230,6 +230,33 @@ test_framework_getenv_int64 (const char *name,
    return default_value;
 }
 
+
+/*
+ *--------------------------------------------------------------------------
+ *
+ * test_framework_get_unix_domain_socket_path --
+ *
+ *       Get the path to Unix Domain Socket .sock of the test MongoDB server.
+ *
+ * Returns:
+ *       A string you must bson_free.
+ *
+ * Side effects:
+ *       None.
+ *
+ *--------------------------------------------------------------------------
+ */
+char *
+test_framework_get_unix_domain_socket_path (void)
+{
+   char *path = test_framework_getenv ("MONGOC_TEST_UNIX_DOMAIN_SOCKET");
+
+   if (path) {
+      return path;
+   }
+   return bson_strdup_printf ("%%2Ftmp%%2Fmongodb-%d.sock",
+         test_framework_get_port());
+}
 /*
  *--------------------------------------------------------------------------
  *
@@ -480,6 +507,45 @@ test_framework_get_ssl (void)
 }
 
 
+/*
+ *--------------------------------------------------------------------------
+ *
+ * test_framework_get_unix_domain_socket_uri_str --
+ *
+ *       Get the connection string (unix domain socket style) of the test
+ *       MongoDB server based on the variables set in the environment.
+ *       Does *not* call isMaster to discover your actual topology.
+ *
+ * Returns:
+ *       A string you must bson_free.
+ *
+ * Side effects:
+ *       Same as test_framework_get_user_password.
+ *
+ *--------------------------------------------------------------------------
+ */
+char *
+test_framework_get_unix_domain_socket_uri_str ()
+{
+   char *path;
+   char *test_uri_str;
+   char *test_uri_str_auth;
+
+   path = test_framework_get_unix_domain_socket_path ();
+   test_uri_str = bson_strdup_printf (
+      "mongodb://%s/%s",
+      path,
+      test_framework_get_ssl () ? "?ssl=true" : "");
+
+   test_uri_str_auth = test_framework_add_user_password_from_env (test_uri_str);
+
+   bson_free (path);
+   bson_free (test_uri_str);
+
+   return test_uri_str_auth;
+}
+
+
 /*
  *--------------------------------------------------------------------------
  *
@@ -916,6 +982,12 @@ test_framework_is_replset (void)
    return is_replset;
 }
 
+int
+test_framework_skip_if_single (void)
+{
+   return (test_framework_is_mongos () || test_framework_is_replset());
+}
+
 int
 test_framework_skip_if_mongos (void)
 {
@@ -928,6 +1000,16 @@ test_framework_skip_if_replset (void)
    return test_framework_is_replset() ? 0 : 1;
 }
 
+int
+test_framework_skip_if_windows (void)
+{
+#ifdef _WIN32
+   return false;
+#else
+   return true;
+#endif
+}
+
 bool
 test_framework_max_wire_version_at_least (int version)
 {

tests/test-libmongoc.h

@@ -35,15 +35,21 @@ char *test_framework_add_user_password (const char *uri_str,
                                         const char *password);
 char *test_framework_get_uri_str_no_auth (const char *database_name);
 char *test_framework_get_uri_str (void);
+char *test_framework_get_unix_domain_socket_uri_str (void);
+char *test_framework_get_unix_domain_socket_path (void);
 mongoc_uri_t *test_framework_get_uri (void);
 void test_framework_set_ssl_opts (mongoc_client_t *client);
 mongoc_client_t *test_framework_client_new (void);
 mongoc_client_pool_t *test_framework_client_pool_new (void);
+bool test_framework_max_wire_version_at_least (int version);
+
 bool test_framework_is_mongos (void);
 bool test_framework_is_replset (void);
-int test_framework_skip_if_mongos (void);
+
+int test_framework_skip_if_mongos  (void);
 int test_framework_skip_if_replset (void);
-bool test_framework_max_wire_version_at_least (int version);
+int test_framework_skip_if_single  (void);
+int test_framework_skip_if_windows (void);
 
 typedef struct _debug_stream_stats_t {
     int n_destroyed;

tests/test-mongoc-client.c

@@ -852,6 +852,33 @@ test_mongoc_client_ipv6 (void)
 }
 
 
+static void
+test_mongoc_client_unix_domain_socket (void *context)
+{
+   mongoc_client_t *client;
+   bson_error_t error;
+   char *uri_str;
+   bson_iter_t iter;
+   bson_t reply;
+
+   uri_str = test_framework_get_unix_domain_socket_uri_str ();
+   client = mongoc_client_new (uri_str);
+
+   assert (client);
+
+   ASSERT_OR_PRINT (mongoc_client_get_server_status (client, NULL,
+                                                     &reply, &error), error);
+
+   assert (bson_iter_init_find (&iter, &reply, "host"));
+   assert (bson_iter_init_find (&iter, &reply, "version"));
+   assert (bson_iter_init_find (&iter, &reply, "ok"));
+
+   bson_destroy (&reply);
+
+   mongoc_client_destroy (client);
+}
+
+
 void
 test_client_install (TestSuite *suite)
 {
@@ -884,6 +911,7 @@ test_client_install (TestSuite *suite)
    TestSuite_Add (suite, "/Client/recovering", test_recovering);
    TestSuite_Add (suite, "/Client/server_status", test_server_status);
    TestSuite_Add (suite, "/Client/database_names", test_get_database_names);
+   TestSuite_AddFull (suite, "/Client/connect/uds", test_mongoc_client_unix_domain_socket, NULL, NULL, test_framework_skip_if_windows);
 
 #ifdef TODO_CDRIVER_689
    TestSuite_Add (suite, "/Client/wire_version", test_wire_version);

tests/test-mongoc-server-selection-errors.c

@@ -4,11 +4,6 @@
 #include "test-conveniences.h"
 #include "test-libmongoc.h"
 
-int skip_if_single (void)
-{
-   return (test_framework_is_mongos () || test_framework_is_replset());
-}
-
 static void
 server_selection_error_dns (const char *uri, const char *errmsg, bool assert_as)
 {
@@ -56,6 +51,7 @@ test_server_selection_error_dns_multi_fail (void)
       false
    );
 }
+
 static void
 test_server_selection_error_dns_multi_success (void *context)
 {
@@ -73,10 +69,63 @@ test_server_selection_error_dns_multi_success (void *context)
    bson_free (uri_str);
 }
 
+static void
+test_server_selection_uds_auth_failure (void *context)
+{
+   mongoc_client_t *client;
+   bson_error_t error;
+   bson_t reply;
+   char *path;
+   char *uri;
+
+   path = test_framework_get_unix_domain_socket_path ();
+   uri = bson_strdup_printf ("mongodb://user:wrongpass@%s", path);
+   client = mongoc_client_new (uri);
+
+   assert (client);
+
+   ASSERT_OR_PRINT (! mongoc_client_get_server_status (client, NULL,
+                                                       &reply, &error), error);
+
+   ASSERT_CMPINT (error.domain, ==, MONGOC_ERROR_CLIENT);
+   ASSERT_CMPINT (error.code, ==, MONGOC_ERROR_CLIENT_AUTHENTICATE);
+
+   bson_destroy (&reply);
+   bson_free (path);
+   bson_free (uri);
+
+   mongoc_client_destroy (client);
+}
+
+static void
+test_server_selection_uds_not_found (void *context)
+{
+   mongoc_client_t *client;
+   bson_error_t error;
+   bson_t reply;
+
+   client = mongoc_client_new ("mongodb:///tmp/mongodb-so-close.sock");
+
+   assert (client);
+
+   ASSERT_OR_PRINT (! mongoc_client_get_server_status (client, NULL,
+                                                       &reply, &error), error);
+
+   ASSERT_CMPINT (error.domain, ==, MONGOC_ERROR_SERVER_SELECTION);
+   ASSERT_CMPINT (error.code, ==, MONGOC_ERROR_SERVER_SELECTION_FAILURE);
+
+   bson_destroy (&reply);
+
+   mongoc_client_destroy (client);
+}
+
+
 void
 test_server_selection_errors_install (TestSuite *suite)
 {
    TestSuite_Add (suite, "/server_selection/errors/dns/single", test_server_selection_error_dns_single);
    TestSuite_Add (suite, "/server_selection/errors/dns/multi/fail", test_server_selection_error_dns_multi_fail);
-   TestSuite_AddFull (suite, "/server_selection/errors/dns/multi/success", test_server_selection_error_dns_multi_success, NULL, NULL, skip_if_single);
+   TestSuite_AddFull (suite, "/server_selection/errors/dns/multi/success", test_server_selection_error_dns_multi_success, NULL, NULL, test_framework_skip_if_single);
+   TestSuite_AddFull (suite, "/server_selection/errors/uds/auth_failure", test_server_selection_uds_auth_failure, NULL, NULL, test_framework_skip_if_windows);
+   TestSuite_AddFull (suite, "/server_selection/errors/uds/not_found", test_server_selection_uds_not_found, NULL, NULL, test_framework_skip_if_windows);
 }