笨鸟编程-零基础入门Pyhton教程 › 首页 ›Flask中文手册 › 查看内容

测试 Flask 应用

第一个测试

是进行第一个应用功能的测试的时候了。让我们检查当我们访问根路径(/)时应用程序是否正确地返回了了“No entries here so far” 字样。为此，我们添加了一个新的测试函数到我们的类当中，如下面的代码所示:

class FlaskrTestCase(unittest.TestCase):

    def setUp(self):
        self.db_fd, flaskr.app.config['DATABASE'] = tempfile.mkstemp()
        self.app = flaskr.app.test_client()
        flaskr.init_db()

    def tearDown(self):
        os.close(self.db_fd)
        os.unlink(flaskr.DATABASE)

    def test_empty_db(self):
        rv = self.app.get('/')
        assert 'No entries here so far' in rv.data

注意到我们的测试函数以 test 开头，这允许 unittest 模块自动识别出哪些方法是一个测试方法，并且运行它。

通过使用 self.app.get 我们可以发送一个 HTTP GET 请求给应用的某个给定路径。返回值将会是一个 response_class 对象。我们可以使用 data 属性来检查程序的返回值(以字符串类型)。在这里，我们检查 'No entries here so far' 是不是输出内容的一部分。

再次运行，您应该看到一个测试成功通过了:

$ python flaskr_tests.py
.
----------------------------------------------------------------------
Ran 1 test in 0.034s

OK

登陆和登出

我们应用的大部分功能只允许具有管理员资格的用户访问。所以我们需要一种方法来帮助我们的测试客户端登陆和登出。为此，我们向登陆和登出页面发送一些请求，这些请求都携带了表单数据（用户名和密码），因为登陆和登出页面都会重定向，我们将客户端设置为 follow_redirects 。

将如下两个方法加入到您的 FlaskrTestCase 类:

def login(self, username, password):
    return self.app.post('/login', data=dict(
        username=username,
        password=password
    ), follow_redirects=True)

def logout(self):
    return self.app.get('/logout', follow_redirects=True)

现在我们可以轻松的测试登陆和登出是正常工作还是因认证失败而出错，添加新的测试函数到类中:

def test_login_logout(self):
    rv = self.login('admin', 'default')
    assert 'You were logged in' in rv.data
    rv = self.logout()
    assert 'You were logged out' in rv.data
    rv = self.login('adminx', 'default')
    assert 'Invalid username' in rv.data
    rv = self.login('admin', 'defaultx')
    assert 'Invalid password' in rv.data

测试消息的添加

我们同时应该测试消息的添加功能是否正常，添加一个新的测试方法如下:

def test_messages(self):
    self.login('admin', 'default')
    rv = self.app.post('/add', data=dict(
        title='<Hello>',
        text='<strong>HTML</strong> allowed here'
    ), follow_redirects=True)
    assert 'No entries here so far' not in rv.data
    assert '&lt;Hello&gt;' in rv.data
    assert '<strong>HTML</strong> allowed here' in rv.data

这里我们测试计划的行为是否能够正常工作，即在正文中可以出现 HTML 标签，而在标题中不允许。

运行这个测试，我们应该得到三个通过的测试:

$ python flaskr_tests.py
...
----------------------------------------------------------------------
Ran 3 tests in 0.332s

OK

关于请求的头信息和状态值等更复杂的测试，请参考 MiniTwit Example ，在这个例子的源代码里包含一套更长的测试。

123 4 / 4 页下一页

		自动登录	找回密码
密码			立即注册