コード例 #1
0
ファイル: tests.py プロジェクト: stefanw/django-lazyinclude
 def setUp(self):
     self.middleware = LazyIncludeMiddleware()
     # need to patch render_template because RequestContext (context processors) are not available
     self.middleware.render_template = self.render_template
     self.rf = RequestFactory()
     self.template = template.Template(self.html)
     self.login_template = template.Template(self.login_html)
コード例 #2
0
ファイル: tests.py プロジェクト: stefanw/django-lazyinclude
class LazyIncludeTest(unittest.TestCase):
    html = """
        {% load lazyinclude_tags %}
        <html>...
        {{ result_of_expensive_database_hit }}
        {% lazyinclude "_login.html" %}
        {{ output_of_user_input }}
        ...</html>
    """
    login_html = """2nd template rendering"""
    
    def setUp(self):
        self.middleware = LazyIncludeMiddleware()
        # need to patch render_template because RequestContext (context processors) are not available
        self.middleware.render_template = self.render_template
        self.rf = RequestFactory()
        self.template = template.Template(self.html)
        self.login_template = template.Template(self.login_html)
        
    def sample_view(self, request, *args, **kwargs):
        c = template.Context({"result_of_expensive_database_hit": "42 (took some time)", 
            "output_of_user_input": """{% lazyinclude evil.html %}{% lazyinclude "evil.html" %}"""})
        return HttpResponse(self.template.render(c))
        
    def render_template(self, name, request):
        self.assertFalse("evil" in name)
        return self.login_template.render(template.Context({}))

    def testLazyInclude(self):
        # post requests result in none
        post_request = self.rf.post('/hello/', {"foo": "bar"})
        self.assertTrue(self.middleware.process_view(post_request, lambda x: HttpResponse(), (), {}) is None)
        
        get_request = self.rf.get('/hello/')
        self.assertTrue(cache.get(_get_cache_key(get_request), None) is None)
        response = self.middleware.process_view(get_request, self.sample_view, (), {})
        cached_response = cache.get(_get_cache_key(get_request), None)
        view_response = self.sample_view(get_request)
        self.assertTrue(cached_response is not None)
        self.assertFalse(self.login_html in cached_response.content)
        self.assertTrue(self.login_html in response.content)
        self.assertEqual(view_response.content, cached_response.content)