Dmy*_*tro 5 python import python-unittest
我尝试使用flask和 来创建简单的Python项目unittest。结构非常简单:
classes
|-sysinfo
|static
|templates
|- index.html
|- layout.html
|__init__.py
|sysinfo.py
|printinfo.py
tests
|test_sysinfo.py
README.md
requirments.txt
Run Code Online (Sandbox Code Playgroud)
非常简单的课程printinfo.py:
#!/usr/bin/python
import psutil
import json
class SysInfo:
.......
def displayInfo(self):
.......
return json.dumps(self.__data)
Run Code Online (Sandbox Code Playgroud)
简单的烧瓶服务器运行sysinfo.py:
from flask import Flask, flash, redirect, render_template, request, session, abort
from printinfo import SysInfo
import json
obj1 = SysInfo("gb")
app = Flask(__name__)
@app.route('/')
def index():
var = json.loads(obj1.displayInfo())
return render_template('index.html',**locals())
@app.route('/healthcheck')
def healthcheck():
return "Ok"
@app.route("/api/all")
def all():
return obj1.displayInfo()
if __name__ == '__main__':
app.run(host='0.0.0.0', port=80)
del obj1
Run Code Online (Sandbox Code Playgroud)
我用 pythonsysinfo.py留在classes/sysinfo文件夹中运行它,一切正常。
所以我决定为我的应用程序运行单元测试。将代码放入classes/tests(也尝试过classes/sysinfo/tests)test_sysinfo.py文件中:
import unittest
import printinfo
from sysinfo import sysinfo
import json
import sys
class TestFlaskApi(unittest.TestCase):
def setUp(self):
self.app = sysinfo.app.test_client()
def simple_test(self):
response = self.app.get('/health')
self.assertEqual(
json.loads(response.get_data().decode(sys.getdefaultencoding())),
{'healthcheck': 'ok'}
)
if __name__ == "__main__":
unittest.main()
Run Code Online (Sandbox Code Playgroud)
当我启动它时,我可以看到错误:
Error Traceback (most recent call last):
File "\Python\Python37-32\lib\unittest\case.py", line 59, in testPartExecutor
yield File "\Python\Python37-32\lib\unittest\case.py", line 615, in run
testMethod() File "\Python\Python37-32\lib\unittest\loader.py", line 34, in testFailure
raise self._exception ImportError: Failed to import test module: test_sysinfo Traceback (most recent call last): File
"\Python\Python37-32\lib\unittest\loader.py", line 154, in
loadTestsFromName
module = __import__(module_name) File "\classes\sysinfo\tests\test_sysinfo.py", line 2, in <module>
import printinfo ModuleNotFoundError: No module named 'printinfo'
Run Code Online (Sandbox Code Playgroud)
我阅读了 StackOverflow 上的几篇文章和一些主题,以了解它与项目结构相关。我尝试创建setup.py和setup.cfg. 我设法用这个设置启动它,但测试仍然不起作用。
您能帮助我进行适用于我的情况的最低设置吗?我找到的所有材料都是针对具体情况编写的,或者过于笼统。我无法将其应用于我的案例。
按照 Flask 教程我编辑__init__.py文件以从这里启动应用程序:
from flask import Flask, flash, redirect, render_template, request, session, abort
from . import printinfo
import json
import os
obj1 = printinfo.SysInfo("gb")
def create_app(test_config=None):
# create and configure the app
app = Flask(__name__, instance_relative_config=True)
app.config.from_mapping(
SECRET_KEY='dev'
)
obj1 = printinfo.SysInfo("gb")
@app.route('/')
def index():
var = json.loads(obj1.displayInfo())
return render_template('index.html', **locals())
@app.route('/healthcheck')
def healthcheck():
return "Ok"
@app.route("/api/all")
def all():
return obj1.displayInfo()
#del obj1
return app
Run Code Online (Sandbox Code Playgroud)
还应该为 Flask 设置环境变量:对于 Linux 和 Mac:
export FLASK_APP=sysinfo
export FLASK_ENV=development
Run Code Online (Sandbox Code Playgroud)
对于Windows cmd,使用set而不是export:
set FLASK_APP=sysinfo
set FLASK_ENV=development
Run Code Online (Sandbox Code Playgroud)
并运行应用程序:
flask run
Run Code Online (Sandbox Code Playgroud)
自开发环境设置以来,它在端口 5000 上的本地主机上运行应用程序。无论如何,我需要添加from . import printinfo到__init__.py.
没有尝试测试,但认为它应该有效。如果感兴趣的话会尽快更新。