Question

我已经看到了这个线程。

mock paramiko.SSHClient().exec_command()

对此我还有进一步的疑问，因为我的声誉不高，因此无法在此处发表评论。

在我的代码中，我将各种命令传递给exec_command并期望使用不同类型的响应，我该如何模拟它们？

我已经使用side_effect创建了一个提供动态输出的函数，但是测试文件越来越大，我问了类似的问题here，所以问。对于每组样本输出，我必须创建一个新的测试函数和side_effect函数块，该块会变得非常大。下面是我测试一组输出的代码。

def mock_device_login(device_type, host, user, password):
    fixture_file = os.path.join("tests/fixtures", "device_health_status.json")
    dummy_stuff = json.loads(open(fixture_file, "r").read())
    class mock_class(object):
        def send_command(self, command):
            return_dict = {'sh module' : dummy_stuff['module_output'],
                            'sh env alarm' : dummy_stuff['environment_output'],
                            'sh process cpu | inc CPU utilization' : dummy_stuff['cpu_usage_output'],
                            'sh ip bgp summary' : 'bgp summary',
                            'sh int desc': 'sh int desc',
                            'sh clock' : "*09:01:10.516 UTC Sun Aug 15 1993"
                            }
            return return_dict[command]


    return  { "response": {"logged_in": True, "details": "successfully logged in", "next_action": "execute command" }, "result": True, "connection": mock_class() }


@patch('device_health_status.device_login', side_effect = mock_device_login)
def test_device_status(self, mock_device_login):
    action = self.get_action_instance()

    actionResult,json_result = action.run(device_type, host, username, password)
    print(json_result['error_summary'])
    self.assertTrue(actionResult, True)
    self.assertEquals(json_result['environment_status'], 'GOOD')
    self.assertEquals(json_result['cpu_usage'], 'GOOD')
    self.assertEquals(json_result['module_status'], 'GOOD')
    self.assertEquals(json_result['overall_device_status'], "GOOD")

模拟paramiko.SSHClient（）。exec_command（）||动态输出

0 个答案: