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

Fixed replicaset_connection test

Browse Source
This commit is contained in:
Ross Lawley 2012-05-01 09:14:38 +01:00
parent b80fda36af
commit 530440b333
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mongoengine/django/sessions.py
View File

@ -46,7 +46,7 @@ class SessionStore(SessionBase):
while True: while True:
self._session_key = self._get_new_session_key() self._session_key = self._get_new_session_key()
try: try:
self.save(must_create=True, safe=True) self.save(must_create=True)
except CreateError: except CreateError:
continue continue
self.modified = True self.modified = True

6
tests/replicaset_connection.py
View File

@ -1,6 +1,6 @@
import unittest import unittest
import pymongo import pymongo
from pymongo import ReadPreference from pymongo import ReadPreference, ReplicaSetConnection
import mongoengine import mongoengine
from mongoengine import * from mongoengine import *
@ -17,11 +17,15 @@ class ConnectionTest(unittest.TestCase):
def test_replicaset_uri_passes_read_preference(self): def test_replicaset_uri_passes_read_preference(self):
"""Requires a replica set called "rs" on port 27017 """Requires a replica set called "rs" on port 27017
""" """
try: try:
conn = connect(db='mongoenginetest', host="mongodb://localhost/mongoenginetest?replicaSet=rs", read_preference=ReadPreference.SECONDARY_ONLY) conn = connect(db='mongoenginetest', host="mongodb://localhost/mongoenginetest?replicaSet=rs", read_preference=ReadPreference.SECONDARY_ONLY)
except ConnectionError, e: except ConnectionError, e:
return return
if not isinstance(conn, ReplicaSetConnection):
return
self.assertEquals(conn.read_preference, ReadPreference.SECONDARY_ONLY) self.assertEquals(conn.read_preference, ReadPreference.SECONDARY_ONLY)
if __name__ == '__main__': if __name__ == '__main__':