Question

我尝试使用flask和unittest创建简单的Python项目。结构非常简单：

classes
  |-sysinfo
      |static
      |templates
         |- index.html
         |- layout.html
      |__init__.py
      |sysinfo.py
      |printinfo.py
  tests
  |test_sysinfo.py
README.md
requirments.txt

printinfo.py中非常简单的类：

 #!/usr/bin/python
import psutil
import json
class SysInfo:
.......
    def displayInfo(self):
        .......
        return json.dumps(self.__data)

简单的烧瓶服务器运行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

我使用python sysinfo.py放在classes/sysinfo文件夹中运行它，一切正常。

因此，我决定为我的应用程序运行unittest。放入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()

当我启动它时，我会看到错误：

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'

我阅读了几篇文章，在StackOverflow上了解了一些主题，以了解它与项目结构有关。我尝试创建setup.py和setup.cfg。我设法通过此设置启动了它，但是测试仍然无法进行。

能否请您帮我解决适用于我的情况的最少设置？我发现的所有材料都是针对特定情况或太笼统编写的。我无法将其应用于我的案件。

Answer 1

跟随烧瓶教程，我编辑__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

还应为Flask设置环境变量：对于Linux和Mac：

export FLASK_APP=sysinfo
export FLASK_ENV=development

对于Windows cmd，请使用set而不是export：

set FLASK_APP=sysinfo
set FLASK_ENV=development

并运行应用程序：

flask run

自从设置开发环境以来，它在端口5000的localhost上运行应用程序。无论如何，我需要将from . import printinfo添加到__init__.py中。

没有尝试测试，但是认为它应该可以工作。如果有兴趣的话会很快更新。

Answer 2

仅遵循Flask教程。创建了conftest.py和test_factory.py。使用pytest运行确定：

import pytest
from sysinfo import create_app
from sysinfo import printinfo

@pytest.fixture
def app():
    app = create_app({
        'TESTING': True
    })

    with app.app_context():
        printinfo.SysInfo("gb")

    yield app

unittest可能使用相同的设置。没有尝试。

python unittest ModuleNotFoundError：没有名为<...>的模块

2 个答案: