макет paramiko.SSHClient (). exec_command () ||динамический вывод - PullRequest
Новая
       4

макет paramiko.SSHClient (). exec_command () ||динамический вывод

0 голосов
/ 08 декабря 2018

Я уже видел эту тему.

mock paramiko.SSHClient (). Exec_command ()

У меня есть еще один вопрос по этому поводу, так как я этого не делаюу меня много репутации, я не могу оставлять там комментарии.

В своем коде я передаю различные команды exec_command и ожидаю разные типы ответов, как я могу их высмеять?

Я использовал side_effect для создания функции для динамического вывода, но тестовый файл становится очень большим, я задал похожий вопрос здесь , поэтому спрашиваю.Для каждого набора выходных данных я должен создать новый блок тестовой функции и функции 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")
...