From 22dcf0be12fa4d8ab41f1a91e1dcc2e9ad7ca32c Mon Sep 17 00:00:00 2001
From: Rocky Burt <rocky@serverzen.com>
Date: Wed, 4 May 2011 22:43:24 -0230
Subject: [PATCH] Finished 100% coverage again

---
 khufu_sqlalchemy/tests.py | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/khufu_sqlalchemy/tests.py b/khufu_sqlalchemy/tests.py
index 8e62648..ff33bfa 100644
--- 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