模拟paramiko.SSHClient()。exec_command()||动态输出

时间:2018-12-08 14:07:39

标签: python unit-testing testing mocking paramiko

我已经看到了这个线程。

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")

0 个答案:

没有答案