Finished 100% coverage again

khufuproject · May 5, 2011 · 22dcf0b · 22dcf0b
1 parent f26a404
commit 22dcf0b
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/khufu_sqlalchemy/tests.py b/khufu_sqlalchemy/tests.py
@@ -65,6 +65,19 @@ def test_context(self):
         req.context.db = marker = object()
         self.assertEqual(dbsession(req), marker)
 
+    def test_object_session(self):
+        from khufu_sqlalchemy import _DBSessionFinder
+        dbsession = _DBSessionFinder()
+        req = Mock(environ={}, context=Mock(),
+                   registry=Mock(settings={}))
+        self.assertEqual(dbsession(req, False), None)
+
+        marker = object()
+        dbsession._object_session = lambda x: marker
+        req = Mock(environ={}, context=Mock(),
+                  registry=Mock(settings={}))
+        self.assertEqual(dbsession(req, False), marker)
+
     def test_create(self):
         from khufu_sqlalchemy import dbsession, DBSESSION_FACTORY