当 Flask 服务器运行另一个线程时 Pytest 挂起

Pytest hangs when a flask server runs another thread

我正在使用 Python3、Flask 0.12 和 Pytest 3.0.7。

我有一个类似这样的烧瓶应用程序:

class AppInitializer:
    def __init__(self):
        pass

    @staticmethod
    def __function_to_be_refreshed():
        while True:
            try:
                time.sleep(5)
            except Exception as e:
                logger.exception(e)


    def create_app(self):
        daemon_thread = threading.Thread(target=self.__function_to_be_refreshed)
        daemon_thread.daemon = True
        daemon_thread.start()
        atexit.register(daemon_thread.join)
        app_ = Flask(__name__)
        return app_


app_initializer = AppInitializer()
app = app_initializer.create_app()

我正尝试使用 pytest 测试此应用程序,如下所示:

import unittest

import pytest


class TestCheckPriceRequestAPI(unittest.TestCase):
    def setUp(self):
        self.app = api.app.test_client()

    def test_random(self):
        pass

当我 运行 使用 pytest 进行此测试时,此测试(连同所有其他测试) 运行 成功但 pytest 挂起。如何停止 运行ning pytest 进程(或者终止守护线程)?

join 命令仅表示线程将等待线程完成,但不会完成它。要以无限循环结束线程,您可以这样做:

class AppInitializer:
    def __init__(self):
        self._run = True
        self._daemon_thread = None

    def __function_to_be_refreshed(self):
        while self._run:
            try:
                time.sleep(5)
            except Exception as e:
                logger.exception(e)

    def __shutdown(self):
        self._run = False
        if self._daemon_thread:
            self._daemon_thread.join()

    def create_app(self):
        self._daemon_thread = threading.Thread(target=self.__function_to_be_refreshed)
        ...
        atexit.register(self.__shutdown)
        ...