瓶測試 - 如何檢索傳遞給Jinja的變量？

Question

在Flask中，如何測試使用render_template返回Jinja模板的變量？

@app.route('/foo/'): 
def foo(): 
    return render_template('foo.html', foo='bar') 

在本例中，我想測試foo等於"bar"。

import unittest 
from app import app 

class TestFoo(unittest.TestCase): 
    def test_foo(self): 
     with app.test_client() as c: 
      r = c.get('/foo/') 
      # Prove that the foo variable is equal to "bar" 

我該怎麼做？

Answer 1

這可以使用signals完成。我將在這裏重現代碼片段：

 
import unittest 
from app import app 
from flask import template_rendered 
from contextlib import contextmanager 

@contextmanager 
def captured_templates(app): 
    recorded = [] 
    def record(sender, template, context, **extra): 
     recorded.append((template, context)) 
    template_rendered.connect(record, app) 
    try: 
     yield recorded 
    finally: 
     template_rendered.disconnect(record, app) 

class TestFoo(unittest.TestCase): 
    def test_foo(self): 
     with app.test_client() as c: 
      with captured_templates(app) as templates: 
       r = c.get('/foo/') 
       template, context = templates[0] 
       self.assertEquals(context['foo'], 'bar') 

---

這裏是另一種實現方式，消除了template一部分，並把它變成一個迭代器。

 
import unittest 
from app import app 
from flask import template_rendered 
from contextlib import contextmanager 

@contextmanager 
def get_context_variables(app): 
    recorded = [] 
    def record(sender, template, context, **extra): 
     recorded.append(context) 
    template_rendered.connect(record, app) 
    try: 
     yield iter(recorded) 
    finally: 
     template_rendered.disconnect(record, app) 

class TestFoo(unittest.TestCase): 
    def test_foo(self): 
     with app.test_client() as c: 
      with get_context_variables(app) as contexts: 
       r = c.get('/foo/') 
       context = next(context) 
       self.assertEquals(context['foo'], 'bar') 

       r = c.get('/foo/?foo=bar') 
       context = next(context) 
       self.assertEquals(context['foo'], 'foo') 

       # This will raise a StopIteration exception because I haven't rendered 
       # and new templates 
       next(context) 

Answer 2

的最佳方式將是使用類似的東西：

self.assertTrue('Hello bar!' in r.body) 

而且在foo.html：

<div>Hello {{ foo }}!</div> 

當然，我不知道你的HTML的結構，所以上面這個例子是隻是一個原型。