pub/mongoengine
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Set default database when using mongomock.

Browse Source 
When passing host `mongomock://localhost/some-default-database` to `connect` the default database was `"test"` instead of `"some-default-database"`.

Fixes: #2130
This commit is contained in:
Nuno Rodrigues 2019-07-12 13:33:19 +01:00
parent aa76ccdd25
commit 290d9df3eb
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mongoengine/connection.py
View File

@ -103,7 +103,13 @@ def _get_connection_settings(
if entity.startswith("mongomock://"):
conn_settings["is_mock"] = True
# `mongomock://` is not a valid url prefix and must be replaced by `mongodb://`
resolved_hosts.append(entity.replace("mongomock://", "mongodb://", 1))
new_entity = entity.replace("mongomock://", "mongodb://", 1)
resolved_hosts.append(new_entity)
uri_dict = uri_parser.parse_uri(new_entity)
if uri_dict.get("database"):
conn_settings["name"] = uri_dict.get("database")
# Handle URI style connections, only updating connection params which
# were explicitly specified in the URI.