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

Add test

Browse Source
This commit is contained in:
Nuno Rodrigues 2019-07-15 16:53:24 +00:00
parent 290d9df3eb
commit 2d02551d0a
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tests/test_connection.py
View File

@ -4,6 +4,8 @@ from pymongo import MongoClient
from pymongo.errors import OperationFailure, InvalidName from pymongo.errors import OperationFailure, InvalidName
from pymongo import ReadPreference from pymongo import ReadPreference
from mongoengine import Document
try: try:
import unittest2 as unittest import unittest2 as unittest
except ImportError: except ImportError:
@ -269,6 +271,14 @@ class ConnectionTest(unittest.TestCase):
conn = get_connection("testdb7") conn = get_connection("testdb7")
self.assertIsInstance(conn, mongomock.MongoClient) 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): def test_connect_with_host_list(self):
"""Ensure that the connect() method works when host is a list """Ensure that the connect() method works when host is a list