フラスコテスト - 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を使用してこれを行うことができます。私はここで、コードsnippitを再現します：ここでは

 
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の構造を知らないので、この例では、上記でありますちょうどプロトタイプ。