-1

我正在尝试为我正在处理的一些 Python 代码编写单元测试，并且其中一些代码在完成后会联系一个 API。我正在尝试运行一个简单的 Flask API 来模拟这个 API 并检查代码是否发送了正确的信息。这是代码：

import unittest
import time
from threading import Thread
from flask import Flask
from flask_restx import Api, Resource
from werkzeug.serving import make_server

mock_app = Flask(__name__)
mock_api = Api(mock_app)


# Mock API
data_in = []
data_out = ""
result_code = 200

@mock_api.route('/jobs')
class MockAPI(Resource):
    def post(self):
        global data_in, data_out, result_code
        data_in.append(mock_api.payload)
        return data_out, result_code


# Unit test class
class TestClass(unittest.TestCase):
    
    def __init__(self, arg):
        super().__init__(arg)
        # Some needed fields
        # ...

        # Mock API Server
        self.mock_server = make_server('localhost', 6000, mock_wfm)
        self.mock_server_thread = Thread(target = self.mock_wfm.serve_forever)

调用 to 的行make_server是导致异常的行。诸如lsof -i :6000不返回任何内容、更改地址或端口之类的命令也无法解决任何问题。

编辑：

在故障行前添加一个简单的打印后，我发现代码实际上被调用了两次，导致错误。我还不知道为什么。

4

1 回答 1

0

问题是该__init__方法显然是为每个测试调用的，所以它每次都在第二个测试中失败。

解决的办法是把server和Thread的创建放在setUp方法中，在tearDown中关闭。

    def setUp(self):
        self.mock_server = make_server('localhost', 6000, mock_wfm)
        self.mock_server_thread = Thread(target = self.mock_server.serve_forever)
        self.mock_server_thread.start()
        time.sleep(1)

    def tearDown(self):
        self.mock_server.shutdown()
        self.mock_server_thread.join()

编辑：

这些方法显然是一个更好的解决方案，因为它们每个类运行一次，而不是每个测试运行一次setupClass。teardownClass

于 2021-12-16T18:37:31.890 回答