From 530440b3335cde0364752da1504079854be960b3 Mon Sep 17 00:00:00 2001
From: Ross Lawley <ross.lawley@gmail.com>
Date: Tue, 1 May 2012 09:14:38 +0100
Subject: [PATCH] Fixed replicaset_connection test

---
 mongoengine/django/sessions.py | 2 +-
 tests/replicaset_connection.py | 6 +++++-
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/mongoengine/django/sessions.py b/mongoengine/django/sessions.py
index 9cea0fd8..d1e9289a 100644
--- a/mongoengine/django/sessions.py
+++ b/mongoengine/django/sessions.py
@@ -46,7 +46,7 @@ class SessionStore(SessionBase):
         while True:
             self._session_key = self._get_new_session_key()
             try:
-                self.save(must_create=True, safe=True)
+                self.save(must_create=True)
             except CreateError:
                 continue
             self.modified = True
diff --git a/tests/replicaset_connection.py b/tests/replicaset_connection.py
index c8a3bbf4..22978143 100644
--- a/tests/replicaset_connection.py
+++ b/tests/replicaset_connection.py
@@ -1,6 +1,6 @@
 import unittest
 import pymongo
-from pymongo import ReadPreference
+from pymongo import ReadPreference, ReplicaSetConnection
 
 import mongoengine
 from mongoengine import *
@@ -17,11 +17,15 @@ class ConnectionTest(unittest.TestCase):
     def test_replicaset_uri_passes_read_preference(self):
         """Requires a replica set called "rs" on port 27017
         """
+
         try:
             conn = connect(db='mongoenginetest', host="mongodb://localhost/mongoenginetest?replicaSet=rs", read_preference=ReadPreference.SECONDARY_ONLY)
         except ConnectionError, e:
             return
 
+        if not isinstance(conn, ReplicaSetConnection):
+            return
+
         self.assertEquals(conn.read_preference, ReadPreference.SECONDARY_ONLY)
 
 if __name__ == '__main__':