我已经看到了这个线程。
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")