From 2d02551d0a41405664dc26604e90669f6e65aca6 Mon Sep 17 00:00:00 2001
From: Nuno Rodrigues <nunomr@gmail.com>
Date: Mon, 15 Jul 2019 16:53:24 +0000
Subject: [PATCH] Add test

---
 tests/test_connection.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/tests/test_connection.py b/tests/test_connection.py
index f9c9d098..78f7e52f 100644
--- a/tests/test_connection.py
+++ b/tests/test_connection.py
@@ -4,6 +4,8 @@ from pymongo import MongoClient
 from pymongo.errors import OperationFailure, InvalidName
 from pymongo import ReadPreference
 
+from mongoengine import Document
+
 try:
     import unittest2 as unittest
 except ImportError:
@@ -269,6 +271,14 @@ class ConnectionTest(unittest.TestCase):
         conn = get_connection("testdb7")
         self.assertIsInstance(conn, mongomock.MongoClient)
 
+        disconnect_all()
+        class SomeDocument(Document): pass
+        conn = connect(host="mongomock://localhost:27017/mongoenginetest8")
+        some_document = SomeDocument()
+        some_document.save()
+        self.assertEqual(conn.get_default_database().name, "mongoenginetest8")
+        self.assertEqual(conn.database_names()[0], "mongoenginetest8")
+
     def test_connect_with_host_list(self):
         """Ensure that the connect() method works when host is a list