Merge PR #20.

2 files changed, 9 insertions, 0 deletions

diff --git a/repoze/lru/__init__.py b/repoze/lru/__init__.py
index 83f22b3..e5c3280 100644
--- a/repoze/lru/__init__.py
+++ b/repoze/lru/__init__.py
@@ -294,6 +294,7 @@ class lru_cache(object):
         _maybe_copy(f, lru_cached, '__module__')
         _maybe_copy(f, lru_cached, '__name__')
         _maybe_copy(f, lru_cached, '__doc__')
+        lru_cached._cache = cache
         return lru_cached
 
 
diff --git a/repoze/lru/tests.py b/repoze/lru/tests.py
index 69a1c35..7922a30 100644
--- a/repoze/lru/tests.py
+++ b/repoze/lru/tests.py
@@ -520,6 +520,14 @@ class DecoratorTests(unittest.TestCase):
         self.assertEqual(result, 2)
         self.assertEqual(len(cache), 2)
 
+    def test_cache_attr(self):
+        cache = DummyLRUCache()
+        decorator = self._makeOne(0, cache)
+        def wrapped(key):
+            return key
+        decorated = decorator(wrapped)
+        self.assertEqual(decorated._cache, cache)
+
     def test_multiargs(self):
         cache = DummyLRUCache()
         decorator = self._makeOne(0, cache)