如何编写python测试连接到rabbitmq？

Question

我对python很陌生，我想养成在编码之前编写测试的习惯，但我不知道如何成功编写连接到rabbitmq服务器的测试。这是我使用 Flask 和 pytest 进行的测试

class TestVerifyService(BaseTestCase):    

    def test_connect_queue(self):
        """
        Ensure RabbitMQ server is properly configured
        @return:
        @rtype:
        """
        self.assertTrue(os.environ.get('RABBIT_HOST'))
        self.assertTrue(os.getenv('RABBIT_USER'))
        self.assertTrue(os.getenv("RABBIT_PASS"))

这是我正在测试的函数：

def connect_queue():
    """
    Create a connection to RabbitMQ server
    @return: connection, channel
    """
    rabbit_url = f"amqp://{os.getenv('RABBIT_USER')}:{os.getenv('RABBIT_PASS')}@{os.getenv('RABBIT_HOST')}"
    connection = pika.BlockingConnection(pika.URLParameters(rabbit_url))
    channel = connection.channel()
    return connection, channel

请问我如何确保我的测试正确地测试了上述功能？谢谢。

Answer 1

我想出了一个方法来编写正确的测试：

def test_connect_queue(self):
    self.assertTrue(os.environ.get('RABBIT_HOST'))
    self.assertTrue(os.getenv('RABBIT_USER'))
    self.assertTrue(os.getenv("RABBIT_PASS"))
    response = connect_queue()
    self.assertTrue(response)