如何修改用于测试目的的方法的行为？

发布于 2025-02-11 14:17:50 字数 849 浏览 1 评论 0原文

# file: cisco_driver.py

class myClass:
    def __init__(self, run_commands):
        self.run_commands = run_commands

    def myFunction(self, interface_name):
        command = f"show interface status {interface_name}"
        output = self.run_commands(command)
        return output

函数myFunction以接口名称为输入，例如：myFunction（“ gig1/1/1/1”）
然后生成命令，在这种情况下：>显示接口状态gig1/1/1
然后使用run_commands将SSH用于设备中，执行命令并

以pytest测试的目的检索其输出，我想测试<<强>正在生成的命令，而不是执行此命令。我想绕过run_commands，然后返回命令本身，而不是其输出。

我该怎么做？是否可以？我可以嘲笑这个吗？ monkeypatch？

我希望它的行为：

    def myFunction(interface_name):
        command = f"show interface status {interface_name}"
        return command

原文

# file: cisco_driver.py

class myClass:
    def __init__(self, run_commands):
        self.run_commands = run_commands

    def myFunction(self, interface_name):
        command = f"show interface status {interface_name}"
        output = self.run_commands(command)
        return output

The function myFunction takes an interface name as input, for example: myFunction("Gig1/1/1")
It then generated a command, in this case: show interface status Gig1/1/1
It then uses run_commands to SSH into a device, execute the command, and retrieve its output

For the purpose of pytest testing, I want to test the command that is being generated, and not the execution of this command. I want to bypass run_commands and return the command itself, not the output of it.

How can I accomplish this? Is it possible? Can I mock this? Monkeypatch?

I want it to behave like this:

    def myFunction(interface_name):
        command = f"show interface status {interface_name}"
        return command

分享到QQ

分享到微博

如果你对这篇内容有疑问，欢迎到本站社区发帖提问参与讨论，获取更多帮助，或者扫码二维码加入 Web 技术交流群。

发布评论

需要登录才能够评论，你可以免费注册一个本站的账号。

-小熊_ 2025-02-18 14:17:50

我玩过它，能够找到一个解决方案：

# pytest_file
def test_case():

    # Initialize new instance of class
    # Use None since "run_commands" will be bypassed
    new_instance = myClass(None)

    # New function that will replace the existing run_commands function
    def mock_run_commands(command)
        # Original behavior of run_commands is to create an SSH connection
        # Instead, I want it to return the same command
        return command

    # Redefine run_commands to my customized function
    new_instance.run_commands = mock_run_commands

    # Pass param through class method
    # This will return "show interface status Gig1/1/1"
    selected_command = new_instance.myFunction("Gig1/1/1")

    # Compare the command produced by class method with what expected outcome
    assert selected_command == expected_command

I played around with it and was able to find a solution:

# pytest_file
def test_case():

    # Initialize new instance of class
    # Use None since "run_commands" will be bypassed
    new_instance = myClass(None)

    # New function that will replace the existing run_commands function
    def mock_run_commands(command)
        # Original behavior of run_commands is to create an SSH connection
        # Instead, I want it to return the same command
        return command

    # Redefine run_commands to my customized function
    new_instance.run_commands = mock_run_commands

    # Pass param through class method
    # This will return "show interface status Gig1/1/1"
    selected_command = new_instance.myFunction("Gig1/1/1")

    # Compare the command produced by class method with what expected outcome
    assert selected_command == expected_command

回复收藏 0 原文

~没有更多了~