代码之家 › 专栏 › 技术社区 › Jake

Python unittest使用单个套接字在tcp服务器上进行多个测试

python-unittest tcp sockets python-3.x python

Jake · 技术社区 · 7 年前

我正在尝试测试一个使用TCP套接字的简单服务的实现。我希望在不为每个测试用例创建连接的情况下测试我的服务器。我想为整个测试套件建立一个连接。目前我有这样的东西。。

class Test_service(unittest.TestCase):

    def test_1(self):
        client_socket = socket(AF_INET, SOCK_STREAM)
        client_socket.connect(('', port_number))
        client_socket.send('message1'.encode())
        self.assertEqual(client_socket.recv(1024).decode(), 'reply1')
        client_socket.close()

    def test_2(self):
        client_socket = socket(AF_INET, SOCK_STREAM)
        client_socket.connect(('', port_number))
        client_socket.send('message2'.encode())
        self.assertEqual(client_socket.recv(1024).decode(), 'reply2')
        client_socket.close()

if __name__ == '__main__':
    unittest.main()

保持单个套接字连接打开并让多个测试用例使用该连接的好方法是什么?

1 回复 | 直到 7 年前

Turn Deven 7 年前

如果你看看 docs 你会发现 setUp 和 tearDown 测试类中将执行一次的方法。因此,您可以执行以下操作:

class Test_service(unittest.TestCase):

    def setUp(self):
        client_socket = socket(AF_INET, SOCK_STREAM)
        client_socket.connect(('', port_number))

    def tearDown(self):
        client_socket.close()

    def test_1(self):
        client_socket.send('message1'.encode())
        self.assertEqual(client_socket.recv(1024).decode(), 'reply1')

    def test_2(self):
        client_socket.send('message2'.encode())
        self.assertEqual(client_socket.recv(1024).decode(), 'reply2')

推荐文章

LondonAppDev · 使用botocore stubber时无法识别客户端异常

6 年前

Bob · 测试函数是否已应用于集合的每个项的干净方法?

6 年前

Alex · 如何设置项目结构以便unittest导入工作?

6 年前

lemi57ssss · 如何模拟iInstance测试中使用的类的补丁?

6 年前

Jieke Wei · python side\u effect-方法的模拟行为

7 年前

00__00__00 · 如何将要测试的类传递给单元测试类

7 年前

TimJ · 单元测试Django模型保存函数

7 年前

Paymahn Moghadasian · 无法在Pycharm中调试单元测试

7 年前

Larissa Leite · 当测试一起运行时,所有测试中使用的外部库模拟修补程序不起作用

7 年前

j12y · ResourceWarning Python 3单元测试中未关闭的套接字

7 年前