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

Question

我对python非常陌生，我想养成在编码之前编写测试的习惯，但我不知道如何成功编写连接到rabbitmq服务器的测试。这是我用烧瓶和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)