Book Image

Flask Framework Cookbook

By : Shalabh Aggarwal
Book Image

Flask Framework Cookbook

By: Shalabh Aggarwal

Overview of this book

Table of Contents (19 chapters)
Flask Framework Cookbook
Credits
About the Author
About the Reviewers
www.PacktPub.com
Preface
Index

Writing more tests for views and logic


In the last recipe, we got started with writing tests for our Flask application. In this recipe, we will build upon the same test file and add more tests for our application; these tests will cover testing the views for behavior and logic.

Getting ready

We will build upon the test file named app_tests.py created in the last recipe.

How to do it…

Before we write any tests, we need to add a small bit of configuration to setUp() to disable the CSRF tokens, as they are not generated by default for test environments:

app.config['WTF_CSRF_ENABLED'] = False

The following are some tests that are created as a part of this recipe. Each test will be described as we go further:

    def test_products(self):
        "Test Products list page"
        rv = self.app.get('/en/products')
        self.assertEqual(rv.status_code, 200)
        self.assertTrue('No Previous Page' in rv.data)
        self.assertTrue('No Next Page' in rv.data)

The preceding test sends a GET request...