运行 flask 服务器、nosetests 和覆盖率

Question

我正在用 Flask 编写 api。我有几个 return json 响应的视图，并且我已经编写了一些单元测试来检查这些视图是否正常工作并且 return 正在输入正确的数据。然后我打开了 nosetests 的覆盖插件（在我的例子中是 nose-cov）。

这就是我的问题开始的地方，覆盖范围没有看到我的观点被测试执行。

首先是一些基本代码给你全貌：

我的看法：

def get_user(uid):
    """Retrieve user.

    Args:
        uid (url): valid uid value

    Usage: ::

        GET user/<uid>/

    Returns:
      obj:
       ::

        {
            'data': {
                `response.User`,
            },
            'success': True,
            'status': 'get'
        }
    """
    if not uid:
        raise exception.ValueError("Uid is empty")

    obj = db_layer.user.get_user(uid=value)

    return {
        'data': {
            obj.to_dict(),  # to_dict is helper method that converts part of orm into dict
        },
        'success': True,
        'status': 'get'
    }

我的测试：

class TestUserViews(base.TestViewsBase):
    def test_get_user(self):
        uid = 'some_uid_from_fixtures'
        name = 'some_name_from_fixtures'

        response = self.get(u'user/uid/{}/'.format(uid))
        self.assertEqual(response.status_code, 200)

        user_data = json.loads(response.text)['data']
        self.assertEqual(name, user_data['username'])
        self.assertEqual(uid, user_data['uid'])

    def get(self, method, headers=None):
        """
        Wrapper around requests.get, reassures that authentication is
        sorted for us.
        """
        kwargs = {
            'headers': self._build_headers(headers),
        }

        return requests.get(self.get_url(method), **kwargs)

    def get_url(self, method):
        return '{}/{}/{}'.format(self.domain, self.version, method)

    def _build_headers(self, headers=None):
        if headers is None:
            headers = {}

        headers.update({
            'X-Auth-Token': 'some-token',
            'X-Auth-Token-Test-User-Id': 'some-uid',
        })

        return headers

对于运行测试套件，我有特殊的 shell 脚本可以为我执行一些操作：

#!/usr/bin/env bash

HOST="0.0.0.0"
PORT="5001"

ENVS="PYTHONPATH=$PYTHONPATH:$PWD"

# start server 
START_SERVER="$ENVS python $PWD/server.py --port=$PORT --host=$HOST"

eval "$START_SERVER&"

PID=$!

eval "$ENVS nosetests -s --nologcapture --cov-report html --with-cov"

kill -9 $PID

之后报告该视图未被执行。

Answer 1

好的伙计们，12 小时后我找到了解决方案。我检查了烧瓶、werkzeug、请求、子进程和线程库。只知道那个问题在别的地方。其实解决起来很简单。必须修改的代码位是 server.py 的执行。我们也需要覆盖它，然后合并 server.py 生成的结果和 nosetests 生成的结果。修改后的 test-runner.sh 看起来如下：

#!/usr/bin/env bash

HOST="0.0.0.0"
PORT="5001"

ENVS="COVERAGE_PROCESS_START=$PWD/.apirc PYTHONPATH=$PYTHONPATH:$PWD"

START_SERVER="$ENVS coverage run --rcfile=.apirc $PWD/server.py --port=$PORT --host=$HOST"

eval "$START_SERVER&"

eval "$ENVS nosetests -s --nologcapture --cov-config=.apirc --cov-report html --with-cov"

# this is important bit, we have to stop flask server gracefully otherwise
# coverage won't get a chance to collect and save all results
eval "curl -X POST http://$HOST:$PORT/0.0/shutdown/"

# this will merge results from both coverage runs
coverage combine  --rcfile=.apirc
coverage html --rcfile=.apirc

我的 .apirc 如下所示：

[run]
branch = True
parallel = True
source = files_to_cover/

[html]
directory = cover

我们需要做的最后一件事是在我们的烧瓶中构建视图，这将使我们能够优雅地关闭服务器。以前我用 kill -9 粗暴地杀死它，它不仅可以杀死服务器，还可以杀死覆盖范围。

遵循此代码段：http://flask.pocoo.org/snippets/67/

我的看法是这样的：

def shutdown():
    if config.SHUTDOWN_ALLOWED:
        func = request.environ.get('werkzeug.server.shutdown')
        if func is None:
            raise RuntimeError('Not running with the Werkzeug Server')
        func()

        return 'Server shutting down...'

使用 nose-cov 而不是标准覆盖插件很重要，因为它使用 rcfile 并允许配置更多。在我们的案例中，平行是关键，请注意 data_files 变量不适用于鼻子覆盖，因此您不能在 .apirc 中覆盖它，您必须使用默认值。

完成所有这些之后，您的覆盖范围将几乎闪耀着有效值。

希望对外面的人有所帮助。

运行 flask 服务器、nosetests 和覆盖率

Running flask server, nosetests and coverage

python

code-coverage

nosetests

flask