在" Flaskr"上运行单元测试Flask中的教程微博应用程序

时间:2017-01-12 12:14:59

标签: python unit-testing flask

我已经从Github克隆了flaskr应用程序,并尝试按照Testing Flask Applications教程进行操作。在Bonus: Testing the Application之后,我已将子目录test添加到顶级flaskr目录中,以便我的目录树如下所示:

.
├── build
│   ├── bdist.linux-x86_64
│   └── lib.linux-x86_64-2.7
│       └── flaskr
│           ├── flaskr.py
│           ├── __init__.py
│           ├── schema.sql
│           ├── static
│           │   └── style.css
│           └── templates
│               ├── layout.html
│               ├── login.html
│               └── show_entries.html
├── dist
│   └── flaskr-0.0.0-py2.7.egg
├── flaskr
│   ├── flaskr.db
│   ├── flaskr.py
│   ├── flaskr.pyc
│   ├── __init__.py
│   ├── __init__.pyc
│   ├── schema.sql
│   ├── static
│   │   └── style.css
│   └── templates
│       ├── layout.html
│       ├── login.html
│       └── show_entries.html
├── flaskr.egg-info
│   ├── dependency_links.txt
│   ├── PKG-INFO
│   ├── requires.txt
│   ├── SOURCES.txt
│   └── top_level.txt
├── MANIFEST.in
├── README
├── setup.cfg
├── setup.py
├── test
│   └── test_flaskr.py
└── tests
    └── test_flaskr.py

请注意,还有内置的'目录tests中的测试;但是,我在test_flaskr.py目录tests中编写测试。到目前为止,我只尝试了一次测试:

import os
import flaskr
import unittest
import tempfile

class FlaskrTestCase(unittest.TestCase):

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

    def tearDown(self):
        os.close(self.db_fd)
        os.unlink(flaskr.app.config['DATABASE'])

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

if __name__ == '__main__':
    unittest.main()

但是,如果我尝试运行此操作,则会收到以下错误:

Traceback (most recent call last):
  File "/home/kurt/dev/scratch/flask/examples/flaskr/test/test_flaskr.py", line 13, in setUp
    flaskr.init_db()
AttributeError: 'module' object has no attribute 'init_db'

我不明白这个错误。我的flaskr.pyhttps://github.com/pallets/flask/blob/master/examples/flaskr/flaskr/flaskr.py上的init_db相同,并且定义了{{1}}个功能。如何进行单元测试?

1 个答案:

答案 0 :(得分:15)

在脚本中导入flaskr包时,可以访问flaskr/__init__.py

中声明和定义的变量,函数等

init_db未在flask/__init__.py中定义,提供的代码需要在init_db中定义flask/__init__.py

有两种方法可以解决您遇到的问题:

  1. flaskr.init_db()替换为flaskr.flaskr.init_db()

  2. 修改test_flaskr.py中的import语句,如下所示:

    import flaskr更改为from flaskr import flaskr