Cómo probar variables de contexto de plantilla con Flask

Question

El cliente de prueba de Django devuelve un objeto de respuesta de prueba que incluye las variables de contexto de plantilla que se utilizaron para representar la plantilla. https://docs.djangoproject.com/en/dev/topics/testing/#django.test.client.Response.context

¿Cómo puedo obtener acceso a las variables de contexto de la plantilla mientras las pruebas en el Frasco?

Ejemplo vista:

@pgt.route('/myview') 
def myview(): 
    context = { 
     'var1': 'value 1', 
     'var2': 'value 2', 
     'var3': 'value 3', 
    } 
    return render_template('mytemplate.html', **context) 

Ejemplo de Ensayo:

class MyViewTestCase(unittest.TestCase): 
    def setUp(self): 
     self.app = create_app() 
     self.client = self.app.test_client() 

    def test_get_success(self): 
     response = self.client.get('/pgt/myview') 

     # I don't want to do this 
     self.assertIn('value 1', response.data) 

     # I want to do something like this 
     self.assertEqual(response.template_context['var1'], 'value 1') 

Answer 1

Gracias a @andrewwatts I utilizado (una versión de) Flask-Testing

from flask.ext.testing import TestCase 


class MyViewTestCase(TestCase): 
    def create_app(self): 
     # This method is required by flask.ext.testing.TestCase. It is called 
     # before setUp(). 
     return create_app() 

    def test_get_success(self): 
     response = self.client.get('/pgt/myview') 
     self.assertEqual(self.get_context_variable('var1'), 'value 1') 

Answer 2

partir de esta información limitada, sugeriría romper el código contexto productoras en una unidad y la prueba separada que directamente. Mapeo del ejemplo anterior:

def get_context(): 
    context = { 
     'var1': 'value 1', 
     'var2': 'value 2', 
     'var3': 'value 3', 
    } 
    return context 

@pgt.route('/myview') 
def myview(): 
    return render_template('mytemplate.html', **get_context())