From c1f7abd2e4531077e4c04519fe03bbcf6bc4e195 Mon Sep 17 00:00:00 2001 From: Jannis Date: Tue, 17 Jan 2017 16:44:38 +0100 Subject: [PATCH] Add test for kwargs in template tag. --- avatar/utils.py | 4 ++-- tests/tests.py | 9 +++++++++ 2 files changed, 11 insertions(+), 2 deletions(-) diff --git a/avatar/utils.py b/avatar/utils.py index e0922f1..38db5b3 100644 --- a/avatar/utils.py +++ b/avatar/utils.py @@ -57,13 +57,13 @@ def cache_result(default_size=settings.AVATAR_DEFAULT_SIZE): return decorator def decorator(func): - def cached_func(user, size=None): + def cached_func(user, size=None, **kwargs): prefix = func.__name__ cached_funcs.add(prefix) key = get_cache_key(user, size or default_size, prefix=prefix) result = cache.get(key) if result is None: - result = func(user, size or default_size) + result = func(user, size or default_size, **kwargs) cache_set(key, result) return result return cached_func diff --git a/tests/tests.py b/tests/tests.py index e8c82f1..28f2501 100644 --- a/tests/tests.py +++ b/tests/tests.py @@ -210,6 +210,15 @@ class AvatarTests(TestCase): self.assertIn('', result) + def test_avatar_tag_works_with_kwargs(self): + upload_helper(self, "test.png") + avatar = get_primary_avatar(self.user) + + result = avatar_tags.avatar(self.user, title="Avatar") + + self.assertIn('test', result) + def test_default_add_template(self): response = self.client.get('/avatar/add/') self.assertContains(response, 'Upload New Image')