CXX-3204 add URI setter for server_selection_try_once (#1374)

Author: kevinAlbs

src/mongocxx/include/mongocxx/v_noabi/mongocxx/uri.hpp

@@ -298,6 +298,16 @@ class uri {
     MONGOCXX_ABI_EXPORT_CDECL(bsoncxx::v_noabi::stdx::optional<bool>)
     server_selection_try_once() const;
 
+    ///
+    /// Sets the value of the option "serverSelectionTryOnce" in the uri.
+    ///
+    /// @param val The new value to apply to as "serverSelectionTryOnce".
+    ///
+    /// @throws mongocxx::v_noabi::exception if there is an error setting the option.
+    ///
+    MONGOCXX_ABI_EXPORT_CDECL(void)
+    server_selection_try_once(bool val);
+
     ///
     /// Returns the value of the option "socketTimeoutMS" if present in the uri.
     ///

src/mongocxx/lib/mongocxx/v_noabi/mongocxx/uri.cpp

@@ -253,6 +253,12 @@ bsoncxx::v_noabi::stdx::optional<bool> uri::server_selection_try_once() const {
     return _bool_option(_impl->uri_t, "serverSelectionTryOnce");
 }
 
+void uri::server_selection_try_once(bool val) {
+    if (!mongoc_uri_set_option_as_bool(_impl->uri_t, "serverSelectionTryOnce", val)) {
+        throw exception{error_code::k_invalid_uri, "failed to set 'serverSelectionTryOnce' option"};
+    }
+}
+
 bsoncxx::v_noabi::stdx::optional<std::int32_t> uri::socket_timeout_ms() const {
     return _int32_option(_impl->uri_t, "socketTimeoutMS");
 }

src/mongocxx/test/uri.cpp

@@ -331,4 +331,46 @@ TEST_CASE("uri::zlib_compression_level()", "[uri]") {
 
 // End special cases.
 
+TEST_CASE("can set server_selection_try_once", "[uri]") {
+    SECTION("URI with no option") {
+        auto uri = mongocxx::uri{"mongodb://host"};
+
+        SECTION("can set to true") {
+            CHECK(!uri.server_selection_try_once());
+            uri.server_selection_try_once(true);
+            CHECK(uri.server_selection_try_once());
+            CHECK(*uri.server_selection_try_once());
+        }
+
+        SECTION("can set to false") {
+            CHECK(!uri.server_selection_try_once());
+            uri.server_selection_try_once(false);
+            CHECK(uri.server_selection_try_once());
+            CHECK(!*uri.server_selection_try_once());
+        }
+    }
+
+    SECTION("URI with serverSelectionTryOnce=true") {
+        auto uri = mongocxx::uri{"mongodb://host/?serverSelectionTryOnce=true"};
+        SECTION("can overwrite to false") {
+            CHECK(uri.server_selection_try_once());
+            CHECK(*uri.server_selection_try_once());
+            uri.server_selection_try_once(false);
+            CHECK(uri.server_selection_try_once());
+            CHECK(!*uri.server_selection_try_once());
+        }
+    }
+
+    SECTION("Test URI with serverSelectionTryOnce=false") {
+        auto uri = mongocxx::uri{"mongodb://host/?serverSelectionTryOnce=false"};
+        SECTION("can overwrite to true") {
+            CHECK(uri.server_selection_try_once());
+            CHECK(!*uri.server_selection_try_once());
+            uri.server_selection_try_once(true);
+            CHECK(uri.server_selection_try_once());
+            CHECK(*uri.server_selection_try_once());
+        }
+    }
+}
+
 } // namespace