我已经从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.py
与https://github.com/pallets/flask/blob/master/examples/flaskr/flaskr/flaskr.py上的init_db
相同,并且定义了{{1}}个功能。如何进行单元测试?
答案 0 :(得分:15)
在脚本中导入flaskr
包时,可以访问flaskr/__init__.py
init_db
未在flask/__init__.py
中定义,提供的代码需要在init_db
中定义flask/__init__.py
。
有两种方法可以解决您遇到的问题:
将flaskr.init_db()
替换为flaskr.flaskr.init_db()
修改test_flaskr.py
中的import语句,如下所示:
将import flaskr
更改为from flaskr import flaskr