Change environment cache key construction

Changing from a tuple of the loader ID and template name to a weakref
to the loader and the template name should avoid situations whereby
the loader has changed, yet the cached templates are returned. This
would occur if the id of the new loader matches the old. A weakref is
preferred over a direct reference so that the loader can be garbaged
collected.

1 files changed, 5 insertions, 3 deletions

diff --git a/tests/test_loader.py b/tests/test_loader.py
index 6b9d9bb4..b916b911 100644
--- a/tests/test_loader.py
+++ b/tests/test_loader.py
@@ -13,6 +13,7 @@ import sys
 import tempfile
 import shutil
 import pytest
+import weakref
 
 from jinja2 import Environment, loaders
 from jinja2._compat import PYPY, PY2
@@ -92,9 +93,10 @@ class TestLoaders():
         assert t2 is env.get_template('two')
         assert t1 is env.get_template('one')
         t3 = env.get_template('three')
-        assert (id(loader), 'one') in env.cache
-        assert (id(loader), 'two') not in env.cache
-        assert (id(loader), 'three') in env.cache
+        loader_ref = weakref.ref(loader)
+        assert (loader_ref, 'one') in env.cache
+        assert (loader_ref, 'two') not in env.cache
+        assert (loader_ref, 'three') in env.cache
 
     def test_cache_loader_change(self):
         loader1 = loaders.DictLoader({'foo': 'one'})