Add test for original fn set

Add a test for the original fn set when cache_on_arguments decorator is
used. This ensures the original fn is located at resulting
<decorated_func>.orignal
This commit is contained in:
Morgan Fainberg 2016-05-03 08:06:42 -07:00
parent 9fe7f57d42
commit f30bc2bfeb
1 changed files with 12 additions and 0 deletions

View File

@ -309,6 +309,18 @@ class CacheDecoratorTest(_GenericBackendFixture, TestCase):
generate.invalidate(1, 2)
eq_(generate(1, 2), 6)
def test_original_fn_set(self):
reg = self._region(backend="dogpile.cache.memory")
counter = itertools.count(1)
def generate(x, y):
return next(counter) + x + y
decorated = reg.cache_on_arguments()(generate)
eq_(decorated.original, generate)
def test_reentrant_call(self):
reg = self._region(backend="dogpile.cache.memory")