fix: fix url parsing for a mongodb+srv url that has commas in the database name (#2789)

Author: pablomatiasgomez

lib/core/uri_parser.js

@@ -52,7 +52,9 @@ function parseSrvConnectionString(uri, options, callback) {
   }
 
   result.domainLength = result.hostname.split('.').length;
-  if (result.pathname && result.pathname.match(',')) {
+
+  const hostname = uri.substring('mongodb+srv://'.length).split('/')[0];
+  if (hostname.match(',')) {
     return callback(new MongoParseError('Invalid URI, cannot contain multiple hostnames'));
   }
 

lib/url_parser.js

@@ -35,7 +35,8 @@ module.exports = function(url, options, callback) {
 
   result.domainLength = result.hostname.split('.').length;
 
-  if (result.pathname && result.pathname.match(',')) {
+  const hostname = url.substring('mongodb+srv://'.length).split('/')[0];
+  if (hostname.match(',')) {
     return callback(new Error('Invalid URI, cannot contain multiple hostnames'));
   }
 

test/functional/mongodb_srv.test.js

@@ -52,6 +52,9 @@ describe('mongodb+srv (spec)', function() {
               expect(object.auth.user).to.equal(test[1].parsed_options.user);
               expect(object.auth.password).to.equal(test[1].parsed_options.password);
             }
+            if (test[1].parsed_options && test[1].parsed_options.dbName) {
+              expect(object.dbName).to.equal(test[1].parsed_options.dbName);
+            }
           }
           done();
         });

test/spec/dns-txt-records/dbname-with-commas.json

@@ -0,0 +1,19 @@
+{
+  "uri": "mongodb+srv://test1.test.build.10gen.cc/some,db?replicaSet=repl0",
+  "seeds": [
+    "test1.test.build.10gen.cc:27017",
+    "test1.test.build.10gen.cc:27018"
+  ],
+  "hosts": [
+    "localhost:27017",
+    "localhost:27018",
+    "localhost:27019"
+  ],
+  "options": {
+    "replicaSet": "repl0",
+    "ssl": true
+  },
+  "parsed_options": {
+    "dbName": "some,db"
+  }
+}

test/spec/dns-txt-records/dbname-with-commas.yml

@@ -0,0 +1,13 @@
+uri: "mongodb+srv://test1.test.build.10gen.cc/some,db?replicaSet=repl0"
+seeds:
+  - test1.test.build.10gen.cc:27017
+  - test1.test.build.10gen.cc:27018
+hosts:
+  - localhost:27017
+  - localhost:27018
+  - localhost:27019
+options:
+  replicaSet: repl0
+  ssl: true
+parsed_options:
+  dbName: some,db

test/spec/initial-dns-seedlist-discovery/dbname-with-commas.json

@@ -0,0 +1,19 @@
+{
+  "uri": "mongodb+srv://test1.test.build.10gen.cc/some,db?replicaSet=repl0",
+  "seeds": [
+    "test1.test.build.10gen.cc:27017",
+    "test1.test.build.10gen.cc:27018"
+  ],
+  "hosts": [
+    "localhost:27017",
+    "localhost:27018",
+    "localhost:27019"
+  ],
+  "options": {
+    "replicaSet": "repl0",
+    "ssl": true
+  },
+  "parsed_options": {
+    "defaultDatabase": "some,db"
+  }
+}

test/spec/initial-dns-seedlist-discovery/dbname-with-commas.yml

@@ -0,0 +1,13 @@
+uri: "mongodb+srv://test1.test.build.10gen.cc/some,db?replicaSet=repl0"
+seeds:
+  - test1.test.build.10gen.cc:27017
+  - test1.test.build.10gen.cc:27018
+hosts:
+  - localhost:27017
+  - localhost:27018
+  - localhost:27019
+options:
+  replicaSet: repl0
+  ssl: true
+parsed_options:
+  defaultDatabase: some,db

test/unit/core/mongodb_srv.test.js

@@ -57,6 +57,9 @@ describe('mongodb+srv', function() {
                 expect(result.auth.username).to.equal(test[1].parsed_options.user);
                 expect(result.auth.password).to.equal(test[1].parsed_options.password);
               }
+              if (test[1].parsed_options && test[1].parsed_options.dbName) {
+                expect(result.defaultDatabase).to.equal(test[1].parsed_options.dbName);
+              }
             }
 
             done();